(self, mock_get_database)
| 483 | |
| 484 | @patch('celery.backends.mongodb.MongoBackend._get_database') |
| 485 | def test_save_group(self, mock_get_database): |
| 486 | self.backend.groupmeta_collection = MONGODB_GROUP_COLLECTION |
| 487 | |
| 488 | mock_database = MagicMock(spec=['__getitem__', '__setitem__']) |
| 489 | mock_collection = Mock() |
| 490 | |
| 491 | mock_get_database.return_value = mock_database |
| 492 | mock_database.__getitem__.return_value = mock_collection |
| 493 | res = [self.app.AsyncResult(i) for i in range(3)] |
| 494 | ret_val = self.backend._save_group( |
| 495 | sentinel.taskset_id, res, |
| 496 | ) |
| 497 | mock_get_database.assert_called_once_with() |
| 498 | mock_database.__getitem__.assert_called_once_with( |
| 499 | MONGODB_GROUP_COLLECTION, |
| 500 | ) |
| 501 | mock_collection.replace_one.assert_called_once_with(ANY, ANY, |
| 502 | upsert=True) |
| 503 | assert res == ret_val |
| 504 | |
| 505 | @patch('celery.backends.mongodb.MongoBackend._get_database') |
| 506 | def test_restore_group(self, mock_get_database): |
nothing calls this directly
no test coverage detected