1import os, sys, tempfile2sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "..", "src"))3from pgar.state import StateStore45def test_continuation_rows_do_not_collide():6 s = StateStore(os.path.join(tempfile.mkdtemp(), "s.sqlite"))7 s.log_session("sid-1", "R1", "executor", "initial", 0, {"argv": []}, "raw")8 s.log_session("sid-1#40", "R1", "executor", "review_repair", 40, {"argv": [], "client_session_id": "sid-1"}, "raw")9 s.finish_session("sid-1#40", session_cost_usd=0.1)10 rows = s.query("SELECT session_id, kind, session_cost_usd FROM sessions WHERE run_id='R1' ORDER BY seq")11 assert [r["kind"] for r in rows] == ["initial", "review_repair"] and rows[1]["session_cost_usd"] == 0.112