ddadminandClaude Opus 4.6
5ca4e6b2ca
- fact_flight_sales_dq: student SK (passenger/route/airplane) → NOTICE, airport_sk → порог 1% (эталон через ods.routes) - routes_dq: закомментирован RI airplane_code → ods.airplanes - ODS DAG: убрана зависимость dq_ods_airplanes → load_ods_routes - 18 студенческих файлов заменены заглушками (ODS/DDS/DM load+dq) - test_ods_sql_contract: SNAPSHOT_ENTITIES без airplanes/seats - test_dags_smoke: убран assert dq_ods_airplanes → dq_ods_routes Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
53 lines
2.1 KiB
Python
53 lines
2.1 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
PROJECT_ROOT = Path(__file__).resolve().parents[1]
|
|
# На main airplanes/seats — заглушки (SELECT 1;), не содержат TRUNCATE-паттерн.
|
|
# Полный список на ветке solution: ("airports", "airplanes", "routes", "seats")
|
|
SNAPSHOT_ENTITIES = ("airports", "routes")
|
|
|
|
|
|
def _read(path: str) -> str:
|
|
return (PROJECT_ROOT / path).read_text(encoding="utf-8")
|
|
|
|
|
|
def test_snapshot_load_scripts_use_truncate() -> None:
|
|
"""Snapshot-таблицы в ODS должны использовать паттерн полной перезагрузки (TRUNCATE)."""
|
|
for entity in SNAPSHOT_ENTITIES:
|
|
sql = _read(f"sql/ods/{entity}_load.sql")
|
|
assert f"TRUNCATE TABLE ods.{entity};" in sql
|
|
assert f"INSERT INTO ods.{entity} (" in sql
|
|
assert "WHERE s.rn = 1;" in sql
|
|
|
|
|
|
def test_snapshot_dq_checks_extra_keys() -> None:
|
|
"""DQ snapshot-таблиц должен ловить лишние ключи в ODS относительно текущего батча STG."""
|
|
for entity in SNAPSHOT_ENTITIES:
|
|
sql = _read(f"sql/ods/{entity}_dq.sql")
|
|
assert "v_extra_keys_count" in sql
|
|
assert "найдены лишние ключи" in sql
|
|
|
|
|
|
def test_ods_batch_resolver_uses_consistent_snapshot_batches() -> None:
|
|
"""Резолвер батча должен искать _load_id, общий для всех snapshot-таблиц STG."""
|
|
dag_code = _read("airflow/dags/bookings_to_gp_ods.py")
|
|
|
|
for table_name in ("stg.airports", "stg.airplanes", "stg.routes", "stg.seats"):
|
|
assert table_name in dag_code
|
|
|
|
assert "INTERSECT" in dag_code
|
|
assert "candidate_batches" in dag_code
|
|
|
|
|
|
def test_flights_load_covers_segment_flight_ids_from_stg_history() -> None:
|
|
"""
|
|
Загрузка flights должна подтягивать рейсы из истории STG,
|
|
если на них ссылаются segments текущего батча.
|
|
"""
|
|
sql = _read("sql/ods/flights_load.sql")
|
|
|
|
assert "segment_flights" in sql
|
|
assert "FROM stg.segments AS s" in sql
|
|
assert "JOIN segment_flights AS sf" in sql
|