operator
52 lines
1.9 KiB
Python
52 lines
1.9 KiB
Python
"""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()
|