78 lines
2.6 KiB
Python
78 lines
2.6 KiB
Python
|
"""Tests for iai-mcp trajectory CLI.
|
||
|
|
|
||
|
|
The `trajectory` subcommand aggregates M1..M6 events via
|
||
|
|
trajectory.aggregate_trajectory and prints one summary line per metric.
|
||
|
|
Supports --since WEEKS to scope history.
|
||
|
|
"""
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import json
|
||
|
|
from datetime import datetime, timedelta, timezone
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from iai_mcp.cli import main as cli_main
|
||
|
|
from iai_mcp.events import write_event
|
||
|
|
from iai_mcp.store import MemoryStore
|
||
|
|
|
||
|
|
|
||
|
|
def test_trajectory_empty_output(tmp_path, capsys, monkeypatch):
|
||
|
|
monkeypatch.setenv("IAI_MCP_STORE", str(tmp_path))
|
||
|
|
# No trajectory data recorded yet.
|
||
|
|
code = cli_main(["trajectory"])
|
||
|
|
assert code == 0
|
||
|
|
out = capsys.readouterr().out
|
||
|
|
assert "no trajectory data" in out.lower() or "no data" in out.lower()
|
||
|
|
|
||
|
|
|
||
|
|
def test_trajectory_renders_m1_to_m6(tmp_path, capsys, monkeypatch):
|
||
|
|
monkeypatch.setenv("IAI_MCP_STORE", str(tmp_path))
|
||
|
|
store = MemoryStore(path=tmp_path)
|
||
|
|
# Seed one event for each metric.
|
||
|
|
for i, m in enumerate(["m1", "m2", "m3", "m4", "m5", "m6"]):
|
||
|
|
write_event(
|
||
|
|
store,
|
||
|
|
kind="trajectory_metric",
|
||
|
|
data={"metric": m, "value": float(i + 1)},
|
||
|
|
severity="info",
|
||
|
|
session_id="s1",
|
||
|
|
)
|
||
|
|
code = cli_main(["trajectory"])
|
||
|
|
assert code == 0
|
||
|
|
out = capsys.readouterr().out
|
||
|
|
# Every metric mentioned (M1 ... M6 uppercase).
|
||
|
|
for m in ("M1", "M2", "M3", "M4", "M5", "M6"):
|
||
|
|
assert m in out
|
||
|
|
|
||
|
|
|
||
|
|
def test_trajectory_since_weeks_flag(tmp_path, capsys, monkeypatch):
|
||
|
|
"""--since=N accepts the flag without crashing. (Filter behaviour is
|
||
|
|
tested at the trajectory.aggregate_trajectory level; the CLI contract
|
||
|
|
here is: flag is recognised and 0 on success.)"""
|
||
|
|
monkeypatch.setenv("IAI_MCP_STORE", str(tmp_path))
|
||
|
|
store = MemoryStore(path=tmp_path)
|
||
|
|
write_event(
|
||
|
|
store, kind="trajectory_metric",
|
||
|
|
data={"metric": "m1", "value": 1.0},
|
||
|
|
severity="info", session_id="s1",
|
||
|
|
)
|
||
|
|
code = cli_main(["trajectory", "--since=2"])
|
||
|
|
assert code == 0
|
||
|
|
|
||
|
|
|
||
|
|
def test_trajectory_prints_aggregate_stats(tmp_path, capsys, monkeypatch):
|
||
|
|
"""Output for a populated M1 mentions min/max/mean."""
|
||
|
|
monkeypatch.setenv("IAI_MCP_STORE", str(tmp_path))
|
||
|
|
store = MemoryStore(path=tmp_path)
|
||
|
|
for v in (1.0, 2.0, 3.0):
|
||
|
|
write_event(
|
||
|
|
store, kind="trajectory_metric",
|
||
|
|
data={"metric": "m1", "value": v},
|
||
|
|
severity="info", session_id="s1",
|
||
|
|
)
|
||
|
|
code = cli_main(["trajectory"])
|
||
|
|
assert code == 0
|
||
|
|
out = capsys.readouterr().out
|
||
|
|
# Some aggregate indicator visible.
|
||
|
|
assert "mean" in out.lower() or "avg" in out.lower() or "=" in out
|