ai-incidents 1.0.0

tests/test_sources.py

@@ -0,0 +1,94 @@
+import os
+import sqlite3
+
+import pytest
+
+from ai_incidents.sources import (
+    SourceError, claude_key, iter_claude_code, iter_opencode, open_readonly, parse_claude_jsonl,
+)
+from helpers import cc, oc_db, oc_session, write_session
+
+
+def test_claude_jsonl_parse(tmp_path):
+    p = write_session(str(tmp_path), "-home-me-proj", "0123456789abcdef", [
+        cc("user", "please clean up"),
+        "not json at all",
+        "",
+        '{"type": "summary"}',
+        cc("assistant", "running it", tool_use="rm -rf ./build"),
+        cc("user", tool_result="removed 3 files"),
+        cc("user", tool_result="boom", is_error=True),
+    ])
+    s = parse_claude_jsonl(p)
+    assert [(t.role, t.text) for t in s.turns] == [("user", "please clean up"), ("assistant", "running it")]
+    assert s.turns[0].date == "2026-07-14"
+    assert [(c.name, c.command) for c in s.calls] == [("Bash", "rm -rf ./build")]
+    assert [(r.is_error, r.text) for r in s.results] == [(False, "removed 3 files"), (True, "boom")]
+
+
+def test_claude_key_shape_and_change_on_growth(tmp_path):
+    p = write_session(str(tmp_path), "proj", "sess", [cc("user", "hi")])
+    k1 = claude_key(p)
+    name, mtime, size = k1.split(":")
+    assert name == "sess.jsonl" and int(size) == os.path.getsize(p) and mtime.isdigit()
+    with open(p, "a") as f:
+        f.write(cc("assistant", "more") + "\n")
+    assert claude_key(p) != k1
+
+
+def test_claude_iter_includes_subagents_and_missing_root(tmp_path):
+    write_session(str(tmp_path), "proj", "aaaaaaaa1111", [cc("user", "x")])
+    write_session(str(tmp_path), "proj/aaaaaaaa1111/subagents", "agent-1", [cc("user", "y")])
+    refs = list(iter_claude_code("claude-code", str(tmp_path)))
+    assert sorted(r.short_id for r in refs) == ["aaaaaaaa", "agent-1"]
+    assert list(iter_claude_code("claude-code", str(tmp_path / "nope"))) == []
+
+
+def test_opencode_parse(tmp_path):
+    db = str(tmp_path / "opencode.db")
+    con = oc_db(db)
+    oc_session(con, "ses_f44d56513ffe8czutv5E", [
+        ("user", [{"type": "text", "text": "tidy the repo"},
+                  {"type": "text", "text": "<file contents>", "synthetic": True}]),
+        ("assistant", [
+            {"type": "reasoning", "text": "thinking"},
+            {"type": "text", "text": "I deleted the wrong folder"},
+            {"type": "tool", "tool": "bash", "state": {"status": "completed", "input": {"command": "rm -rf data"},
+                                                      "output": "ok"}},
+            {"type": "tool", "tool": "read", "state": {"status": "error", "input": {"filePath": "/x"},
+                                                      "error": "Traceback (most recent call last)"}},
+        ]),
+    ])
+    con.close()
+    [ref] = list(iter_opencode("opencode", str(tmp_path / "opencode*.db")))
+    assert ref.short_id == "f44d5651"
+    s = ref.load()
+    assert [(t.role, t.text) for t in s.turns] == [("user", "tidy the repo"), ("assistant", "I deleted the wrong folder")]
+    assert s.turns[0].date == "2026-07-14"
+    assert [(c.name, c.command) for c in s.calls] == [("bash", "rm -rf data"), ("read", "")]
+    assert [(r.is_error, r.text) for r in s.results] == [(False, "ok"), (True, "Traceback (most recent call last)")]
+
+
+def test_opencode_key_changes_when_session_updates(tmp_path):
+    db = str(tmp_path / "opencode.db")
+    con = oc_db(db)
+    oc_session(con, "ses_1", [("user", [{"type": "text", "text": "hi"}])])
+    k1 = next(iter_opencode("opencode", db)).key
+    con.execute("UPDATE session SET time_updated = time_updated + 5")
+    con.commit()
+    assert next(iter_opencode("opencode", db)).key != k1
+
+
+def test_opencode_is_opened_read_only(tmp_path):
+    db = str(tmp_path / "opencode.db")
+    oc_db(db).close()
+    con = open_readonly(db)
+    with pytest.raises(sqlite3.OperationalError):
+        con.execute("INSERT INTO session VALUES ('x','p',NULL,'s','/','t','v',1,1)")
+
+
+def test_opencode_rejects_a_foreign_database(tmp_path):
+    db = str(tmp_path / "opencode.db")
+    sqlite3.connect(db).execute("CREATE TABLE unrelated (x)").connection.close()
+    with pytest.raises(SourceError):
+        list(iter_opencode("opencode", db))