feat(gateway): expose /api/v1/queue on chromebox-gateway and configure Cloudflare ingress

2026-10-10 15:38:14 +00:00
parent 5d37659255
commit 7ea70c5c5d
2 changed files with 87 additions and 2 deletions
@@ -0,0 +1,51 @@
+"""Test for chromebox-gateway /api/v1/queue endpoint."""
+import os
+import sys
+import json
+import unittest
+from unittest.mock import patch, MagicMock
+
+REPO_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+BIN_DIR = os.path.join(REPO_ROOT, "bin")
+if BIN_DIR not in sys.path:
+    sys.path.insert(0, BIN_DIR)
+
+import importlib.util
+spec = importlib.util.spec_from_file_location("chromebox_gateway", os.path.join(BIN_DIR, "chromebox-gateway.py"))
+cbg = importlib.util.module_from_spec(spec)
+spec.loader.exec_module(cbg)
+
+class TestChromeboxGatewayQueue(unittest.TestCase):
+    def test_health_endpoints_list(self):
+        handler = MagicMock()
+        handler.path = "/health"
+        cbg.Handler.do_GET(handler)
+        handler._json.assert_called_once()
+        args, _ = handler._json.call_args
+        self.assertEqual(args[0], 200)
+        self.assertIn("/api/v1/queue", args[1].get("endpoints", []))
+
+    @patch.object(cbg, "check_token", return_value=None)
+    def test_queue_unauthorized(self, mock_tok):
+        handler = MagicMock()
+        handler.path = "/api/v1/queue"
+        handler.headers = {"Authorization": "Bearer invalid"}
+        cbg.Handler.do_GET(handler)
+        handler._json.assert_called_once_with(401, {"error": "unauthorized"})
+
+    @patch.object(cbg, "check_token", return_value="master")
+    @patch.object(cbg, "rate_ok", return_value=True)
+    def test_queue_authorized_success(self, mock_rate, mock_tok):
+        handler = MagicMock()
+        handler.path = "/api/v1/queue"
+        handler.headers = {"Authorization": "Bearer master-token"}
+        cbg.Handler.do_GET(handler)
+        handler._json.assert_called_once()
+        code, body = handler._json.call_args[0]
+        self.assertEqual(code, 200)
+        self.assertTrue(body.get("ok"))
+        self.assertIn("counts", body)
+        self.assertIn("tasks", body)
+
+if __name__ == "__main__":
+    unittest.main()