paradizelost/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-04-09 09:58:20 -05:00
Code Issues Packages Projects Releases Wiki Activity

Missed one mock

Browse Source
This commit is contained in:
Trenton Holmes 2022-09-15 19:58:14 -07:00 committed by Trenton H
parent f7cd6974c5
commit 1262c121f0
1 changed files with 19 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
src/documents/tests/test_tasks.py
View File

@ -11,6 +11,7 @@ from documents.models import DocumentType
from documents.models import Tag from documents.models import Tag
from documents.sanity_checker import SanityCheckFailedException from documents.sanity_checker import SanityCheckFailedException
from documents.sanity_checker import SanityCheckMessages from documents.sanity_checker import SanityCheckMessages
from documents.tests.test_classifier import dummy_preprocess
from documents.tests.utils import DirectoriesMixin from documents.tests.utils import DirectoriesMixin
@ -75,21 +76,26 @@ class TestClassifier(DirectoriesMixin, TestCase):
doc = Document.objects.create(correspondent=c, content="test", title="test") doc = Document.objects.create(correspondent=c, content="test", title="test")
self.assertFalse(os.path.isfile(settings.MODEL_FILE)) self.assertFalse(os.path.isfile(settings.MODEL_FILE))
tasks.train_classifier() with mock.patch(
self.assertTrue(os.path.isfile(settings.MODEL_FILE)) "documents.classifier.DocumentClassifier.preprocess_content",
mtime = os.stat(settings.MODEL_FILE).st_mtime ) as pre_proc_mock:
pre_proc_mock.side_effect = dummy_preprocess
tasks.train_classifier() tasks.train_classifier()
self.assertTrue(os.path.isfile(settings.MODEL_FILE)) self.assertTrue(os.path.isfile(settings.MODEL_FILE))
mtime2 = os.stat(settings.MODEL_FILE).st_mtime mtime = os.stat(settings.MODEL_FILE).st_mtime
self.assertEqual(mtime, mtime2)
doc.content = "test2" tasks.train_classifier()
doc.save() self.assertTrue(os.path.isfile(settings.MODEL_FILE))
tasks.train_classifier() mtime2 = os.stat(settings.MODEL_FILE).st_mtime
self.assertTrue(os.path.isfile(settings.MODEL_FILE)) self.assertEqual(mtime, mtime2)
mtime3 = os.stat(settings.MODEL_FILE).st_mtime
self.assertNotEqual(mtime2, mtime3) doc.content = "test2"
doc.save()
tasks.train_classifier()
self.assertTrue(os.path.isfile(settings.MODEL_FILE))
mtime3 = os.stat(settings.MODEL_FILE).st_mtime
self.assertNotEqual(mtime2, mtime3)
class TestSanityCheck(DirectoriesMixin, TestCase): class TestSanityCheck(DirectoriesMixin, TestCase):