From 2bb73627d6b17ebd6fc9f4a9d4dffd3c240fa0a1 Mon Sep 17 00:00:00 2001 From: shamoon <4887959+shamoon@users.noreply.github.com> Date: Thu, 12 Feb 2026 22:36:16 -0800 Subject: [PATCH] Make this dumber --- src/documents/tests/test_consumer.py | 15 +++++---------- 1 file changed, 5 insertions(+), 10 deletions(-) diff --git a/src/documents/tests/test_consumer.py b/src/documents/tests/test_consumer.py index 01cf93c62..551719807 100644 --- a/src/documents/tests/test_consumer.py +++ b/src/documents/tests/test_consumer.py @@ -103,13 +103,8 @@ def fake_magic_from_file(file, *, mime=False): # NOSONAR filepath = Path(file) if filepath.name.startswith("invalid_pdf"): return "application/octet-stream" - if filepath.suffix == "": - try: - with Path(filepath).open("rb") as handle: - if handle.read(4) == b"%PDF": - return "application/pdf" - except OSError: - pass + if filepath.name.startswith("valid_pdf"): + return "application/pdf" if filepath.suffix == ".pdf": return "application/pdf" elif filepath.suffix == ".png": @@ -772,7 +767,7 @@ class TestConsumer( version_file = self.get_test_file2() status = DummyProgressManager(version_file.name, None) overrides = DocumentMetadataOverrides( - filename="version-upload", + filename="valid_pdf_version-upload", actor_id=999999, ) doc = ConsumableDocument( @@ -800,7 +795,7 @@ class TestConsumer( ) consumer.setup() try: - self.assertEqual(consumer.filename, "version-upload_v0") + self.assertEqual(consumer.filename, "valid_pdf_version-upload_v0") consumer.run() finally: consumer.cleanup() @@ -810,7 +805,7 @@ class TestConsumer( ) self.assertIsNotNone(version) assert version is not None - self.assertEqual(version.original_filename, "version-upload_v0") + self.assertEqual(version.original_filename, "valid_pdf_version-upload_v0") self.assertTrue(bool(version.content)) @mock.patch("documents.consumer.load_classifier")