"""Startup cleanup of orphan runs (interrupted by server restart)."""

from agenteval.models import EvalRun, RunStatus
from agenteval.storage.repository import RunRepository


def _make_run(session, status: RunStatus) -> str:
    run = RunRepository(session).create(EvalRun(
        target_id="t-1", scenario_id="s-1", status=status,
    ))
    return run.id


def test_mark_orphans_failed(db_session):
    repo = RunRepository(db_session)
    running_id = _make_run(db_session, RunStatus.RUNNING)
    pending_id = _make_run(db_session, RunStatus.PENDING)
    completed_id = _make_run(db_session, RunStatus.COMPLETED)
    failed_id = _make_run(db_session, RunStatus.FAILED)

    count = repo.mark_orphans_failed()

    assert count == 2
    for rid in (running_id, pending_id):
        run = repo.get(rid)
        assert run.status == RunStatus.FAILED
        assert run.summary["error"]["code"] == "interrupted"
        assert run.completed_at is not None
    # 已完结的运行不受影响
    assert repo.get(completed_id).status == RunStatus.COMPLETED
    assert repo.get(failed_id).status == RunStatus.FAILED
    assert repo.get(completed_id).summary is None


def test_mark_orphans_failed_noop_when_clean(db_session):
    repo = RunRepository(db_session)
    _make_run(db_session, RunStatus.COMPLETED)
    assert repo.mark_orphans_failed() == 0