genichinandClaude Opus 4.8
a30efa8f3b
dcy-007(core 자율 안전 매도) 전체 delivery. BAT-010 기반 인프라가 미커밋 상태였고 BAT-011 트리거 레이어와 공유 파일에서 섞여 있어 한 커밋으로 묶음. BAT-010 (REQ-034/035/036/041, release-candidate): - WatchdogStore(application/watchdog.py) + snapshot RPC 워치독 갱신 - Slot stop_loss_price/take_profit_price 필드 + 004_add_price_thresholds.sql - SlotService.sell() 자율 매도 내부 경로 + idempotency - SafetySellConfig env 5종 파싱 BAT-011 (REQ-037/038/039/040/042, release-candidate): - SafetySellService(application/safety_sell.py): forced_close 만기 / stop-loss· take-profit KIS polling / 장 마감 강제 청산 + trex-safety-sell 7번째 스레드 - BrokerPort.get_current_price + KIS inquire-price(TTTC8397R/VTTC8397R, stck_prpr) + paper stub - SlotService.update_thresholds() + intent RPC update_thresholds 액션 (부분 갱신/null 해제, 워치독 last_seen_at 리셋) - baseline 3문서 정합(interface-contract/runtime-flows/project-structure) 거버넌스: REQ-034~042 Implemented 동기화, BAT-010/011 release-candidate, dcy-007 discovery 포함. 검증: 신규 38 테스트 green, 단위 236 passed/23 skipped, 인프라 게이트(PG+RabbitMQ) 259 passed(마이그레이션 001~004 round-trip 포함), ruff clean. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
97 lines
3.6 KiB
Python
97 lines
3.6 KiB
Python
"""Snapshot RPC handler watchdog integration tests (REQ-034 AC-1 / BAT-010)."""
|
|
from __future__ import annotations
|
|
|
|
from datetime import date
|
|
|
|
import pytest
|
|
|
|
from trex_core.application.messaging import RpcRequest
|
|
from trex_core.application.messaging_handlers import make_snapshot_rpc_handler
|
|
from trex_core.application.watchdog import WatchdogStore
|
|
from trex_core.domain.slot import SlotService, SlotState
|
|
from trex_core.domain.slot.in_memory_repository import InMemorySlotRepository
|
|
|
|
|
|
@pytest.fixture
|
|
def trading_day():
|
|
return date(2026, 6, 7)
|
|
|
|
|
|
@pytest.fixture
|
|
def repo():
|
|
return InMemorySlotRepository()
|
|
|
|
|
|
@pytest.fixture
|
|
def service(repo):
|
|
return SlotService(repo)
|
|
|
|
|
|
def _make_req(trading_day):
|
|
return RpcRequest(
|
|
payload={"trading_date": trading_day.isoformat(), "view": "all"},
|
|
correlation_id="test-corr",
|
|
)
|
|
|
|
|
|
def _buy_slot(service, repo, trading_day, symbol="005930"):
|
|
"""Preallocate + buy one slot into HOLDING state."""
|
|
service.preallocate_daily_slots(trading_day, 1)
|
|
slot = repo.find_idle_slot(trading_day)
|
|
pending = service.transit_slot(
|
|
slot.slot_id, SlotState.IDLE, SlotState.PENDING_BUY,
|
|
reason="buy", actor="test", symbol=symbol,
|
|
)
|
|
holding = service.transit_slot(
|
|
pending.slot_id, SlotState.PENDING_BUY, SlotState.HOLDING,
|
|
reason="filled", actor="test", avg_price=100.0, qty=10.0,
|
|
)
|
|
return holding
|
|
|
|
|
|
class TestSnapshotRpcWatchdog:
|
|
def test_holding_slot_watchdog_updated_on_snapshot_rpc(self, repo, service, trading_day):
|
|
"""AC-1: snapshot RPC updates watchdog for each holding slot in response."""
|
|
holding = _buy_slot(service, repo, trading_day)
|
|
watchdog = WatchdogStore()
|
|
handler = make_snapshot_rpc_handler(repo, service, watchdog_store=watchdog)
|
|
handler(_make_req(trading_day))
|
|
assert str(holding.slot_id) in watchdog._store
|
|
|
|
def test_idle_slot_watchdog_not_updated(self, repo, service, trading_day):
|
|
"""Only HOLDING slots update the watchdog — idle slots are ignored."""
|
|
service.preallocate_daily_slots(trading_day, 1)
|
|
idle = repo.find_idle_slot(trading_day)
|
|
watchdog = WatchdogStore()
|
|
handler = make_snapshot_rpc_handler(repo, service, watchdog_store=watchdog)
|
|
handler(_make_req(trading_day))
|
|
assert str(idle.slot_id) not in watchdog._store
|
|
|
|
def test_no_watchdog_store_does_not_raise(self, repo, service, trading_day):
|
|
"""Without watchdog_store= the handler works as before (backward compat)."""
|
|
_buy_slot(service, repo, trading_day)
|
|
handler = make_snapshot_rpc_handler(repo, service)
|
|
resp = handler(_make_req(trading_day))
|
|
assert resp is not None
|
|
|
|
def test_multiple_holding_slots_all_updated(self, repo, service, trading_day):
|
|
"""All holding slots in the response have their watchdog updated."""
|
|
service.preallocate_daily_slots(trading_day, 3)
|
|
slots = []
|
|
for sym in ("A", "B", "C"):
|
|
slot = repo.find_idle_slot(trading_day)
|
|
p = service.transit_slot(
|
|
slot.slot_id, SlotState.IDLE, SlotState.PENDING_BUY,
|
|
reason="buy", actor="test", symbol=sym,
|
|
)
|
|
h = service.transit_slot(
|
|
p.slot_id, SlotState.PENDING_BUY, SlotState.HOLDING,
|
|
reason="filled", actor="test", avg_price=100.0, qty=1.0,
|
|
)
|
|
slots.append(h)
|
|
watchdog = WatchdogStore()
|
|
handler = make_snapshot_rpc_handler(repo, service, watchdog_store=watchdog)
|
|
handler(_make_req(trading_day))
|
|
for s in slots:
|
|
assert str(s.slot_id) in watchdog._store, f"{s.slot_id} not in watchdog"
|