AdrianoDev
91 lines
2.8 KiB
Python
91 lines
2.8 KiB
Python
from fastapi.testclient import TestClient
|
|
from option_mcp_common.auth import Principal, TokenStore
|
|
from option_mcp_common.server import build_app
|
|
|
|
|
|
def test_build_app_health():
|
|
store = TokenStore(tokens={})
|
|
app = build_app(name="test-mcp", version="0.0.1", token_store=store)
|
|
client = TestClient(app)
|
|
r = client.get("/health")
|
|
assert r.status_code == 200
|
|
body = r.json()
|
|
assert body["status"] == "healthy"
|
|
assert body["name"] == "test-mcp"
|
|
assert body["version"] == "0.0.1"
|
|
assert "uptime_seconds" in body
|
|
assert "data_timestamp" in body
|
|
assert r.headers.get("X-Duration-Ms") is not None
|
|
|
|
|
|
def test_build_app_adds_token_store():
|
|
store = TokenStore(tokens={"t1": Principal("x", {"core"})})
|
|
app = build_app(name="t", version="v", token_store=store)
|
|
assert app.state.token_store is store
|
|
|
|
|
|
def test_timestamp_injector_dict_response():
|
|
"""CER-P5-001: dict response gets data_timestamp + X-Data-Timestamp header."""
|
|
store = TokenStore(tokens={})
|
|
app = build_app(name="t", version="v", token_store=store)
|
|
|
|
@app.post("/tools/foo")
|
|
def foo():
|
|
return {"ok": True}
|
|
|
|
client = TestClient(app)
|
|
r = client.post("/tools/foo")
|
|
assert r.status_code == 200
|
|
body = r.json()
|
|
assert body["ok"] is True
|
|
assert "data_timestamp" in body
|
|
assert r.headers.get("X-Data-Timestamp") is not None
|
|
|
|
|
|
def test_timestamp_injector_list_of_dicts():
|
|
"""CER-P5-001: list of dicts → each item gets data_timestamp."""
|
|
store = TokenStore(tokens={})
|
|
app = build_app(name="t", version="v", token_store=store)
|
|
|
|
@app.post("/tools/list_items")
|
|
def list_items():
|
|
return [{"x": 1}, {"x": 2}]
|
|
|
|
client = TestClient(app)
|
|
r = client.post("/tools/list_items")
|
|
body = r.json()
|
|
assert isinstance(body, list)
|
|
assert len(body) == 2
|
|
for item in body:
|
|
assert "data_timestamp" in item
|
|
assert r.headers.get("X-Data-Timestamp") is not None
|
|
|
|
|
|
def test_timestamp_injector_preserves_existing():
|
|
"""CER-P5-001: se già presente, non override."""
|
|
store = TokenStore(tokens={})
|
|
app = build_app(name="t", version="v", token_store=store)
|
|
|
|
@app.post("/tools/already")
|
|
def already():
|
|
return {"data_timestamp": "2020-01-01T00:00:00Z", "x": 1}
|
|
|
|
client = TestClient(app)
|
|
body = client.post("/tools/already").json()
|
|
assert body["data_timestamp"] == "2020-01-01T00:00:00Z"
|
|
|
|
|
|
def test_timestamp_injector_empty_list_gets_header_only():
|
|
"""CER-P5-001: list vuota — no body modification, ma header presente."""
|
|
store = TokenStore(tokens={})
|
|
app = build_app(name="t", version="v", token_store=store)
|
|
|
|
@app.post("/tools/empty_list")
|
|
def empty_list():
|
|
return []
|
|
|
|
client = TestClient(app)
|
|
r = client.post("/tools/empty_list")
|
|
assert r.json() == []
|
|
assert r.headers.get("X-Data-Timestamp") is not None
|