paradizelost/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2026-02-16 00:19:32 -06:00
Code Issues Packages Projects Releases Wiki Activity

mypy too

Browse Source
This commit is contained in:
shamoon
2026-02-11 23:06:50 -08:00
parent c414857ac4
commit 595603f695
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -679,6 +679,7 @@ class TestConsumer(
consumer.run() consumer.run()
document = Document.objects.first() document = Document.objects.first()
assert document is not None
self.assertEqual(document.version_label, "v1") self.assertEqual(document.version_label, "v1")
@@ -1260,7 +1261,7 @@ class PostConsumeTestCase(DirectoriesMixin, GetConsumerMixin, TestCase):
consumer.run_post_consume_script(doc) consumer.run_post_consume_script(doc)
@mock.patch("documents.consumer.run_subprocess") @mock.patch("documents.consumer.run_subprocess")
def test_post_consume_script_simple(self, m) -> None: def test_post_consume_script_simple(self, m: mock.MagicMock) -> None:
with tempfile.NamedTemporaryFile() as script: with tempfile.NamedTemporaryFile() as script:
with override_settings(POST_CONSUME_SCRIPT=script.name): with override_settings(POST_CONSUME_SCRIPT=script.name):
doc = Document.objects.create(title="Test", mime_type="application/pdf") doc = Document.objects.create(title="Test", mime_type="application/pdf")
@@ -1271,7 +1272,10 @@ class PostConsumeTestCase(DirectoriesMixin, GetConsumerMixin, TestCase):
m.assert_called_once() m.assert_called_once()
@mock.patch("documents.consumer.run_subprocess") @mock.patch("documents.consumer.run_subprocess")
def test_post_consume_script_with_correspondent_and_type(self, m) -> None: def test_post_consume_script_with_correspondent_and_type(
self,
m: mock.MagicMock,
) -> None:
with tempfile.NamedTemporaryFile() as script: with tempfile.NamedTemporaryFile() as script:
with override_settings(POST_CONSUME_SCRIPT=script.name): with override_settings(POST_CONSUME_SCRIPT=script.name):
c = Correspondent.objects.create(name="my_bank") c = Correspondent.objects.create(name="my_bank")