Fix dashboard separation mode switch

dev/test_dashboard_modes.py

@@ -0,0 +1,65 @@
+import importlib.util
+import json
+import os
+import tempfile
+import unittest
+from pathlib import Path
+from unittest.mock import patch
+
+
+ROOT = Path(__file__).resolve().parents[1]
+MODULE_PATH = ROOT / "platform/llama-dashboard/app.py"
+
+
+class _Response:
+    status = 202
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, *_args):
+        return False
+
+    def read(self):
+        return b'{"status":"accepted"}'
+
+
+class DashboardModeTests(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        cls.tempdir = tempfile.TemporaryDirectory()
+        with patch.dict(os.environ, {
+            "DASHBOARD_HISTORY_DB": str(Path(cls.tempdir.name) / "history.sqlite3"),
+            "ROUTER_URL": "http://router.test:8081",
+            "ROUTER_API_KEY": "test-key",
+        }):
+            spec = importlib.util.spec_from_file_location("dashboard_app_test", MODULE_PATH)
+            cls.dashboard = importlib.util.module_from_spec(spec)
+            assert spec.loader is not None
+            spec.loader.exec_module(cls.dashboard)
+
+    @classmethod
+    def tearDownClass(cls):
+        cls.tempdir.cleanup()
+
+    def test_separation_mode_is_forwarded_to_router(self):
+        with patch.object(self.dashboard.urllib.request, "urlopen", return_value=_Response()) as urlopen:
+            status, body = self.dashboard.change_mode("separation")
+
+        self.assertEqual(status, 202)
+        self.assertEqual(body, {"status": "accepted"})
+        request = urlopen.call_args.args[0]
+        self.assertEqual(json.loads(request.data), {"mode": "separation"})
+        self.assertEqual(request.get_header("Authorization"), "Bearer test-key")
+
+    def test_unknown_mode_is_rejected_without_router_request(self):
+        with patch.object(self.dashboard.urllib.request, "urlopen") as urlopen:
+            status, body = self.dashboard.change_mode("unknown")
+
+        self.assertEqual(status, 400)
+        self.assertEqual(body, {"error": "invalid mode"})
+        urlopen.assert_not_called()
+
+
+if __name__ == "__main__":
+    unittest.main()