ddadmin
5972bcc2d8
- Зачем:
- STG-слой использовал legacy-имена (batch_id, load_dttm, src_created_at_ts),
тогда как ODS/DDS/DM уже работали с каноном (_load_id, _load_ts, event_ts).
Студент видел разные имена для одного понятия — это убрано.
- Что:
- переименованы колонки в 9 STG DDL: batch_id→_load_id, load_dttm→_load_ts,
src_created_at_ts→event_ts; добавлен NOT NULL для _load_id во всех таблицах.
- обновлены 9 STG Load, 9 STG DQ, 9 ODS Load, 9 ODS DQ (INSERT/SELECT/WHERE).
- обновлены DAG-файлы bookings_to_gp_stage.py и bookings_to_gp_ods.py
(встроенный SQL резолвера, комментарии; Python-идентификаторы не тронуты).
- обновлены тесты и ~15 документов (naming_conventions, PRD, db_schema,
design-docs, qa-plan, README, TESTING и др.).
- Проверка:
- grep -rn 'load_dttm\|src_created_at_ts' sql/ airflow/ tests/ — 0 совпадений.
- make test — 4 passed.
- e2e-etl: day1 прошёл полностью, day2 стартовал без ошибок.
51 lines
2.0 KiB
Python
51 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
PROJECT_ROOT = Path(__file__).resolve().parents[1]
|
|
SNAPSHOT_ENTITIES = ("airports", "airplanes", "routes", "seats")
|
|
|
|
|
|
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
|