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>
112 lines
4.5 KiB
Python
112 lines
4.5 KiB
Python
"""SlotService.sell() unit tests (REQ-036 / BAT-010)."""
|
|
from __future__ import annotations
|
|
|
|
from datetime import date
|
|
|
|
import pytest
|
|
|
|
from trex_core.domain.slot import (
|
|
SlotNotFoundError,
|
|
SlotService,
|
|
SlotState,
|
|
SlotTransitionError,
|
|
)
|
|
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 _holding_slot(service, repo, trading_day):
|
|
"""Create one slot in HOLDING state."""
|
|
service.preallocate_daily_slots(trading_day, 1)
|
|
idle = repo.find_idle_slot(trading_day)
|
|
pending = service.transit_slot(
|
|
idle.slot_id, SlotState.IDLE, SlotState.PENDING_BUY,
|
|
reason="buy", actor="test", symbol="005930",
|
|
)
|
|
return service.transit_slot(
|
|
pending.slot_id, SlotState.PENDING_BUY, SlotState.HOLDING,
|
|
reason="filled", actor="test", avg_price=50000.0, qty=1.0,
|
|
)
|
|
|
|
|
|
class TestSell:
|
|
def test_sell_holding_transitions_to_pending_sell(self, service, repo, trading_day):
|
|
"""AC-1: sell() on a holding slot transitions it to pending_sell."""
|
|
holding = _holding_slot(service, repo, trading_day)
|
|
service.sell(holding.slot_id, actor="safety-sell", reason="stop_loss")
|
|
updated = repo.fetch_slot(holding.slot_id)
|
|
assert updated.status is SlotState.PENDING_SELL
|
|
|
|
def test_sell_records_transition_log(self, service, repo, trading_day):
|
|
"""sell() appends a transition log entry."""
|
|
holding = _holding_slot(service, repo, trading_day)
|
|
service.sell(holding.slot_id, actor="safety-sell", reason="stop_loss")
|
|
log = repo.transition_log_for(holding.slot_id)
|
|
sell_entries = [e for e in log if e.to_state is SlotState.PENDING_SELL]
|
|
assert len(sell_entries) == 1
|
|
assert sell_entries[0].actor == "safety-sell"
|
|
assert sell_entries[0].reason == "stop_loss"
|
|
|
|
def test_sell_idempotent_on_pending_sell(self, service, repo, trading_day):
|
|
"""AC-2: sell() on already-pending_sell slot returns silently, no second transition."""
|
|
holding = _holding_slot(service, repo, trading_day)
|
|
service.sell(holding.slot_id, actor="safety-sell", reason="stop_loss")
|
|
# Call again — must not raise and must not add another transition
|
|
service.sell(holding.slot_id, actor="safety-sell", reason="stop_loss")
|
|
log = repo.transition_log_for(holding.slot_id)
|
|
sell_entries = [e for e in log if e.to_state is SlotState.PENDING_SELL]
|
|
assert len(sell_entries) == 1 # exactly one, not two
|
|
|
|
def test_sell_idle_slot_raises_transition_error(self, service, repo, trading_day):
|
|
"""AC-3: sell() on an idle slot raises SlotTransitionError."""
|
|
service.preallocate_daily_slots(trading_day, 1)
|
|
idle = repo.find_idle_slot(trading_day)
|
|
with pytest.raises(SlotTransitionError):
|
|
service.sell(idle.slot_id, actor="safety-sell", reason="stop_loss")
|
|
# Slot state unchanged
|
|
assert repo.fetch_slot(idle.slot_id).status is SlotState.IDLE
|
|
|
|
def test_sell_unknown_slot_raises_not_found(self, service, trading_day):
|
|
"""sell() with unknown slot_id raises SlotNotFoundError."""
|
|
from uuid import uuid4
|
|
with pytest.raises(SlotNotFoundError):
|
|
service.sell(uuid4(), actor="safety-sell", reason="stop_loss")
|
|
|
|
|
|
class TestSellPreservesPriceFields:
|
|
def test_sell_preserves_stop_loss_price(self, service, repo, trading_day):
|
|
"""Price threshold fields survive the sell transition."""
|
|
from decimal import Decimal
|
|
service.preallocate_daily_slots(trading_day, 1)
|
|
idle = repo.find_idle_slot(trading_day)
|
|
pending = service.transit_slot(
|
|
idle.slot_id, SlotState.IDLE, SlotState.PENDING_BUY,
|
|
reason="buy", actor="test", symbol="005930",
|
|
stop_loss_price=Decimal("45000"),
|
|
take_profit_price=Decimal("60000"),
|
|
)
|
|
holding = service.transit_slot(
|
|
pending.slot_id, SlotState.PENDING_BUY, SlotState.HOLDING,
|
|
reason="filled", actor="test", avg_price=50000.0, qty=1.0,
|
|
)
|
|
assert holding.stop_loss_price == Decimal("45000")
|
|
assert holding.take_profit_price == Decimal("60000")
|
|
service.sell(holding.slot_id, actor="safety-sell", reason="stop_loss")
|
|
updated = repo.fetch_slot(holding.slot_id)
|
|
assert updated.stop_loss_price == Decimal("45000")
|
|
assert updated.take_profit_price == Decimal("60000")
|