"""Sanity tests for SQL migration files — no DB or network required."""
from __future__ import annotations

from pathlib import Path

REPO_ROOT = Path(__file__).parent.parent.parent.parent
INIT_DIR = REPO_ROOT / "services" / "kb-postgres" / "init"


def test_migration_001_exists():
    assert (INIT_DIR / "001_envelope.sql").exists(), \
        f"Expected {INIT_DIR / '001_envelope.sql'} to exist"


def test_migration_001_enables_vector_extension():
    sql = (INIT_DIR / "001_envelope.sql").read_text()
    assert "CREATE EXTENSION" in sql
    assert "vector" in sql.lower()


def test_migration_001_creates_envelope_table():
    sql = (INIT_DIR / "001_envelope.sql").read_text()
    assert "CREATE TABLE" in sql
    assert "envelope" in sql.lower()


def test_migration_001_has_all_frozen_columns():
    sql = (INIT_DIR / "001_envelope.sql").read_text()
    frozen_columns = ("id", "source", "ts", "geo", "raw_ref", "entities")
    for col in frozen_columns:
        assert col in sql, f"Frozen column '{col}' missing from 001_envelope.sql"


def test_migration_001_geo_is_nullable():
    sql = (INIT_DIR / "001_envelope.sql").read_text()
    # geo must NOT have NOT NULL constraint
    lines = [l for l in sql.splitlines() if "geo" in l.lower()]
    assert lines, "No 'geo' line found in migration SQL"
    for line in lines:
        assert "NOT NULL" not in line.upper(), \
            f"geo must be nullable but found: {line.strip()}"


def test_migration_001_entities_has_default():
    sql = (INIT_DIR / "001_envelope.sql").read_text()
    lines = [l for l in sql.splitlines() if "entities" in l.lower()]
    assert any("DEFAULT" in l.upper() for l in lines), \
        "entities column must have a DEFAULT '[]'"