genichinandClaude Opus 4.8
e4be4d60c2
자산 영구 슬롯(dcy-020)의 수량 가변 전제 위에 보유 포지션 수량 조정 구현. 명령형 계약(전략이 cash+quantity 로 qty+price 계산), core 는 실행 substrate, 수량 정본 = 브로커 잔고 재동기화로 "주문 수량 vs 보유 수량" 충돌 해소(별도 필드 없음). - state.py: holding→pending_buy 전이 추가(증액) → 13 transitions (REQ-057) - service.py: PENDING_SELL 진입 qty=caller as-is(None=전량 sentinel, 값=부분 주문) (REQ-062) - intents.py: 증액 admission(holding 슬롯 매수=increase, M 캡 미적용), SellIntent.qty 부분 매도 (REQ-061) - execution.py: _balance_map; 매도 dispatch=브로커잔고(전량)/min(주문,잔고)(부분); 매수 체결 avg_price 가중평균+브로커 정본 resync; 매도 체결 잔고>0→holding/0→inactive; 증액 롤백 holding 복귀 (REQ-066/062) - messaging_handlers.py: sell payload qty 파싱(부분 매도) - baseline 4문서(data-model/runtime-flows/interface-contract/project-structure) 정합 (REQ-067) 검증: 단위 378 / 통합(PG+RabbitMQ) 422 passed, ruff clean. run-the-app 사이징 e2e: 매수 10@1000 → 증액 +5@1200 → 15@1066.67(가중평균) → 부분 매도 6 → 9@1066.67(cost basis) → 전량→inactive. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
108 lines
3.6 KiB
Python
108 lines
3.6 KiB
Python
"""State machine unit tests — asset-centric (dcy-020 / REQ-057, BAT-030).
|
|
|
|
`inactive` is the only no-position state. The pool-model `idle`/`closed`
|
|
aliases were removed at the cutover (D2); migration 010 rewrites legacy rows.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from trex_core.domain.slot.state import (
|
|
ALLOWED_TRANSITIONS,
|
|
SLOT_ADVISORY_LOCK_ID,
|
|
SlotState,
|
|
is_allowed_transition,
|
|
slot_advisory_lock_id,
|
|
)
|
|
|
|
|
|
def test_slot_state_members():
|
|
"""dcy-020 (BAT-030): 5 canonical members — idle/closed aliases removed."""
|
|
assert {s.value for s in SlotState} == {
|
|
"inactive",
|
|
"pending_buy",
|
|
"holding",
|
|
"pending_sell",
|
|
"error",
|
|
}
|
|
|
|
|
|
def test_legacy_alias_values_rejected():
|
|
"""`idle`/`closed` are no longer valid SlotState values (migration 010
|
|
rewrites any remaining DB rows before this code reads them)."""
|
|
with pytest.raises(ValueError):
|
|
SlotState("idle")
|
|
with pytest.raises(ValueError):
|
|
SlotState("closed")
|
|
|
|
|
|
def test_advisory_lock_id_is_stable_constant():
|
|
"""SLOT_ADVISORY_LOCK_ID base must not change between releases (REQ-002 NFR-1)."""
|
|
assert isinstance(SLOT_ADVISORY_LOCK_ID, int)
|
|
assert SLOT_ADVISORY_LOCK_ID == 0x5107_4D31
|
|
|
|
|
|
def test_slot_advisory_lock_id_is_instance_scoped():
|
|
"""dcy-020/REQ-057: distinct instance_key -> distinct lock id; same -> same."""
|
|
a = slot_advisory_lock_id("inst-a")
|
|
b = slot_advisory_lock_id("inst-b")
|
|
assert a != b
|
|
assert slot_advisory_lock_id("inst-a") == a
|
|
# 'default' keeps a stable derived id
|
|
assert slot_advisory_lock_id("default") == slot_advisory_lock_id("default")
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"from_state,to_state",
|
|
[
|
|
# canonical lifecycle
|
|
(SlotState.INACTIVE, SlotState.PENDING_BUY),
|
|
(SlotState.PENDING_BUY, SlotState.HOLDING),
|
|
(SlotState.PENDING_BUY, SlotState.INACTIVE),
|
|
(SlotState.HOLDING, SlotState.PENDING_BUY), # dcy-021 increase
|
|
(SlotState.HOLDING, SlotState.PENDING_SELL),
|
|
(SlotState.PENDING_SELL, SlotState.INACTIVE),
|
|
(SlotState.PENDING_SELL, SlotState.HOLDING),
|
|
# to error (every non-error)
|
|
(SlotState.INACTIVE, SlotState.ERROR),
|
|
(SlotState.PENDING_BUY, SlotState.ERROR),
|
|
(SlotState.HOLDING, SlotState.ERROR),
|
|
(SlotState.PENDING_SELL, SlotState.ERROR),
|
|
# operator resolve out of error
|
|
(SlotState.ERROR, SlotState.INACTIVE),
|
|
(SlotState.ERROR, SlotState.HOLDING),
|
|
],
|
|
)
|
|
def test_allowed_transition(from_state: SlotState, to_state: SlotState):
|
|
assert is_allowed_transition(from_state, to_state)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"from_state,to_state",
|
|
[
|
|
# skipping intermediate states
|
|
(SlotState.INACTIVE, SlotState.HOLDING),
|
|
(SlotState.INACTIVE, SlotState.PENDING_SELL),
|
|
(SlotState.HOLDING, SlotState.INACTIVE),
|
|
(SlotState.PENDING_BUY, SlotState.PENDING_SELL),
|
|
(SlotState.PENDING_SELL, SlotState.PENDING_BUY),
|
|
# error has restricted exits — only inactive and holding
|
|
(SlotState.ERROR, SlotState.PENDING_BUY),
|
|
(SlotState.ERROR, SlotState.PENDING_SELL),
|
|
(SlotState.ERROR, SlotState.ERROR),
|
|
],
|
|
)
|
|
def test_disallowed_transition(from_state: SlotState, to_state: SlotState):
|
|
assert not is_allowed_transition(from_state, to_state)
|
|
|
|
|
|
def test_self_loops_are_not_allowed():
|
|
"""No state may transition to itself."""
|
|
for s in SlotState:
|
|
assert not is_allowed_transition(s, s)
|
|
|
|
|
|
def test_allowed_transitions_set_size():
|
|
"""7 base (incl. dcy-021 holding→pending_buy) + 4 to-error + 2 operator-resolve = 13."""
|
|
assert len(ALLOWED_TRANSITIONS) == 13
|