fixed compatibility issue

backend/db.py

@@ -6,6 +6,7 @@ from contextlib import contextmanager
 from dataclasses import dataclass
 from pathlib import Path
 from typing import Any, Iterator
+from graph import build_station_testpoint_map
 
 
 APP_ROOT = Path(__file__).resolve().parent
@@ -36,6 +37,7 @@ class TestRecord:
 	status: str = "pending"
 	excluded: bool = False
 	raw_payload: dict[str, Any] | None = None
+	station_testpoint_map: str | None = None
 
 
 @dataclass(frozen=True)
@@ -141,6 +143,7 @@ def init_db(db_path: str | Path = DB_PATH) -> None:
 		_ensure_column(conn, "tests", "victim_band", "TEXT")
 		_ensure_column(conn, "tests", "excluded", "INTEGER NOT NULL DEFAULT 0")
 		_ensure_column(conn, "tests", "throttled", "INTEGER NOT NULL DEFAULT 0")
+		_ensure_column(conn, "tests", "station_testpoint_map", "TEXT")
 
 		# Seed runtime defaults for schedule estimation.
 		conn.execute(
@@ -154,6 +157,12 @@ def init_db(db_path: str | Path = DB_PATH) -> None:
 		)
 
 
+def _serialize_station_testpoint_map(config: dict[str, dict[str, str | None]]) -> str:
+	"""Compute and serialize the station-to-testpoint map from a test config."""
+	station_map = build_station_testpoint_map(config)
+	return json.dumps(station_map)
+
+
 def upsert_tests(records: list[TestRecord], db_path: str | Path = DB_PATH) -> int:
 	if not records:
 		return 0
@@ -163,9 +172,9 @@ def upsert_tests(records: list[TestRecord], db_path: str | Path = DB_PATH) -> in
 			"""
 			INSERT INTO tests(
 				test_id, device, test_type, rotation, rx_tx, has_coe_pair,
-				coe_pairing_json, config_json, priority, victim_band, throttled, estimated_minutes, status, excluded, raw_payload
+				coe_pairing_json, config_json, priority, victim_band, throttled, estimated_minutes, status, excluded, raw_payload, station_testpoint_map
 			)
-			VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+			VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
 			ON CONFLICT(test_id, device) DO UPDATE SET
 				test_type = excluded.test_type,
 				device = excluded.device,
@@ -181,6 +190,7 @@ def upsert_tests(records: list[TestRecord], db_path: str | Path = DB_PATH) -> in
 				status = excluded.status,
 				excluded = excluded.excluded,
 				raw_payload = excluded.raw_payload,
+				station_testpoint_map = excluded.station_testpoint_map,
 				updated_at = CURRENT_TIMESTAMP
 			""",
 			[
@@ -200,6 +210,7 @@ def upsert_tests(records: list[TestRecord], db_path: str | Path = DB_PATH) -> in
 					r.status,
 					int(r.excluded),
 					json.dumps(r.raw_payload or {}),
+					_serialize_station_testpoint_map(r.config),
 				)
 				for r in records
 			],
@@ -403,7 +414,7 @@ def list_tests_for_device(device: str, db_path: str | Path = DB_PATH) -> list[Te
 				test_id, device, test_type, rotation, rx_tx, has_coe_pair,
 				config_json, victim_band,
 				coe_pairing_json, priority, throttled, estimated_minutes,
-				excluded, status, raw_payload
+				excluded, status, raw_payload, station_testpoint_map
 			FROM tests
 			WHERE device = ?
 			""",
@@ -429,6 +440,7 @@ def list_tests_for_device(device: str, db_path: str | Path = DB_PATH) -> list[Te
 				status=row["status"],
 				excluded=bool(row["excluded"]),
 				raw_payload=json.loads(row["raw_payload"] or "{}"),
+				station_testpoint_map=row["station_testpoint_map"],
 			)
 		)