39146288e1
Adds an optional Jinja2/HTML banner displayed above the bootstrap commands on /setup. Empty by default; admin authors it at /admin/setup-banner. autoescape=True — safe for HTML context. Render failures return "" so a broken banner never breaks /setup. Schema v22: setup_banner singleton table, auto-migration v21→v22.
54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
"""v21 → v22 migration: adds setup_banner singleton table."""
|
|
|
|
from pathlib import Path
|
|
|
|
import duckdb
|
|
|
|
from src.db import SCHEMA_VERSION, _ensure_schema, get_schema_version
|
|
|
|
|
|
def _open(path: Path) -> duckdb.DuckDBPyConnection:
|
|
return duckdb.connect(str(path))
|
|
|
|
|
|
def test_v22_creates_setup_banner_table(tmp_path):
|
|
db_path = tmp_path / "system.duckdb"
|
|
conn = _open(db_path)
|
|
# Pretend we're on v21: run schema then roll version back.
|
|
_ensure_schema(conn)
|
|
conn.execute("UPDATE schema_version SET version = 21")
|
|
conn.execute("DROP TABLE IF EXISTS setup_banner")
|
|
conn.close()
|
|
|
|
# Re-open: migration ladder runs.
|
|
conn = _open(db_path)
|
|
_ensure_schema(conn)
|
|
assert get_schema_version(conn) == SCHEMA_VERSION
|
|
# Singleton row must exist with NULL content (= no banner).
|
|
rows = conn.execute(
|
|
"SELECT id, content, updated_at, updated_by FROM setup_banner"
|
|
).fetchall()
|
|
assert len(rows) == 1
|
|
assert rows[0][0] == 1 # singleton id
|
|
assert rows[0][1] is None # NULL = no banner
|
|
|
|
|
|
def test_fresh_install_seeds_setup_banner(tmp_path):
|
|
db_path = tmp_path / "system.duckdb"
|
|
conn = _open(db_path)
|
|
_ensure_schema(conn)
|
|
assert get_schema_version(conn) == SCHEMA_VERSION
|
|
rows = conn.execute("SELECT id, content FROM setup_banner").fetchall()
|
|
assert len(rows) == 1
|
|
assert rows[0][0] == 1
|
|
assert rows[0][1] is None
|
|
|
|
|
|
def test_welcome_template_unaffected_by_v22(tmp_path):
|
|
"""welcome_template table must still coexist after v22 migration."""
|
|
db_path = tmp_path / "system.duckdb"
|
|
conn = _open(db_path)
|
|
_ensure_schema(conn)
|
|
rows = conn.execute("SELECT id, content FROM welcome_template").fetchall()
|
|
assert len(rows) == 1
|
|
assert rows[0][0] == 1
|