From 595603f695afde161f174e0f8ae99ca9febf7833 Mon Sep 17 00:00:00 2001 From: shamoon <4887959+shamoon@users.noreply.github.com> Date: Wed, 11 Feb 2026 23:06:50 -0800 Subject: [PATCH] mypy too --- src/documents/tests/test_consumer.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/documents/tests/test_consumer.py b/src/documents/tests/test_consumer.py index a9e3e949a..cea01bc16 100644 --- a/src/documents/tests/test_consumer.py +++ b/src/documents/tests/test_consumer.py @@ -679,6 +679,7 @@ class TestConsumer( consumer.run() document = Document.objects.first() + assert document is not None self.assertEqual(document.version_label, "v1") @@ -1260,7 +1261,7 @@ class PostConsumeTestCase(DirectoriesMixin, GetConsumerMixin, TestCase): consumer.run_post_consume_script(doc) @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 override_settings(POST_CONSUME_SCRIPT=script.name): doc = Document.objects.create(title="Test", mime_type="application/pdf") @@ -1271,7 +1272,10 @@ class PostConsumeTestCase(DirectoriesMixin, GetConsumerMixin, TestCase): m.assert_called_once() @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 override_settings(POST_CONSUME_SCRIPT=script.name): c = Correspondent.objects.create(name="my_bank")