(
self,
mock_chord_error_from_stack,
mock_maybe_signature,
mock_group_result_restore,
mock_firestore_ttl,
)
| 386 | @patch('celery.backends.gcs.maybe_signature') |
| 387 | @patch.object(GCSBackend, 'chord_error_from_stack') |
| 388 | def test_restore_deps( |
| 389 | self, |
| 390 | mock_chord_error_from_stack, |
| 391 | mock_maybe_signature, |
| 392 | mock_group_result_restore, |
| 393 | mock_firestore_ttl, |
| 394 | ): |
| 395 | gid = 'group_id' |
| 396 | request = MagicMock() |
| 397 | mock_group_result_restore.return_value = MagicMock() |
| 398 | |
| 399 | backend = GCSBackend(app=self.app) |
| 400 | deps = backend._restore_deps(gid, request) |
| 401 | |
| 402 | mock_group_result_restore.assert_called_once_with( |
| 403 | gid, backend=backend |
| 404 | ) |
| 405 | assert deps is not None |
| 406 | mock_chord_error_from_stack.assert_not_called() |
| 407 | |
| 408 | mock_group_result_restore.side_effect = Exception('restore error') |
| 409 | deps = backend._restore_deps(gid, request) |
| 410 | mock_maybe_signature.assert_called_with(request.chord, app=self.app) |
| 411 | mock_chord_error_from_stack.assert_called_once() |
| 412 | assert deps is None |
| 413 | |
| 414 | mock_group_result_restore.side_effect = None |
| 415 | mock_group_result_restore.return_value = None |
| 416 | deps = backend._restore_deps(gid, request) |
| 417 | mock_chord_error_from_stack.assert_called() |
| 418 | assert deps is None |
| 419 | |
| 420 | @patch.object(GCSBackend, '_is_firestore_ttl_policy_enabled') |
| 421 | @patch.object(GCSBackend, '_firestore_document') |
nothing calls this directly
no test coverage detected