Minor updates
This commit is contained in:
@@ -85,15 +85,27 @@ class GatewayLLMTests(unittest.IsolatedAsyncioTestCase):
|
||||
async def test_llm_endpoint_requires_api_key(self):
|
||||
module = load_gateway_module(llm_enabled=True)
|
||||
|
||||
response = await self._request(
|
||||
module,
|
||||
"POST",
|
||||
"/ai/chat",
|
||||
json_payload={"messages": [{"role": "user", "content": "hello"}]},
|
||||
)
|
||||
with self.assertLogs("gateway", level="INFO") as captured:
|
||||
response = await self._request(
|
||||
module,
|
||||
"POST",
|
||||
"/ai/chat",
|
||||
json_payload={"messages": [{"role": "user", "content": "hello"}]},
|
||||
)
|
||||
|
||||
self.assertEqual(response.status_code, 401)
|
||||
self.assertEqual(response.json()["error"]["code"], "unauthorized")
|
||||
joined_logs = "\n".join(captured.output)
|
||||
self.assertIn("incoming_request service=gateway", joined_logs)
|
||||
self.assertIn("completed_request service=gateway", joined_logs)
|
||||
self.assertIn("caller=127.0.0.1", joined_logs)
|
||||
self.assertIn("method=POST", joined_logs)
|
||||
self.assertIn("path=/ai/chat", joined_logs)
|
||||
self.assertIn("status_code=401", joined_logs)
|
||||
self.assertIn("response_time_ms=", joined_logs)
|
||||
self.assertIn("cpu_time_ms=", joined_logs)
|
||||
self.assertIn("cpu_percent=", joined_logs)
|
||||
self.assertIn("rss_mb=", joined_logs)
|
||||
|
||||
async def test_llm_disabled_returns_503(self):
|
||||
module = load_gateway_module(llm_enabled=False)
|
||||
|
||||
Reference in New Issue
Block a user