from __future__ import annotations

from promptboost.db import list_boost_runs, save_boost_run
from promptboost.pipeline import boost


def test_boost_run_persistence(temp_db) -> None:
    result = boost("Fix a code bug and add tests.", "codex", "gpt")
    saved = save_boost_run(result, temp_db)
    runs = list_boost_runs(db_path=temp_db)

    assert saved.id == 1
    assert len(runs) == 1
    assert runs[0].raw_prompt == result.raw_prompt
    assert runs[0].adapter_rules["harness"]["id"] == "codex"