group_fqcd_jr/tests/unit/test_advisor_migration_contract.py

from __future__ import annotations

import re
from pathlib import Path

from alembic.config import Config
from alembic.script import ScriptDirectory

ROOT = Path(__file__).resolve().parents[2]
VERSIONS = ROOT / "alembic" / "versions"
BASELINE = ROOT / "alembic" / "baseline_generated.sql"

ADVISOR_FILES = (
    "20260910_advisor_investment_goal.py",
    "20260910_advisor_product_industry_exposure.py",
    "20260910_advisor_portfolio_projection_checkpoint.py",
    "20260910_advisor_product_metric_snapshot.py",
    "20260910_advisor_product_asset_classification.py",
    "20260910_advisor_product_reference_snapshot.py",
    "20260910_advisor_offsite_fund_reference.py",
    "20260910_advisor_product_price_history.py",
    "20260910_advisor_product_governance_reference.py",
    "20260910_advisor_governance_monitor_and_quotes.py",
    "20260910_advisor_market_quote_resilience.py",
    "20260910_advisor_data_quality_backtest.py",
    "20260911_advisor_goal_conversation.py",
    "20260911_advisor_profile_tag_governance.py",
)


def created_tables(path: Path) -> set[str]:
    content = path.read_text(encoding="utf-8")
    return set(re.findall(r"CREATE TABLE(?: IF NOT EXISTS)?\s+`?([A-Za-z0-9_]+)`?", content))


def test_advisor_migrations_form_one_chain_from_qyqy_head() -> None:
    script = ScriptDirectory.from_config(Config(str(ROOT / "alembic.ini")))
    assert len(script.get_heads()) == 1

    # ⚠️ 这里**不写死 head 名**。原先断言
    #   `script.get_heads()[0] == "20260911_merge_adv_risk_heads"`
    # 那是"投顾迁移刚加完那一刻"的快照；之后任何人新增迁移（2026-09-12 的
    # 客户权益迁移 `20260912_customer_benefit` 即是一例）都会让本用例变红，
    # 而变红的原因与投顾链的对错**无关** —— 断言测到的是时间，不是契约。
    #
    # 原意是"投顾链确实接在这条主链上、没另起一条分支"。改为断言
    # **投顾链尾是当前 head 的祖先**：既保住这个意思，又不受后续迁移影响。
    head = script.get_heads()[0]
    ancestors = {rev.revision for rev in script.iterate_revisions(head, "base")}
    advisor_tail = re.search(
        r'revision = "([^"]+)"',
        (VERSIONS / ADVISOR_FILES[-1]).read_text(encoding="utf-8"),
    )
    assert advisor_tail is not None
    assert advisor_tail.group(1) in ancestors

    first = (VERSIONS / ADVISOR_FILES[0]).read_text(encoding="utf-8")
    assert 'down_revision = "20260910_drop_review_separation"' in first
    for previous, current in zip(ADVISOR_FILES, ADVISOR_FILES[1:], strict=False):
        previous_content = (VERSIONS / previous).read_text(encoding="utf-8")
        current_content = (VERSIONS / current).read_text(encoding="utf-8")
        previous_revision = re.search(r'revision = "([^"]+)"', previous_content)
        assert previous_revision is not None
        assert f'down_revision = "{previous_revision.group(1)}"' in current_content


def test_advisor_migrations_only_create_additive_tables() -> None:
    baseline_tables = created_tables(BASELINE)
    advisor_tables: set[str] = set()
    for filename in ADVISOR_FILES:
        content = (VERSIONS / filename).read_text(encoding="utf-8")
        assert "ALTER TABLE" not in content
        assert "DROP TABLE" not in content
        advisor_tables.update(created_tables(VERSIONS / filename))

    assert advisor_tables
    assert not advisor_tables & baseline_tables
    assert "fin_sim_order" not in advisor_tables
    assert "fin_transaction" not in advisor_tables
    assert "fin_holding" not in advisor_tables