import os, sys, tempfile
sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "..", "src"))
from pgar.state import StateStore

def test_continuation_rows_do_not_collide():
    s = StateStore(os.path.join(tempfile.mkdtemp(), "s.sqlite"))
    s.log_session("sid-1", "R1", "executor", "initial", 0, {"argv": []}, "raw")
    s.log_session("sid-1#40", "R1", "executor", "review_repair", 40, {"argv": [], "client_session_id": "sid-1"}, "raw")
    s.finish_session("sid-1#40", session_cost_usd=0.1)
    rows = s.query("SELECT session_id, kind, session_cost_usd FROM sessions WHERE run_id='R1' ORDER BY seq")
    assert [r["kind"] for r in rows] == ["initial", "review_repair"] and rows[1]["session_cost_usd"] == 0.1