Organized

tests/test_ocr_categories.py

@@ -0,0 +1,81 @@
+#!/usr/bin/env python3
+from __future__ import annotations
+
+import tempfile
+import unittest
+from pathlib import Path
+
+from redmine_ticket_ocr import (
+    build_ocr_note,
+    extract_text_from_file,
+    looks_like_generic_scan_subject,
+    suggest_subject_from_text,
+)
+from redmine_tickets import KNOWN_COMMANDS, build_parser, inject_default_sync_command, is_closed_status
+
+
+class OcrSuggestTest(unittest.TestCase):
+    def test_generic_subject(self):
+        self.assertTrue(looks_like_generic_scan_subject("Scanned document from HP"))
+        self.assertFalse(looks_like_generic_scan_subject("Invoice — Acme Corp"))
+
+    def test_suggest_w2(self):
+        text = "Form W-2 Wage and Tax Statement\nName: Jane Doe\nCompany: Acme LLC\nJanuary 15, 2026"
+        subj = suggest_subject_from_text(text, original_subject="Scanned document")
+        self.assertIn("W-2", subj)
+        self.assertIn("Jane Doe", subj)
+
+    def test_suggest_vendor_invoice_skips_fee_bullet(self):
+        text = (
+            "- Return Check Fee up to $45.00\n"
+            "- A 1.5% Service Charge, 18% Annually ($1.00 Minimum) will be charged\n"
+            "Best Way Disposal\n"
+            "Invoice Number: 2404778\n"
+            "Invoice Due Date 9/25/2026\n"
+            "Total Due $307.52\n"
+        )
+        subj = suggest_subject_from_text(text, original_subject="Scanned document from LDR Office")
+        self.assertIn("Invoice 2404778", subj)
+        self.assertIn("Best Way Disposal", subj)
+        self.assertNotIn("Return Check Fee", subj)
+
+    def test_txt_extract(self):
+        with tempfile.TemporaryDirectory() as td:
+            p = Path(td) / "a.txt"
+            p.write_text("hello ocr world " * 5, encoding="utf-8")
+            text, method = extract_text_from_file(p)
+            self.assertEqual("read", method)
+            self.assertIn("hello", text)
+
+    def test_note_builder(self):
+        note = build_ocr_note(
+            issue_id=1,
+            files=[{"filename": "a.pdf", "method": "pdftotext", "text": "body"}],
+            proposed_subject="Invoice — X",
+        )
+        self.assertIn("proposed_subject: Invoice — X", note)
+        self.assertIn("a.pdf", note)
+
+
+class CliRegistrationTest(unittest.TestCase):
+    def test_commands(self):
+        self.assertIn("ocr-rename", KNOWN_COMMANDS)
+        self.assertIn("categories", KNOWN_COMMANDS)
+        args = build_parser().parse_args(
+            inject_default_sync_command(
+                ["ocr-rename", "--issue-id", "1", "--no-queue-note", "--no-queue-subject"]
+            )
+        )
+        self.assertEqual("ocr-rename", args.command)
+        args2 = build_parser().parse_args(
+            inject_default_sync_command(["categories", "list", "--refresh"])
+        )
+        self.assertEqual("categories", args2.command)
+        args3 = build_parser().parse_args(
+            inject_default_sync_command(["walk", "--open-only", "--max-tickets", "3"])
+        )
+        self.assertTrue(args3.open_only)
+
+
+if __name__ == "__main__":
+    unittest.main()