From d4ee469b01d5c46b3a8b825c67c578f7588adff7 Mon Sep 17 00:00:00 2001 From: Trenton H <797416+stumpylog@users.noreply.github.com> Date: Thu, 1 Feb 2024 19:22:07 -0800 Subject: [PATCH] More test fixing for coverage --- src/documents/tests/test_api_documents.py | 6 +++++- src/documents/views.py | 2 +- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/src/documents/tests/test_api_documents.py b/src/documents/tests/test_api_documents.py index e75ad4174..038bc859b 100644 --- a/src/documents/tests/test_api_documents.py +++ b/src/documents/tests/test_api_documents.py @@ -1303,7 +1303,10 @@ class TestDocumentApi(DirectoriesMixin, DocumentConsumeDelayMixin, APITestCase): classifier_checksum = b"thisisachecksum" - mocked_load.return_value = mock.Mock(last_auto_type_hash=classifier_checksum) + mocked_load.side_effect = [ + mock.Mock(last_auto_type_hash=classifier_checksum), + mock.Mock(last_auto_type_hash=classifier_checksum), + ] last_modified = timezone.now() cache.set(CLASSIFIER_MODIFIED_KEY, last_modified, CACHE_50_MINUTES) @@ -1351,6 +1354,7 @@ class TestDocumentApi(DirectoriesMixin, DocumentConsumeDelayMixin, APITestCase): response = self.client.get(f"/api/documents/{doc.pk}/suggestions/") self.assertEqual(response.status_code, status.HTTP_200_OK) + self.assertFalse(True) @mock.patch("documents.parsers.parse_date_generator") @override_settings(NUMBER_OF_SUGGESTED_DATES=0) diff --git a/src/documents/views.py b/src/documents/views.py index c398f1f9e..5c1b9b24c 100644 --- a/src/documents/views.py +++ b/src/documents/views.py @@ -525,7 +525,7 @@ class DocumentViewSet( if classifier is not None: cache.set( doc_key, - (classifier.last_auto_type_hash.decode(), resp_data), + (classifier.last_auto_type_hash, resp_data), CACHE_5_MINUTES, )