feat: add user_version-based schema migrations

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
This commit is contained in:
2026-06-10 19:02:53 -04:00
parent a7851e7007
commit 8870b8f3ef
3 changed files with 103 additions and 4 deletions
+55 -1
View File
@@ -1,9 +1,13 @@
from pathlib import Path
import pytest
from sqlalchemy import inspect
from hqt.config import Settings
from hqt.db import migrations
from hqt.db.engine import ensure_db, get_engine, get_session_factory
from hqt.db.models import Harness, Project, Session
from hqt.db.migrations import migrate
from hqt.db.models import Base, Harness, Project, Session
def _tmp_settings(tmp_path: Path) -> Settings:
@@ -69,3 +73,53 @@ def test_session_with_fk(tmp_path):
assert s.project.name == "proj"
assert s.harness.name == "claude-code"
assert s in p.sessions
def _user_version(engine) -> int:
with engine.connect() as conn:
return conn.exec_driver_sql("PRAGMA user_version").scalar() or 0
def test_fresh_db_stamped_latest_version(tmp_path):
settings = _tmp_settings(tmp_path)
ensure_db(settings)
assert _user_version(get_engine(settings)) == migrations.LATEST_VERSION
def test_unversioned_existing_db_stamped_baseline(tmp_path):
# Simulates a user DB created before versioning existed: tables present,
# user_version still 0.
settings = _tmp_settings(tmp_path)
settings.db_path.parent.mkdir(parents=True, exist_ok=True)
engine = get_engine(settings)
Base.metadata.create_all(engine)
migrate(engine)
assert _user_version(engine) == migrations.BASELINE_VERSION
def test_db_from_newer_hqt_raises(tmp_path):
settings = _tmp_settings(tmp_path)
ensure_db(settings)
engine = get_engine(settings)
with engine.begin() as conn:
conn.exec_driver_sql("PRAGMA user_version = 9999")
with pytest.raises(RuntimeError, match="newer"):
migrate(engine)
def test_pending_migrations_applied_in_order_once(tmp_path, monkeypatch):
settings = _tmp_settings(tmp_path)
ensure_db(settings) # stamps BASELINE_VERSION
applied: list[int] = []
monkeypatch.setattr(
migrations,
"MIGRATIONS",
[(2, lambda conn: applied.append(2)), (3, lambda conn: applied.append(3))],
)
monkeypatch.setattr(migrations, "LATEST_VERSION", 3)
engine = get_engine(settings)
migrate(engine)
assert applied == [2, 3]
assert _user_version(engine) == 3
migrate(engine) # re-run is a no-op
assert applied == [2, 3]