- Зачем: - необходимо визуализировать выбор типа хранения (Heap vs Append-Only) для учебных целей. - автоматизировать проверку всей цепочки DWH для исключения ручных ошибок. - сделать процесс отладки прозрачным и наглядным через стандартные инструменты Airflow. - Что: - внедрена клауза WITH (appendonly=...) во все DDL; ODS-справочники переведены на AO Row и TRUNCATE+INSERT. - создан скрипт scripts/e2e_etl.sh для полного прогона ETL (DDL + 2 дня данных) через REST API. - исправлены баги типизации (INTEGER[]), именования полей (amount, passenger_id) и удалены фантомные колонки (contact_data). - обновлен e2e-etl-test-protocol.md: добавлен раздел по отладке, чтению логов и перезапуску задач через API. - исправлены pytest-контракты под новую логику загрузки. - Проверка: - успешный прогон `make e2e-smoke` (полный цикл от очистки до витрины).
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:
|
|
"""Резолвер батча должен искать batch_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
|