Initial release: iai-mcp v0.1.0

Co-Authored-By: Claude <noreply@anthropic.com>
Co-Authored-By: XNLLLLH <XNLLLLH@users.noreply.github.com>

tests/test_cli_trajectory.py

@@ -0,0 +1,77 @@
+"""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