V2: Added Webserver for faster formatting

tests/test_formatter.py

@@ -0,0 +1,130 @@
+from datetime import date
+from hashlib import sha256
+from pathlib import Path
+from tempfile import TemporaryDirectory
+import unittest
+
+import fitz  # type: ignore
+from PIL import Image
+
+from roster_formatter import (
+    DEFAULT_LAYOUT,
+    PAGE_HEIGHT,
+    PAGE_WIDTH,
+    PNG_DPI,
+    PNG_HEIGHT,
+    PNG_WIDTH,
+    LayoutSettings,
+    RosterFormatError,
+    RosterPeriod,
+    format_roster,
+    validate_input_pdf,
+)
+from tests.helpers import make_geometry_pdf
+
+
+LEGACY_GEOMETRY_SHA256 = "8b000127605f4fd78f9f39606de955eb02392ca113e8ddd808421d4f04c09c31"
+
+
+class RosterPeriodTests(unittest.TestCase):
+    def test_original_month_abbreviation_rules_are_preserved(self) -> None:
+        expected = {
+            1: "JAN 2026",
+            2: "FEB 2026",
+            3: "MARCH 2026",
+            4: "APRIL 2026",
+            5: "MAY 2026",
+            6: "JUNE 2026",
+            7: "JULY 2026",
+            8: "AUG 2026",
+            9: "SEP 2026",
+            10: "OCT 2026",
+            11: "NOV 2026",
+            12: "DEC 2026",
+        }
+        self.assertEqual(
+            expected,
+            {month: RosterPeriod(2026, month).label for month in range(1, 13)},
+        )
+
+    def test_suggested_period_changes_on_the_fifteenth(self) -> None:
+        self.assertEqual(RosterPeriod(2026, 7), RosterPeriod.suggested(date(2026, 7, 14)))
+        self.assertEqual(RosterPeriod(2026, 8), RosterPeriod.suggested(date(2026, 7, 15)))
+        self.assertEqual(RosterPeriod(2027, 1), RosterPeriod.suggested(date(2026, 12, 31)))
+
+    def test_parse_accepts_browser_and_cli_formats(self) -> None:
+        self.assertEqual(RosterPeriod(2026, 9), RosterPeriod.parse("2026-09"))
+        self.assertEqual(RosterPeriod(2026, 9), RosterPeriod.parse("September 2026"))
+        self.assertEqual(RosterPeriod(2026, 9), RosterPeriod.parse("SEP 2026"))
+
+    def test_parse_rejects_untrusted_filename_text(self) -> None:
+        with self.assertRaises(RosterFormatError):
+            RosterPeriod.parse("../../September 2026")
+
+
+class LayoutRegressionTests(unittest.TestCase):
+    def test_production_defaults_are_explicit(self) -> None:
+        self.assertEqual(792, PAGE_WIDTH)
+        self.assertEqual(612, PAGE_HEIGHT)
+        self.assertEqual(150, PNG_DPI)
+        self.assertEqual(1650, PNG_WIDTH)
+        self.assertEqual(1275, PNG_HEIGHT)
+        self.assertEqual(
+            LayoutSettings(
+                content_shift=100,
+                bottom_margin=60,
+                logo_x=50,
+                logo_y=-10,
+                logo_scale=0.27,
+                date_x=460,
+                date_y=88,
+                date_font_size=48,
+            ),
+            DEFAULT_LAYOUT,
+        )
+
+    def test_legacy_default_is_byte_identical(self) -> None:
+        with TemporaryDirectory() as temp_dir:
+            root = Path(temp_dir)
+            source = make_geometry_pdf(root / "source.pdf")
+            output = root / "output.png"
+            format_roster(source, output, RosterPeriod(2026, 7))
+
+            self.assertEqual(LEGACY_GEOMETRY_SHA256, sha256(output.read_bytes()).hexdigest())
+            with Image.open(output) as image:
+                self.assertEqual((PNG_WIDTH, PNG_HEIGHT), image.size)
+                self.assertEqual("RGB", image.mode)
+
+    def test_explicit_adjustment_changes_pixels_but_not_canvas(self) -> None:
+        with TemporaryDirectory() as temp_dir:
+            root = Path(temp_dir)
+            source = make_geometry_pdf(root / "source.pdf")
+            default_output = root / "default.png"
+            adjusted_output = root / "adjusted.png"
+            format_roster(source, default_output, RosterPeriod(2026, 7))
+            format_roster(
+                source,
+                adjusted_output,
+                RosterPeriod(2026, 7),
+                LayoutSettings(logo_x=55),
+            )
+
+            self.assertNotEqual(default_output.read_bytes(), adjusted_output.read_bytes())
+            with Image.open(adjusted_output) as image:
+                self.assertEqual((PNG_WIDTH, PNG_HEIGHT), image.size)
+
+    def test_input_geometry_is_validated(self) -> None:
+        with TemporaryDirectory() as temp_dir:
+            source = Path(temp_dir) / "portrait.pdf"
+            doc = fitz.open()
+            page = doc.new_page(width=612, height=792)
+            page.insert_text((20, 20), "not landscape")
+            doc.save(source)
+            doc.close()
+
+            with self.assertRaisesRegex(RosterFormatError, "landscape"):
+                validate_input_pdf(source)
+
+
+if __name__ == "__main__":
+    unittest.main()