feat: add governed nationwide AOI orchestration and CUDA enforcement

backend/tests/test_runtime_reconciliation_service.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 from datetime import datetime, timezone
 from unittest.mock import MagicMock
 
-from app.models import AnalysisRun, Job
+from app.models import AoiOperation, AoiOperationPartition, AnalysisRun, Job
 from app.services.runtime_reconciliation_service import RuntimeReconciliationService
 
 
@@ -11,9 +11,15 @@ def test_reconciliation_terminalizes_only_running_work() -> None:
     db = MagicMock()
     jobs = MagicMock()
     runs = MagicMock()
+    resumable_partitions = MagicMock()
+    exhausted_partitions = MagicMock()
+    operations = MagicMock()
     jobs.filter.return_value.update.return_value = 5
     runs.filter.return_value.update.return_value = 2
-    db.query.side_effect = [jobs, runs]
+    resumable_partitions.filter.return_value.update.return_value = 3
+    exhausted_partitions.filter.return_value.update.return_value = 1
+    operations.filter.return_value.update.return_value = 2
+    db.query.side_effect = [jobs, runs, resumable_partitions, exhausted_partitions, operations]
     finished_at = datetime(2026, 7, 17, 20, 0, tzinfo=timezone.utc)
 
     result = RuntimeReconciliationService.reconcile(
@@ -23,8 +29,13 @@ def test_reconciliation_terminalizes_only_running_work() -> None:
 
     assert result.interrupted_jobs == 5
     assert result.interrupted_analysis_runs == 2
+    assert result.resumed_aoi_partitions == 3
+    assert result.exhausted_aoi_partitions == 1
     jobs.filter.assert_called_once()
     runs.filter.assert_called_once()
+    resumable_partitions.filter.assert_called_once()
+    exhausted_partitions.filter.assert_called_once()
+    operations.filter.assert_called_once()
     job_values = jobs.filter.return_value.update.call_args.args[0]
     run_values = runs.filter.return_value.update.call_args.args[0]
     assert job_values[Job.status] == "failed"
@@ -33,6 +44,12 @@ def test_reconciliation_terminalizes_only_running_work() -> None:
     assert run_values[AnalysisRun.status] == "failed"
     assert run_values[AnalysisRun.finished_at] == finished_at
     assert "PROCESS_INTERRUPTED" in run_values[AnalysisRun.error_message]
+    resumed_values = resumable_partitions.filter.return_value.update.call_args.args[0]
+    exhausted_values = exhausted_partitions.filter.return_value.update.call_args.args[0]
+    operation_values = operations.filter.return_value.update.call_args.args[0]
+    assert resumed_values[AoiOperationPartition.status] == "queued"
+    assert exhausted_values[AoiOperationPartition.status] == "failed"
+    assert operation_values[AoiOperation.status] == "queued"
     db.commit.assert_called_once_with()