genichinandClaude Opus 4.8
c04fd276b3
실 DB 점검 중 마이그레이션 009 부분 유니크 인덱스가 legacy 풀 데이터(같은 자산이 closed+재진입 holding 으로 portfolio_id 중복)에 적용 불가 → cutover· 데몬 부팅 모두 차단되는 결함 확인. 데이터 보존 cutover 를 폐기하고 DB wipe + 자산 중심 재초기화로 전환(사용자 결정). 빈 테이블에는 009 인덱스가 정상 생성되고 자산 중심에서 (instance_key, portfolio_id)는 설계상 유일. - trex_core/runtime/reset.py 신규(trex-core-reset): slot-core 7테이블 wipe + apply_migrations 재적용 → 자산 중심 스키마 착지 검증. dry-run 기본/--yes 적용, 파괴적·DB 전역. 보유 carry-over 없음(재초기화 후 reconcile_core_holdings 재편입). - runtime/cutover.py + trex-core-cutover console script + test_cutover_rehearsal.py 삭제. - pyproject: trex-core-reset 등록. - tests/runtime/test_reset.py 신규 5건(PG-gated): wipe+착지·dry-run 무변경·멱등· wipe 후 admission·테이블 커버리지. - change-req REQ-064 CHG-20260610-03(cutover→재초기화 재정의) + srs index 갱신. - baseline 4문서(data-model/runtime-flows/project-structure/interface-contract) + BAT-030 design/planning/index/implementation/verification 도구 서술 reset 으로 교체. 검증: ruff clean, 단위 367 passed, PG+RabbitMQ 통합 411 passed/1 skip(KIS sandbox). Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
90 lines
3.6 KiB
Python
90 lines
3.6 KiB
Python
"""DB reset tool tests (dcy-020 / REQ-064, BAT-030 M4).
|
|
|
|
PostgreSQL-gated (TREX_POSTGRESQL_TEST_DSN) — the wipe-and-reinitialize tool
|
|
that replaced the data-preserving cutover: wipes the slot-core tables and
|
|
reapplies migrations so the DB lands on the clean asset-centric schema.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
from trex_core.adapters.persistence.postgresql import PostgreSQLSlotRepository
|
|
from trex_core.domain.slot import EXPECTED_SLOT_FIELDS, SlotService, SlotState
|
|
from trex_core.runtime.reset import SLOT_CORE_TABLES, run_reset
|
|
|
|
INST = "default"
|
|
|
|
|
|
def _seed(dsn):
|
|
"""Create asset slots + drive one into holding so there are rows to wipe."""
|
|
repo = PostgreSQLSlotRepository(dsn)
|
|
svc = SlotService(repo, instance_key=INST)
|
|
repo.reconcile_universe(INST, {"pf-a", "pf-b"})
|
|
s = repo.find_slot_by_portfolio(INST, "pf-a")
|
|
svc.transit_slot(s.slot_id, SlotState.INACTIVE, SlotState.PENDING_BUY,
|
|
reason="r", actor="t", symbol="005930", qty=10)
|
|
return repo
|
|
|
|
|
|
def test_reset_wipes_and_lands_asset_centric(pg_clean):
|
|
repo = _seed(pg_clean)
|
|
assert len(repo.list_slots(INST)) == 2
|
|
|
|
report = run_reset(pg_clean, dry_run=False)
|
|
|
|
assert report.dry_run is False
|
|
assert report.schema_ok is True
|
|
assert report.wiped["daily_slot"] == 2
|
|
assert repo.list_slots(INST) == [] # wiped
|
|
# schema is the clean asset-centric form
|
|
import psycopg
|
|
with psycopg.connect(pg_clean) as c, c.cursor() as cur:
|
|
cur.execute("SELECT column_name FROM information_schema.columns "
|
|
"WHERE table_name='daily_slot'")
|
|
cols = {r[0] for r in cur.fetchall()}
|
|
assert cols == set(EXPECTED_SLOT_FIELDS)
|
|
cur.execute("SELECT pg_get_constraintdef(oid) FROM pg_constraint "
|
|
"WHERE conname='daily_slot_status_check'")
|
|
assert "inactive" in cur.fetchone()[0]
|
|
cur.execute("SELECT indexname FROM pg_indexes WHERE tablename='daily_slot'")
|
|
assert "uq_daily_slot_instance_portfolio" in {r[0] for r in cur.fetchall()}
|
|
|
|
|
|
def test_dry_run_reports_without_wiping(pg_clean):
|
|
repo = _seed(pg_clean)
|
|
report = run_reset(pg_clean, dry_run=True)
|
|
assert report.dry_run is True
|
|
assert report.wiped["daily_slot"] == 2 # reported
|
|
assert len(repo.list_slots(INST)) == 2 # but NOT wiped
|
|
|
|
|
|
def test_reset_is_idempotent(pg_clean):
|
|
_seed(pg_clean)
|
|
run_reset(pg_clean, dry_run=False)
|
|
report2 = run_reset(pg_clean, dry_run=False) # second pass on clean DB
|
|
assert report2.schema_ok is True
|
|
assert report2.wiped["daily_slot"] == 0
|
|
|
|
|
|
def test_reset_usable_for_admission_after_wipe(pg_clean):
|
|
"""After reset the DB is immediately usable: reconcile + activate work."""
|
|
_seed(pg_clean)
|
|
run_reset(pg_clean, dry_run=False)
|
|
repo = PostgreSQLSlotRepository(pg_clean)
|
|
svc = SlotService(repo, instance_key=INST)
|
|
repo.reconcile_universe(INST, {"pf-x"})
|
|
s = repo.find_slot_by_portfolio(INST, "pf-x")
|
|
assert s.status is SlotState.INACTIVE
|
|
svc.transit_slot(s.slot_id, SlotState.INACTIVE, SlotState.PENDING_BUY,
|
|
reason="r", actor="t", symbol="005930")
|
|
assert repo.count_active(INST) == 1
|
|
|
|
|
|
def test_covers_all_slot_core_tables(pg_clean):
|
|
"""Sanity: the tool targets every slot-core table the migrations create."""
|
|
import psycopg
|
|
with psycopg.connect(pg_clean) as c, c.cursor() as cur:
|
|
cur.execute("SELECT table_name FROM information_schema.tables "
|
|
"WHERE table_schema='public' AND table_name = ANY(%s)",
|
|
(list(SLOT_CORE_TABLES),))
|
|
present = {r[0] for r in cur.fetchall()}
|
|
assert present == set(SLOT_CORE_TABLES)
|