76 lines
3.5 KiB
Python
76 lines
3.5 KiB
Python
 
|
"""B-97 (DAT-75, DAT-56, ARC-62, ARC-64): the Alembic chain is executed, not assumed.
|
||
|
|
|
||
|
|
The two expression indexes that carry invariants — `ix_rounds_single_active` ("at
|
||
|
|
most one active round", B-09) and `ix_users_username_lower` ("usernames are
|
||
|
|
case-insensitive", B-57) — are invisible to autogenerate, and `batch_alter_table`
|
||
|
|
on SQLite rebuilds the table and drops them with only a warning. Every other test
|
||
|
|
builds its schema with `Base.metadata.create_all`, so a migration that lost one of
|
||
|
|
them would reach production unnoticed. This one runs `alembic upgrade head` on an
|
||
|
|
empty database, exactly as the container does at startup.
|
||
|
|
|
||
|
|
Alembic runs in a subprocess: env.py calls `asyncio.run` and `fileConfig`, neither of
|
||
|
|
which belongs inside the pytest process.
|
||
|
|
"""
|
||
|
|
|
||
|
|
import os
|
||
|
|
import re
|
||
|
|
import sqlite3
|
||
|
|
import subprocess
|
||
|
|
import sys
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
_ROOT = Path(__file__).resolve().parents[2]
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture(scope="module")
|
||
|
|
def migrated_db(tmp_path_factory) -> Path:
|
||
|
|
db_path = tmp_path_factory.mktemp("migrations") / "migrated.db"
|
||
|
|
env = {**os.environ, "DATABASE_URL": f"sqlite+aiosqlite:///{db_path}"}
|
||
|
|
result = subprocess.run(
|
||
|
|
[sys.executable, "-m", "alembic", "upgrade", "head"],
|
||
|
|
cwd=_ROOT,
|
||
|
|
env=env,
|
||
|
|
capture_output=True,
|
||
|
|
text=True,
|
||
|
|
timeout=120,
|
||
|
|
)
|
||
|
|
assert result.returncode == 0, f"alembic upgrade head failed:\n{result.stderr[-3000:]}"
|
||
|
|
return db_path
|
||
|
|
|
||
|
|
|
||
|
|
def _index_sql(db_path: Path, name: str) -> str | None:
|
||
|
|
with sqlite3.connect(db_path) as conn:
|
||
|
|
row = conn.execute("SELECT sql FROM sqlite_master WHERE type = 'index' AND name = ?", (name,)).fetchone()
|
||
|
|
return re.sub(r"\s+", " ", row[0]).lower() if row else None
|
||
|
|
|
||
|
|
|
||
|
|
def test_the_expression_indexes_survive_the_whole_migration_chain(migrated_db):
|
||
|
|
"""DAT-75 / ARC-62 / B-97: both indexes exist, with their definition, at head."""
|
||
|
|
single_active = _index_sql(migrated_db, "ix_rounds_single_active")
|
||
|
|
assert single_active is not None, "ix_rounds_single_active was lost by a migration"
|
||
|
|
assert "unique index" in single_active
|
||
|
|
assert "on rounds ((1))" in single_active
|
||
|
|
for status in ("open", "closing", "drawing", "paying_out"):
|
||
|
|
assert f"'{status}'" in single_active
|
||
|
|
|
||
|
|
username_lower = _index_sql(migrated_db, "ix_users_username_lower")
|
||
|
|
assert username_lower is not None, "ix_users_username_lower was lost by a migration"
|
||
|
|
assert "unique index" in username_lower
|
||
|
|
assert "lower(username)" in username_lower
|
||
|
|
|
||
|
|
|
||
|
|
def test_the_migrated_database_still_enforces_both_invariants(migrated_db):
|
||
|
|
"""DAT-75 / B-97: a second active round and a case-variant username are refused."""
|
||
|
|
with sqlite3.connect(migrated_db) as conn:
|
||
|
|
conn.execute("INSERT INTO rounds (status, opened_at) VALUES ('open', '2026-01-01 00:00:00')")
|
||
|
|
with pytest.raises(sqlite3.IntegrityError):
|
||
|
|
conn.execute("INSERT INTO rounds (status, opened_at) VALUES ('closing', '2026-01-01 00:00:00')")
|
||
|
|
# A closed round is not active, so any number of them may coexist.
|
||
|
|
conn.execute("INSERT INTO rounds (status, opened_at) VALUES ('closed', '2026-01-01 00:00:00')")
|
||
|
|
|
||
|
|
columns = "(username, password_hash, derivation_index, address, cached_balance_sats, token_version, created_at)"
|
||
|
|
conn.execute(f"INSERT INTO users {columns} VALUES ('Alice', 'x', 0, 'plm1qa', 0, 0, '2026-01-01 00:00:00')")
|
||
|
|
with pytest.raises(sqlite3.IntegrityError):
|
||
|
|
conn.execute(f"INSERT INTO users {columns} VALUES ('aLICE', 'x', 1, 'plm1qb', 0, 0, '2026-01-01 00:00:00')")
|