lauadminandClaude Sonnet 5
f4072be451
Wires the CRM (exoCRM GetOrders) into the dashboard as a locally persisted order queue instead of the previous static mockup: - CrmClient Protocol + ExoCrmClient/StubCrmClient for the CRM's signed JSON-RPC API - Order model + migration, synced from CRM on each queue view; soft-deleted orders stay hidden across re-syncs - GET/DELETE /api/v1/orders with "no receipt"/"receipt issued" tabs (the latter is empty until Checkbox fiscalization lands) - Dashboard: real order list, item-detail modal, tab switcher, one-click delete Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
91 lines
4.1 KiB
Python
91 lines
4.1 KiB
Python
"""Миграции обязаны точно соответствовать моделям.
|
|
|
|
Расхождение между `Base.metadata` и миграциями обнаруживается только на живой
|
|
БД и обычно уже в проде. Проверка статическая: сравниваем таблицы, колонки,
|
|
nullability и индексы, которые создают миграции, с тем, что описано моделями.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import re
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from app.db.models import Base
|
|
|
|
VERSIONS_DIR = Path(__file__).resolve().parents[1] / "alembic" / "versions"
|
|
MIGRATION_FILES = sorted(VERSIONS_DIR.glob("*.py"))
|
|
SOURCES = {path.name: path.read_text(encoding="utf-8") for path in MIGRATION_FILES}
|
|
# Склеенный текст всех миграций — таблица может быть создана в любой из них.
|
|
COMBINED_SOURCE = "\n".join(SOURCES.values())
|
|
|
|
MODEL_TABLES = set(Base.metadata.tables)
|
|
|
|
|
|
def _migration_block(table: str) -> str:
|
|
"""Тело вызова op.create_table для указанной таблицы.
|
|
|
|
Границей служит следующий op.create_table; для последней таблицы — конец файла.
|
|
"""
|
|
start = COMBINED_SOURCE.index(f'op.create_table(\n "{table}",')
|
|
next_table = COMBINED_SOURCE.find("op.create_table(", start + 1)
|
|
end = next_table if next_table != -1 else len(COMBINED_SOURCE)
|
|
return COMBINED_SOURCE[start:end]
|
|
|
|
|
|
def test_migrations_create_every_model_table() -> None:
|
|
created = set(re.findall(r'op\.create_table\(\n\s+"(\w+)"', COMBINED_SOURCE))
|
|
assert created == MODEL_TABLES, (
|
|
f"Миграции и модели разошлись. Только в моделях: {MODEL_TABLES - created}; "
|
|
f"только в миграциях: {created - MODEL_TABLES}"
|
|
)
|
|
|
|
|
|
def test_each_migration_downgrade_drops_its_own_tables() -> None:
|
|
"""Каждая миграция откатывает ровно те таблицы, которые сама создаёт."""
|
|
for name, source in SOURCES.items():
|
|
created = set(re.findall(r'op\.create_table\(\n\s+"(\w+)"', source))
|
|
downgrade = source[source.index("def downgrade()") :]
|
|
dropped = set(re.findall(r'op\.drop_table\("(\w+)"\)', downgrade))
|
|
assert dropped == created, f"{name}: downgrade не удаляет {created - dropped}"
|
|
|
|
|
|
@pytest.mark.parametrize("table_name", sorted(MODEL_TABLES))
|
|
def test_columns_match(table_name: str) -> None:
|
|
block = _migration_block(table_name)
|
|
in_migration = set(re.findall(r'sa\.Column\(\n?\s*"(\w+)"', block))
|
|
in_model = {column.name for column in Base.metadata.tables[table_name].columns}
|
|
|
|
assert in_migration == in_model, (
|
|
f"Таблица {table_name}: только в модели {in_model - in_migration}, "
|
|
f"только в миграции {in_migration - in_model}"
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize("table_name", sorted(MODEL_TABLES))
|
|
def test_nullability_matches(table_name: str) -> None:
|
|
block = _migration_block(table_name)
|
|
model_table = Base.metadata.tables[table_name]
|
|
|
|
for column in model_table.columns:
|
|
match = re.search(rf'sa\.Column\(\n?\s*"{column.name}".*?\n', block)
|
|
assert match, f"{table_name}.{column.name} отсутствует в миграции"
|
|
|
|
migration_nullable = "nullable=True" in match.group(0)
|
|
assert migration_nullable == column.nullable, (
|
|
f"{table_name}.{column.name}: в модели nullable={column.nullable}, "
|
|
f"в миграции nullable={migration_nullable}"
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize("table_name", sorted(MODEL_TABLES))
|
|
def test_indexes_match(table_name: str) -> None:
|
|
model_indexes = {index.name for index in Base.metadata.tables[table_name].indexes}
|
|
migration_indexes = set(
|
|
re.findall(r'op\.create_index\(\s*(?:op\.f\()?"(\w+)"', COMBINED_SOURCE)
|
|
)
|
|
|
|
missing = model_indexes - migration_indexes
|
|
assert not missing, f"Таблица {table_name}: миграция не создаёт индексы {missing}"
|