Files
airflow-greenplum/tests/test_ods_sql_contract.py
T
ddadmin 22a0590e1b feat(sql): явное управление storage и автоматизированный E2E-тест через REST API
- Зачем:
  - необходимо визуализировать выбор типа хранения (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` (полный цикл от очистки до витрины).
2026-03-03 22:38:26 +03:00

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