forked from luopingyi/xc_agent_platform_demo
61 lines
1.9 KiB
Python
61 lines
1.9 KiB
Python
|
"""Tests for /health and /status handlers."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import json
|
||
|
|
import os
|
||
|
|
import unittest
|
||
|
|
from unittest import mock
|
||
|
|
|
||
|
|
from src.health import handle_get, health_payload, status_payload
|
||
|
|
|
||
|
|
|
||
|
|
class TestHealth(unittest.TestCase):
|
||
|
|
def test_health_payload(self) -> None:
|
||
|
|
self.assertEqual(health_payload(), {"status": "ok"})
|
||
|
|
|
||
|
|
def test_handle_get_health(self) -> None:
|
||
|
|
code, headers, body = handle_get("/health")
|
||
|
|
self.assertEqual(code, 200)
|
||
|
|
self.assertIn("application/json", headers["Content-Type"])
|
||
|
|
data = json.loads(body.decode("utf-8"))
|
||
|
|
self.assertEqual(data, {"status": "ok"})
|
||
|
|
|
||
|
|
def test_handle_get_unknown(self) -> None:
|
||
|
|
code, _, body = handle_get("/nope")
|
||
|
|
self.assertEqual(code, 404)
|
||
|
|
self.assertIn("not_found", body.decode("utf-8"))
|
||
|
|
|
||
|
|
def test_status_booleans_only_no_secret_values(self) -> None:
|
||
|
|
with mock.patch.dict(
|
||
|
|
os.environ,
|
||
|
|
{
|
||
|
|
"STRATEGY_ID": "secret-strategy-id-xyz",
|
||
|
|
"EXTERNAL_SERVICE_TOKEN": "secret-token-abc",
|
||
|
|
},
|
||
|
|
clear=False,
|
||
|
|
):
|
||
|
|
payload = status_payload()
|
||
|
|
self.assertTrue(payload["strategy_id_present"])
|
||
|
|
self.assertTrue(payload["external_service_token_present"])
|
||
|
|
blob = json.dumps(payload)
|
||
|
|
self.assertNotIn("secret-strategy-id-xyz", blob)
|
||
|
|
self.assertNotIn("secret-token-abc", blob)
|
||
|
|
|
||
|
|
def test_status_absent(self) -> None:
|
||
|
|
env = {k: v for k, v in os.environ.items() if k not in (
|
||
|
|
"STRATEGY_ID",
|
||
|
|
"EXTERNAL_SERVICE_TOKEN",
|
||
|
|
"XC_TOKEN",
|
||
|
|
"XC_TOKEN_FILE",
|
||
|
|
"MODELHUB_XC_TOKEN",
|
||
|
|
)}
|
||
|
|
with mock.patch.dict(os.environ, env, clear=True):
|
||
|
|
payload = status_payload()
|
||
|
|
self.assertFalse(payload["strategy_id_present"])
|
||
|
|
self.assertFalse(payload["external_service_token_present"])
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|