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

Make this dumber

Browse Source
This commit is contained in:
shamoon
2026-02-12 22:36:16 -08:00
parent e049f3c7de
commit 2bb73627d6
1 changed files with 5 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -103,13 +103,8 @@ def fake_magic_from_file(file, *, mime=False): # NOSONAR
filepath = Path(file) filepath = Path(file)
if filepath.name.startswith("invalid_pdf"): if filepath.name.startswith("invalid_pdf"):
return "application/octet-stream" return "application/octet-stream"
if filepath.suffix == "": if filepath.name.startswith("valid_pdf"):
try: return "application/pdf"
with Path(filepath).open("rb") as handle:
if handle.read(4) == b"%PDF":
return "application/pdf"
except OSError:
pass
if filepath.suffix == ".pdf": if filepath.suffix == ".pdf":
return "application/pdf" return "application/pdf"
elif filepath.suffix == ".png": elif filepath.suffix == ".png":
@@ -772,7 +767,7 @@ class TestConsumer(
version_file = self.get_test_file2() version_file = self.get_test_file2()
status = DummyProgressManager(version_file.name, None) status = DummyProgressManager(version_file.name, None)
overrides = DocumentMetadataOverrides( overrides = DocumentMetadataOverrides(
filename="version-upload", filename="valid_pdf_version-upload",
actor_id=999999, actor_id=999999,
) )
doc = ConsumableDocument( doc = ConsumableDocument(
@@ -800,7 +795,7 @@ class TestConsumer(
) )
consumer.setup() consumer.setup()
try: try:
self.assertEqual(consumer.filename, "version-upload_v0") self.assertEqual(consumer.filename, "valid_pdf_version-upload_v0")
consumer.run() consumer.run()
finally: finally:
consumer.cleanup() consumer.cleanup()
@@ -810,7 +805,7 @@ class TestConsumer(
) )
self.assertIsNotNone(version) self.assertIsNotNone(version)
assert version is not None 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)) self.assertTrue(bool(version.content))
@mock.patch("documents.consumer.load_classifier") @mock.patch("documents.consumer.load_classifier")