From 1e7aa283c2f6ea0ceb3e4064dd4196dfe67aaa09 Mon Sep 17 00:00:00 2001 From: Mouse Reeve Date: Sat, 2 Jan 2021 09:08:37 -0800 Subject: [PATCH] Patches celery calls in connector tests --- bookwyrm/tests/connectors/test_abstract_connector.py | 7 +++++-- bookwyrm/tests/models/test_import_model.py | 10 ++++++---- 2 files changed, 11 insertions(+), 6 deletions(-) diff --git a/bookwyrm/tests/connectors/test_abstract_connector.py b/bookwyrm/tests/connectors/test_abstract_connector.py index 6bcfc76b..6e912858 100644 --- a/bookwyrm/tests/connectors/test_abstract_connector.py +++ b/bookwyrm/tests/connectors/test_abstract_connector.py @@ -1,4 +1,5 @@ ''' testing book data connectors ''' +from unittest.mock import patch from django.test import TestCase import responses @@ -104,8 +105,10 @@ class AbstractConnector(TestCase): 'https://example.com/book/abcd', json=self.edition_data ) - result = self.connector.get_or_create_book( - 'https://example.com/book/abcd') + with patch( + 'bookwyrm.connectors.abstract_connector.load_more_data.delay'): + result = self.connector.get_or_create_book( + 'https://example.com/book/abcd') self.assertEqual(result, self.book) self.assertEqual(models.Edition.objects.count(), 1) self.assertEqual(models.Edition.objects.count(), 1) diff --git a/bookwyrm/tests/models/test_import_model.py b/bookwyrm/tests/models/test_import_model.py index 4781aa37..3d890141 100644 --- a/bookwyrm/tests/models/test_import_model.py +++ b/bookwyrm/tests/models/test_import_model.py @@ -165,9 +165,11 @@ class ImportJob(TestCase): status=200) with patch( - 'bookwyrm.connectors.connector_manager.first_search_result' - ) as search: - search.return_value = result - book = self.item_1.get_book_from_isbn() + 'bookwyrm.connectors.abstract_connector.load_more_data.delay'): + with patch( + 'bookwyrm.connectors.connector_manager.first_search_result' + ) as search: + search.return_value = result + book = self.item_1.get_book_from_isbn() self.assertEqual(book.title, 'Sabriel')