145 lines
4.6 KiB
Python
145 lines
4.6 KiB
Python
from pathlib import Path
|
|
import json
|
|
|
|
from src.storage import HistoryDB
|
|
|
|
|
|
def test_history_db_job_and_events_roundtrip(tmp_path: Path) -> None:
|
|
db = HistoryDB(tmp_path / "screenjob_test.db")
|
|
job_id = "job_test_001"
|
|
db.create_job(
|
|
job_id=job_id,
|
|
objective="Open example.com",
|
|
model="gpt-5.4-mini",
|
|
created_at="2026-05-27T00:00:00Z",
|
|
safety_override=False,
|
|
disabled_tools=["click"],
|
|
)
|
|
db.add_event(
|
|
job_id=job_id,
|
|
ts="2026-05-27T00:00:01Z",
|
|
step=1,
|
|
event_type="tool_called",
|
|
payload={"tool": "see_screen"},
|
|
)
|
|
db.update_job(
|
|
job_id,
|
|
status="completed",
|
|
ended_at="2026-05-27T00:00:02Z",
|
|
result="Done",
|
|
response_json=json.dumps({"return": "Done", "data": {"files": ["a.txt", "b.txt"]}}, ensure_ascii=False),
|
|
steps=2,
|
|
estimated_cost_usd=0.1234,
|
|
)
|
|
|
|
job = db.get_job(job_id)
|
|
assert job is not None
|
|
assert job["status"] == "completed"
|
|
assert job["model"] == "gpt-5.4-mini"
|
|
assert job["disabled_tools"] == ["click"]
|
|
assert job["return"] == "Done"
|
|
assert job["data"]["files"] == ["a.txt", "b.txt"]
|
|
assert job["usage"]["estimated_cost_usd"] == 0.1234
|
|
|
|
events = db.get_job_events(job_id, limit=10)
|
|
assert len(events) == 1
|
|
assert events[0]["event_type"] == "tool_called"
|
|
assert events[0]["payload"]["tool"] == "see_screen"
|
|
|
|
jobs = db.list_jobs(limit=10)
|
|
assert len(jobs) == 1
|
|
assert jobs[0]["job_id"] == job_id
|
|
|
|
stats = db.stats()
|
|
assert stats["total_jobs"] == 1
|
|
assert stats["completed_jobs"] == 1
|
|
assert abs(stats["total_estimated_cost"] - 0.1234) < 1e-9
|
|
|
|
|
|
def test_storage_response_fallback_uses_result_when_json_missing(tmp_path: Path) -> None:
|
|
db = HistoryDB(tmp_path / "screenjob_test_fallback.db")
|
|
job_id = "job_test_002"
|
|
db.create_job(
|
|
job_id=job_id,
|
|
objective="Fallback check",
|
|
model="gpt-5.4-mini",
|
|
created_at="2026-05-27T00:00:00Z",
|
|
safety_override=False,
|
|
disabled_tools=[],
|
|
)
|
|
db.update_job(job_id, status="completed", result="Legacy result string")
|
|
job = db.get_job(job_id)
|
|
assert job is not None
|
|
assert job["return"] == "Legacy result string"
|
|
assert job["data"] is None
|
|
|
|
|
|
def test_history_db_analytics_groups_by_category(tmp_path: Path) -> None:
|
|
db = HistoryDB(tmp_path / "screenjob_test_analytics.db")
|
|
|
|
db.create_job(
|
|
job_id="job_browser_ok",
|
|
objective="Open amazon.de and checkout",
|
|
model="gpt-5.4-mini",
|
|
created_at="2026-05-27T00:00:01Z",
|
|
safety_override=False,
|
|
disabled_tools=[],
|
|
)
|
|
db.update_job("job_browser_ok", status="completed", steps=4, estimated_cost_usd=0.12)
|
|
|
|
db.create_job(
|
|
job_id="job_browser_fail",
|
|
objective="Open website and login",
|
|
model="gpt-5.4-mini",
|
|
created_at="2026-05-28T00:00:01Z",
|
|
safety_override=False,
|
|
disabled_tools=[],
|
|
)
|
|
db.update_job("job_browser_fail", status="failed", steps=6, estimated_cost_usd=0.24)
|
|
|
|
db.create_job(
|
|
job_id="job_terminal_ok",
|
|
objective="Run a shell command to inspect files",
|
|
model="gpt-5.4-mini",
|
|
created_at="2026-05-28T00:00:02Z",
|
|
safety_override=False,
|
|
disabled_tools=[],
|
|
)
|
|
db.update_job("job_terminal_ok", status="completed", steps=10, estimated_cost_usd=0.05)
|
|
|
|
analytics = db.analytics()
|
|
browser = next(row for row in analytics["by_category"] if row["label"] == "Browser / web")
|
|
terminal = next(row for row in analytics["by_category"] if row["label"] == "Files / terminal")
|
|
assert browser["finished_jobs"] == 2
|
|
assert browser["success_rate"] == 50.0
|
|
assert browser["avg_steps"] == 5.0
|
|
assert terminal["success_rate"] == 100.0
|
|
|
|
|
|
def test_prune_older_than_removes_old_terminal_jobs(tmp_path: Path) -> None:
|
|
db = HistoryDB(tmp_path / "screenjob_prune.db")
|
|
db.create_job(
|
|
job_id="job_old",
|
|
objective="Old",
|
|
model="gpt-5.4-mini",
|
|
created_at="2000-01-01T00:00:00+00:00",
|
|
safety_override=False,
|
|
disabled_tools=[],
|
|
)
|
|
db.update_job("job_old", status="completed")
|
|
db.add_event(job_id="job_old", ts="2000-01-01T00:00:01+00:00", step=1, event_type="done", payload={})
|
|
|
|
db.create_job(
|
|
job_id="job_running",
|
|
objective="Running",
|
|
model="gpt-5.4-mini",
|
|
created_at="2000-01-01T00:00:00+00:00",
|
|
safety_override=False,
|
|
disabled_tools=[],
|
|
)
|
|
db.update_job("job_running", status="running")
|
|
|
|
assert db.prune_older_than(7) == 1
|
|
assert db.get_job("job_old") is None
|
|
assert db.get_job("job_running") is not None
|