Files
trex-core/tests/application/test_snapshot_handler.py
T
genichinandClaude Opus 4.8 7d727824dc feat(slot): BAT-030 — 자산 중심 cutover 완성 (dcy-020 BAT-δ, REQ-056·057·062·064·065)
M1 literal inactive 전면화: SlotState 5-state(idle/closed 별칭 제거),
마이그레이션 010(CHECK 교체+행 rewrite), ops_checks 를 transition_log 기반 재작성.
M2 집행 매도 수량 브로커 정본(D3): dispatch 매도 qty=broker hldg_qty,
미러 fallback 없음, 브로커 0 → skip+WARN, 리스트 기반 dispatch(head-block 방지).
M3 컬럼 drop + legacy pool 제거: 마이그레이션 011(trading_date·entry_count DROP),
repo 계약 instance 스코프 개편(집행 큐 포함), preallocation 폐기(데몬 7스레드),
admission 자산 중심 단일 경로(slot_id 정본), capacity 재정의(available=M−active),
daily_symbol_entries 코드 경로 제거.
M4 cutover 도구(D1·D4): trex-core-cutover(전체 rebuild, preflight/브로커 carry-over/
--allow-unmapped/--promote-unique/dry-run) + PG 리허설 테스트 8건.
M5 baseline 마감: data-model/runtime-flows/project-structure/interface-contract
자산 중심 최종 갱신 + run-trex-core-daemon 스킬 정합.

검증: 단위 367 passed + ruff clean; PG-gated 29 passed(010/011 round-trip·드리프트
가드·cutover 리허설); RabbitMQ 21 passed; run-the-app e2e(test DB+실 MQ, M=1):
reconcile 슬롯 생성→slot_id 매수→dedup→capacity_full→fill→already_active→
매도→inactive(자산 바인딩 보존)→영구 슬롯 재활성 전 구간 green.
실 cutover 는 운영자 실행(REL 단계, D4).

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
2026-06-10 15:16:15 +09:00

96 lines
3.3 KiB
Python

"""Snapshot RPC handler envelope path (REQ-028 AC-1/AC-2/AC-3)."""
from __future__ import annotations
from datetime import date
from trex_core.application.messaging import RpcRequest
from trex_core.application.messaging_handlers import make_snapshot_rpc_handler
from trex_core.application.portfolio import (
DegradedReason,
InMemoryPauseStore,
PortfolioItem,
SnapshotStore,
)
from trex_core.domain.slot import SlotService
from trex_core.domain.slot.in_memory_repository import InMemorySlotRepository
DAY = date(2026, 6, 5)
def _wire(items=None):
repo = InMemorySlotRepository()
svc = SlotService(repo)
repo.reconcile_universe("default", {"pf-A"})
store = SnapshotStore()
store.swap(items or [
PortfolioItem(portfolio_id="pf-A", symbol="005930",
cash_balance=1_000_000, asset_quantity=0,
active_strategy_id="target-value-v1",
strategy_params={"k": 1}),
])
pause = InMemoryPauseStore()
handler = make_snapshot_rpc_handler(
repo, svc,
snapshot_store=store, pause_store=pause, max_active_slots=5,
)
return store, pause, handler
def test_envelope_shape_returned():
_, _, handler = _wire()
resp = handler(RpcRequest(
payload={"view": "all", "trading_date": DAY.isoformat()},
correlation_id="snap-1",
))
body = resp.payload
# BAT-022: `slots[]` additive operator-facing per-slot state.
assert set(body.keys()) == {"assets", "view", "available_slots", "degraded",
"reason", "message", "slots"}
assert isinstance(body["assets"], list)
assert isinstance(body["slots"], list)
assert body["view"] == "all"
assert body["degraded"] is False
assert body["reason"] is None
def test_envelope_strategy_key_filter():
_, _, handler = _wire([
PortfolioItem(portfolio_id="x", symbol="X", cash_balance=0, asset_quantity=0,
active_strategy_id="alpha"),
PortfolioItem(portfolio_id="y", symbol="Y", cash_balance=0, asset_quantity=0,
active_strategy_id="beta"),
])
resp = handler(RpcRequest(
payload={"view": "all", "strategy_key": "alpha"}, correlation_id="snap-2",
))
pids = {a["asset_id"] for a in resp.payload["assets"]}
assert pids == {"x"}
def test_degraded_signal_does_not_leak_credentials():
store, _, handler = _wire()
store.set_degraded(
DegradedReason.PORTFOLIO_HUB_UNAVAILABLE,
"GET https://hub failed: 401 Bearer secret-token-abc",
)
resp = handler(RpcRequest(payload={"view": "all"}, correlation_id="snap-3"))
body = resp.payload
assert body["degraded"] is True
assert body["reason"] == "portfolio_hub_unavailable"
assert body["message"] is not None
assert "secret-token-abc" not in body["message"]
def test_bat007_legacy_path_when_stores_unset():
"""Make sure pre-BAT-008 callers still work (slots+capacity stub shape)."""
repo = InMemorySlotRepository()
svc = SlotService(repo)
repo.reconcile_universe("default", {"pf-A", "pf-B", "pf-C"})
handler = make_snapshot_rpc_handler(repo, svc)
resp = handler(RpcRequest(
payload={"view": "all", "trading_date": DAY.isoformat()}, correlation_id="legacy",
))
body = resp.payload
assert "slots" in body and "capacity" in body
assert "assets" not in body