Minor tweak to password test to ensure the right lines were hit

This commit is contained in:
Trenton H 2023-01-27 11:11:09 -08:00
parent 4fce5aba63
commit 9784ea4a60

View File

@ -845,9 +845,14 @@ class TestBarcode(DirectoriesMixin, TestCase):
- Scanning handles the exception without crashing - Scanning handles the exception without crashing
""" """
test_file = os.path.join(self.SAMPLE_DIR, "password-is-test.pdf") test_file = os.path.join(self.SAMPLE_DIR, "password-is-test.pdf")
doc_barcode_info = barcodes.scan_file_for_barcodes( with self.assertLogs("paperless.barcodes", level="WARNING") as cm:
test_file, doc_barcode_info = barcodes.scan_file_for_barcodes(
) test_file,
)
warning = cm.output[0]
expected_str = "WARNING:paperless.barcodes:File is likely password protected, not checking for barcodes"
self.assertTrue(warning.startswith(expected_str))
separator_page_numbers = barcodes.get_separating_barcodes( separator_page_numbers = barcodes.get_separating_barcodes(
doc_barcode_info.barcodes, doc_barcode_info.barcodes,
) )