clean up calendar display, added separated device runtime overrides

backend/test_runtime_overrides.py

@@ -0,0 +1,62 @@
+from __future__ import annotations
+
+import csv
+from pathlib import Path
+import sys
+from tempfile import TemporaryDirectory
+import unittest
+
+sys.path.append(str(Path(__file__).resolve().parent))
+
+from db import DEVICE_DUT, DEVICE_REF
+from parser import P2P_COE_REQUIRED_COLUMNS, parse_target_csv
+
+
+def _write_p2p_csv(directory: str, test_id: str = "P2PRXAX001") -> Path:
+    path = Path(directory) / "runtime-overrides.csv"
+    row = {column: "" for column in P2P_COE_REQUIRED_COLUMNS}
+    row["Priority"] = "1"
+    row["Index"] = "1"
+    row["TC ID"] = test_id
+
+    with path.open("w", encoding="utf-8", newline="") as handle:
+        writer = csv.DictWriter(handle, fieldnames=P2P_COE_REQUIRED_COLUMNS)
+        writer.writeheader()
+        writer.writerow(row)
+
+    return path
+
+
+class RuntimeOverridesParseTests(unittest.TestCase):
+    def test_parse_target_csv_applies_device_specific_runtime_overrides(self) -> None:
+        with TemporaryDirectory() as tmpdir:
+            csv_path = _write_p2p_csv(tmpdir)
+
+            parsed = parse_target_csv(
+                csv_path,
+                runtime_overrides={
+                    DEVICE_DUT: {"P2P": 91},
+                    DEVICE_REF: {"P2P": 123},
+                },
+            )
+
+        by_device = {record.device: record for record in parsed.tests}
+        self.assertEqual(by_device[DEVICE_DUT].estimated_minutes, 91)
+        self.assertEqual(by_device[DEVICE_REF].estimated_minutes, 123)
+
+    def test_parse_target_csv_keeps_legacy_runtime_override_for_both_devices(self) -> None:
+        with TemporaryDirectory() as tmpdir:
+            csv_path = _write_p2p_csv(tmpdir)
+
+            parsed = parse_target_csv(
+                csv_path,
+                runtime_overrides={"P2P": 88},
+            )
+
+        by_device = {record.device: record for record in parsed.tests}
+        self.assertEqual(by_device[DEVICE_DUT].estimated_minutes, 88)
+        self.assertEqual(by_device[DEVICE_REF].estimated_minutes, 88)
+
+
+if __name__ == "__main__":
+    unittest.main()