paradizelost/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-05-21 12:52:13 -05:00
Code Issues Packages Projects Releases Wiki Activity

Fix classifier test, use Path

Browse Source
This commit is contained in:
shamoon 2024-12-25 10:00:20 -08:00 committed by Sebastian Steinbeißer
parent 02386d6513
commit 4993bcf6ab
No known key found for this signature in database
GPG Key ID: E8200F0F0380C0E0
2 changed files with 21 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/documents/tests/test_api_status.py
View File

@ -233,7 +233,11 @@ class TestSystemStatus(APITestCase):
self.assertEqual(response.data["tasks"]["classifier_status"], "WARNING") self.assertEqual(response.data["tasks"]["classifier_status"], "WARNING")
self.assertIsNotNone(response.data["tasks"]["classifier_error"]) self.assertIsNotNone(response.data["tasks"]["classifier_error"])
def test_system_status_classifier_error(self): @mock.patch(
"documents.classifier.load_classifier",
side_effect=ClassifierModelCorruptError(),
)
def test_system_status_classifier_error(self, mock_load_classifier):
""" """
GIVEN: GIVEN:
- The classifier does exist but is corrupt - The classifier does exist but is corrupt
@ -250,8 +254,6 @@ class TestSystemStatus(APITestCase):
) as does_exist, ) as does_exist,
override_settings(MODEL_FILE=Path(does_exist.name)), override_settings(MODEL_FILE=Path(does_exist.name)),
): ):
with mock.patch("documents.classifier.load_classifier") as mock_load:
mock_load.side_effect = ClassifierModelCorruptError()
Document.objects.create( Document.objects.create(
title="Test Document", title="Test Document",
) )

2
src/documents/views.py
View File

@ -2159,7 +2159,7 @@ class SystemStatusView(PassUserMixin):
matching_algorithm=Tag.MATCH_AUTO, matching_algorithm=Tag.MATCH_AUTO,
).exists() ).exists()
) )
and not os.path.isfile(settings.MODEL_FILE) and not settings.MODEL_FILE.exists()
): ):
# if classifier file doesn't exist just classify as a warning # if classifier file doesn't exist just classify as a warning
classifier_error = "Classifier file does not exist (yet). Re-training may be pending." classifier_error = "Classifier file does not exist (yet). Re-training may be pending."