Files
trex-core/tests/application/test_messaging_snapshot_consumer.py
T
genichinandClaude Opus 4.8 d09b93195f feat(cli): dcy-016 운영자 CLI 확장 (universe/slots/enter/exit) + 워치독 억제 (BAT-020)
운영자가 조회·매수·매도를 CLI 로 수행하되 데몬 RabbitMQ RPC 로 통일해 단일
writer admission·불변식을 보존(DB 직접 쓰기 없음)하고, 운영자 조회가 전략
생존 신호로 오인되지 않게 워치독을 억제한다.

거버넌스: 전부 change-req (신규 REQ 없음, dcy-016 OQ-3).
- REQ-032: CLI universe/slots(snapshot RPC) + enter(buy)/exit(sell) intent RPC
  + 인스턴스 라우팅(--config/--instance/default) + enter/exit 안전장치
  (확인 프롬프트 기본 ON / --yes(paper) / --yes --live(이중 ack) / --dry-run)
  + 전 명령 --json. pause/resume 는 direct-DB 유지
- REQ-022: CLI request-reply RPC 클라이언트(call_rpc) 재사용
- REQ-021/034: snapshot 요청 워치독 억제 플래그(touch_watchdog/actor=operator)
  → 핸들러가 last_seen_at 갱신 스킵. 기본값 현행 유지(하위호환). 운영자
  조회가 죽은 전략의 safety-sell 을 무력화하던 결함(R-4) 제거
- REQ-026: interface-contract/runtime-flows 정합

CLI 는 rpc/confirm 주입 가능(테스트 broker/DB-free). snapshot envelope 은
asset-centric 이라 exit 은 운영자 제공 slot_id 사용(slot_id 노출은 본 주기 밖).

검증: 단위 327 passed/34 skipped, 인프라 게이트(PG+RabbitMQ) 361 passed, ruff clean.
신규 test_cli_operator(14) + snapshot 워치독 억제(2). BAT-020 release-candidate.

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

120 lines
4.2 KiB
Python

"""Snapshot RPC handler tests (REQ-021 AC-2, BAT-007 stub-shape)."""
from __future__ import annotations
from datetime import date
from trex_core.adapters.persistence.in_memory_intent_store import InMemoryIntentStore
from trex_core.application.intents import BuyIntent, IntentService
from trex_core.application.messaging import RpcRequest
from trex_core.application.messaging_handlers import make_snapshot_rpc_handler
from trex_core.domain.slot import SlotService
from trex_core.domain.slot.in_memory_repository import InMemorySlotRepository
DAY = date(2026, 6, 5)
def _wire():
repo = InMemorySlotRepository()
svc = SlotService(repo)
intents = IntentService(svc, repo, InMemoryIntentStore())
svc.preallocate_daily_slots(DAY, 3)
return repo, svc, intents
def _req(view="all", strategy_id=None):
return RpcRequest(
payload={
"view": view,
"strategy_id": strategy_id,
"trading_date": DAY.isoformat(),
},
correlation_id="snap-cid",
)
def test_snapshot_returns_all_slots_and_capacity():
repo, svc, intents = _wire()
intents.submit_buy(BuyIntent(
intent_id="b1", trading_date=DAY, symbol="005930", qty=10,
strategy_id="A", portfolio_id="PF-1",
))
handler = make_snapshot_rpc_handler(repo, svc)
resp = handler(_req())
assert resp.payload["trading_date"] == DAY.isoformat()
assert resp.payload["view"] == "all"
assert len(resp.payload["slots"]) == 3 # all 3 slots returned
cap = resp.payload["capacity"]
assert cap["pending_count"] == 1 and cap["idle_count"] == 2
def test_snapshot_open_view_filters_to_active_slots():
repo, svc, intents = _wire()
intents.submit_buy(BuyIntent(
intent_id="b1", trading_date=DAY, symbol="005930", qty=10, strategy_id="A",
portfolio_id="PF-1",
))
handler = make_snapshot_rpc_handler(repo, svc)
resp = handler(_req(view="open"))
statuses = {s["status"] for s in resp.payload["slots"]}
assert "idle" not in statuses
assert statuses == {"pending_buy"}
def test_snapshot_strategy_id_filter():
repo, svc, intents = _wire()
intents.submit_buy(BuyIntent(
intent_id="b1", trading_date=DAY, symbol="005930", qty=10, strategy_id="A",
portfolio_id="PF-1",
))
intents.submit_buy(BuyIntent(
intent_id="b2", trading_date=DAY, symbol="000660", qty=5, strategy_id="B",
portfolio_id="PF-2",
))
handler = make_snapshot_rpc_handler(repo, svc)
resp = handler(_req(view="open", strategy_id="A"))
assert len(resp.payload["slots"]) == 1
assert resp.payload["slots"][0]["strategy_id"] == "A"
assert resp.payload["slots"][0]["symbol"] == "005930"
# --- dcy-016 / REQ-021/034: operator query watchdog suppression -------------
class _RecWatchdog:
def __init__(self):
self.updated = []
def update(self, slot_id):
self.updated.append(str(slot_id))
def _holding_slot(repo, svc):
from trex_core.domain.slot import SlotState
slot = next(s for s in repo.list_slots(DAY) if s.status is SlotState.IDLE)
svc.transit_slot(slot.slot_id, SlotState.IDLE, SlotState.PENDING_BUY,
reason="t", actor="t", symbol="005930", portfolio_id="PF-1", qty=10)
svc.transit_slot(slot.slot_id, SlotState.PENDING_BUY, SlotState.HOLDING,
reason="t", actor="t", avg_price=1000)
return str(slot.slot_id)
def test_strategy_poll_touches_watchdog_by_default():
repo, svc, intents = _wire()
sid = _holding_slot(repo, svc)
wd = _RecWatchdog()
handler = make_snapshot_rpc_handler(repo, svc, watchdog_store=wd)
handler(_req()) # no actor flag => legacy liveness touch (REQ-034)
assert sid in wd.updated
def test_operator_query_suppresses_watchdog_touch():
repo, svc, intents = _wire()
_holding_slot(repo, svc)
wd = _RecWatchdog()
handler = make_snapshot_rpc_handler(repo, svc, watchdog_store=wd)
req = RpcRequest(
payload={"view": "all", "trading_date": DAY.isoformat(),
"actor": "operator", "touch_watchdog": False},
correlation_id="op",
)
handler(req)
assert wd.updated == [] # dcy-016 R-4: operator query is not a liveness signal