genichinandClaude Opus 4.8
d334f34215
운영자 CLI 의 slots 조회 편의: --symbol, --slot-id 클라이언트 측 필터(--status 와 조합 가능) 추가. 두 필터 중 하나라도 주면 한 줄 요약 대신 슬롯 상세 블록 (status/symbol/portfolio_id/qty/avg_price/limit/allocated_capital/strategy_id/ sell_strategy_id/forced_close_at/universe_detached, holding 이면 watchdog)으로 출력. 필터 없으면 기존 terse 테이블 유지(무회귀). 구조/계약/런타임 흐름 영향 없는 표시 기능. tests/runtime/test_cli_operator.py 필터 4건 추가(24 passed), ruff clean. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
262 lines
10 KiB
Python
262 lines
10 KiB
Python
"""Operator CLI tests (dcy-016 / REQ-032) — RPC injected, no broker/DB needed."""
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
|
|
import pytest
|
|
|
|
from trex_core.runtime import cli
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _env(monkeypatch):
|
|
monkeypatch.setenv("TREX_POSTGRESQL_DSN", "postgresql://x/y")
|
|
monkeypatch.setenv("TREX_BROKER", "paper")
|
|
monkeypatch.delenv("TREX_KIS_ALLOW_LIVE", raising=False)
|
|
|
|
|
|
class _RpcRecorder:
|
|
def __init__(self, response):
|
|
self.response = response
|
|
self.calls = []
|
|
|
|
def __call__(self, queue, payload, *, correlation_id, timeout):
|
|
self.calls.append({"queue": queue, "payload": payload, "cid": correlation_id})
|
|
return self.response
|
|
|
|
|
|
_SNAPSHOT = {
|
|
"assets": [{"asset_id": "PF-1", "symbol": "005930", "asset_quantity": 0,
|
|
"active_strategy_id": "orb"}],
|
|
"view": "all", "available_slots": 9, "degraded": False, "reason": None,
|
|
# BAT-022: envelope per-slot state (operator-facing). `slots` queries read
|
|
# this, not the universe `assets[]`.
|
|
"slots": [
|
|
{"slot_id": "S-1", "status": "holding",
|
|
"symbol": "005930", "qty": 10.0, "avg_price": 28700.0, "entry_price": None,
|
|
"allocated_capital": None, "strategy_id": "orb", "sell_strategy_id": None,
|
|
"portfolio_id": "PF-1", "forced_close_at": None},
|
|
{"slot_id": "S-2", "status": "inactive",
|
|
"symbol": None, "qty": None, "avg_price": None, "entry_price": None,
|
|
"allocated_capital": None, "strategy_id": None, "sell_strategy_id": None,
|
|
"portfolio_id": None, "forced_close_at": None},
|
|
],
|
|
}
|
|
_ADMITTED = {"intent_id": "i", "status": "admitted", "slot_id": "S-1", "reason": None}
|
|
_REJECTED = {"intent_id": "i", "status": "rejected", "slot_id": None,
|
|
"reason": "unknown_portfolio"}
|
|
|
|
|
|
# --- universe / slots (snapshot RPC + watchdog suppression) ------------------
|
|
def test_universe_uses_snapshot_queue_with_operator_suppression(capsys):
|
|
rpc = _RpcRecorder(_SNAPSHOT)
|
|
rc = cli.main(["universe"], rpc=rpc)
|
|
assert rc == 0
|
|
call = rpc.calls[0]
|
|
assert call["queue"] == "trex.snapshot.req" # default instance
|
|
assert call["payload"]["actor"] == "operator"
|
|
assert call["payload"]["touch_watchdog"] is False # FR-2a / R-4
|
|
assert "PF-1" in capsys.readouterr().out
|
|
|
|
|
|
def test_universe_json_is_scoped_to_universe(capsys):
|
|
"""BAT-022: `universe --json` emits universe data only — `slots[]`/`watchdog[]`
|
|
(operator slot table) are NOT leaked into the universe view."""
|
|
rpc = _RpcRecorder(_SNAPSHOT)
|
|
cli.main(["universe", "--json"], rpc=rpc)
|
|
out = json.loads(capsys.readouterr().out)
|
|
assert set(out.keys()) == {"instance", "view", "assets", "available_slots",
|
|
"degraded", "reason"}
|
|
assert "slots" not in out and "watchdog" not in out
|
|
|
|
|
|
def test_slots_shows_real_slot_state_not_universe(capsys):
|
|
"""BAT-022: `slots` shows envelope `slots[]` (slot table) with status,
|
|
not the universe `assets[]`."""
|
|
rpc = _RpcRecorder(_SNAPSHOT)
|
|
cli.main(["slots"], rpc=rpc)
|
|
out = capsys.readouterr().out
|
|
# request stays non-invasive (operator suppression)
|
|
assert rpc.calls[0]["payload"]["actor"] == "operator"
|
|
assert rpc.calls[0]["payload"]["touch_watchdog"] is False
|
|
# both slots shown, with status column (slot state, not universe assets)
|
|
assert "count=2" in out
|
|
assert "S-1\tholding" in out and "S-2\tinactive" in out
|
|
|
|
|
|
def test_slots_status_filters_by_slot_status(capsys):
|
|
"""BAT-022: `--status` filters by slot status client-side (no view mapping)."""
|
|
rpc = _RpcRecorder(_SNAPSHOT)
|
|
cli.main(["slots", "--status", "holding"], rpc=rpc)
|
|
out = capsys.readouterr().out
|
|
assert "count=1" in out and "S-1\tholding" in out and "S-2" not in out
|
|
|
|
|
|
def test_slots_symbol_filters_to_one_symbol(capsys):
|
|
"""`--symbol` shows only the slot(s) bound to that symbol (client-side filter)."""
|
|
rpc = _RpcRecorder(_SNAPSHOT)
|
|
cli.main(["slots", "--symbol", "005930"], rpc=rpc)
|
|
out = capsys.readouterr().out
|
|
assert "count=1" in out and "S-1" in out and "S-2" not in out
|
|
|
|
|
|
def test_slots_symbol_combines_with_status(capsys):
|
|
"""`--symbol` composes with `--status` (both must match)."""
|
|
rpc = _RpcRecorder(_SNAPSHOT)
|
|
cli.main(["slots", "--status", "holding", "--symbol", "999999"], rpc=rpc)
|
|
out = capsys.readouterr().out
|
|
assert "count=0" in out # S-1 is holding but symbol != 999999
|
|
|
|
|
|
def test_slots_slot_id_filters_to_one_slot(capsys):
|
|
"""`--slot-id` shows only the matching slot, in the detailed block."""
|
|
rpc = _RpcRecorder(_SNAPSHOT)
|
|
cli.main(["slots", "--slot-id", "S-2"], rpc=rpc)
|
|
out = capsys.readouterr().out
|
|
assert "count=1" in out and "slot S-2" in out and "S-1" not in out
|
|
# detailed block fields present
|
|
assert "portfolio_id" in out and "universe_detached" in out
|
|
|
|
|
|
def test_slots_symbol_detailed_output(capsys):
|
|
"""`--symbol` switches to the detailed per-slot block (not the terse table)."""
|
|
rpc = _RpcRecorder(_SNAPSHOT)
|
|
cli.main(["slots", "--symbol", "005930"], rpc=rpc)
|
|
out = capsys.readouterr().out
|
|
assert "slot S-1" in out and "status" in out and "avg_price" in out
|
|
|
|
|
|
def test_slots_marks_watchdog_owned(capsys):
|
|
"""dcy-017: slots flags strategy-absent (watchdog-owned) positions."""
|
|
resp = {
|
|
"assets": [], "view": "all", "available_slots": 9, "degraded": False,
|
|
"reason": None,
|
|
"slots": [{"slot_id": "S-1",
|
|
"status": "holding", "symbol": "005930", "qty": 10.0,
|
|
"avg_price": 28700.0, "entry_price": None,
|
|
"allocated_capital": None, "strategy_id": "orb",
|
|
"sell_strategy_id": None, "portfolio_id": "PF-1",
|
|
"forced_close_at": None}],
|
|
"watchdog": [{"slot_id": "S-1", "portfolio_id": "PF-1", "symbol": "005930",
|
|
"expired": True, "last_seen": None}],
|
|
}
|
|
cli.main(["slots", "--status", "held"], rpc=_RpcRecorder(resp))
|
|
out = capsys.readouterr().out
|
|
assert "watchdog_owned=1" in out and "WATCHDOG-OWNED" in out
|
|
|
|
|
|
def test_instance_shortcut_routes_to_instance_queue():
|
|
rpc = _RpcRecorder(_SNAPSHOT)
|
|
cli.main(["universe", "--instance", "inst-2"], rpc=rpc)
|
|
assert rpc.calls[0]["queue"] == "trex.snapshot.req.inst-2"
|
|
|
|
|
|
def test_json_output_is_valid_json(capsys):
|
|
rpc = _RpcRecorder(_SNAPSHOT)
|
|
cli.main(["universe", "--json"], rpc=rpc)
|
|
parsed = json.loads(capsys.readouterr().out)
|
|
assert parsed["assets"][0]["asset_id"] == "PF-1"
|
|
|
|
|
|
# --- enter (buy intent RPC) + safeguards ------------------------------------
|
|
def test_enter_aborts_without_confirmation_no_rpc():
|
|
rpc = _RpcRecorder(_ADMITTED)
|
|
rc = cli.main(["enter", "PF-1", "--capital", "100000"],
|
|
rpc=rpc, confirm=lambda _p: False)
|
|
assert rc == 1 and rpc.calls == [] # aborted -> no order
|
|
|
|
|
|
def test_enter_with_yes_submits_buy_intent():
|
|
rpc = _RpcRecorder(_ADMITTED)
|
|
rc = cli.main(["enter", "PF-1", "--capital", "100000", "--yes"], rpc=rpc)
|
|
assert rc == 0
|
|
p = rpc.calls[0]["payload"]
|
|
assert rpc.calls[0]["queue"] == "trex.intent.req"
|
|
assert p["intent_type"] == "buy" and p["portfolio_id"] == "PF-1"
|
|
assert p["allocated_capital"] == 100000
|
|
assert "symbol" not in p # server-resolved (dcy-011)
|
|
|
|
|
|
def test_enter_with_forced_close_at_maps_to_payload():
|
|
# REQ-032 AC-5: --forced-close-at (absolute) → buy intent forced_close_at.
|
|
rpc = _RpcRecorder(_ADMITTED)
|
|
rc = cli.main(["enter", "PF-1", "--capital", "100000", "--yes",
|
|
"--forced-close-at", "2026-06-09T15:20:00+09:00"], rpc=rpc)
|
|
assert rc == 0
|
|
p = rpc.calls[0]["payload"]
|
|
assert p["forced_close_at"] == "2026-06-09T15:20:00+09:00"
|
|
|
|
|
|
def test_enter_without_forced_close_at_omits_field():
|
|
# REQ-032 AC-6: unspecified → field absent (backward compat).
|
|
rpc = _RpcRecorder(_ADMITTED)
|
|
rc = cli.main(["enter", "PF-1", "--capital", "100000", "--yes"], rpc=rpc)
|
|
assert rc == 0
|
|
assert "forced_close_at" not in rpc.calls[0]["payload"]
|
|
|
|
|
|
def test_enter_rejects_malformed_forced_close_at_before_submit():
|
|
# REQ-032 AC-6: invalid timestamp rejected at CLI, no RPC submitted.
|
|
rpc = _RpcRecorder(_ADMITTED)
|
|
rc = cli.main(["enter", "PF-1", "--capital", "100000", "--yes",
|
|
"--forced-close-at", "notatime"], rpc=rpc)
|
|
assert rc == 1 and rpc.calls == []
|
|
|
|
|
|
def test_enter_rejected_returns_nonzero():
|
|
rpc = _RpcRecorder(_REJECTED)
|
|
rc = cli.main(["enter", "PF-x", "--yes"], rpc=rpc)
|
|
assert rc == 1
|
|
|
|
|
|
def test_enter_dry_run_does_not_submit(capsys):
|
|
rpc = _RpcRecorder(_ADMITTED)
|
|
rc = cli.main(["enter", "PF-1", "--capital", "5", "--dry-run"], rpc=rpc)
|
|
assert rc == 0 and rpc.calls == []
|
|
assert "dry-run" in capsys.readouterr().out
|
|
|
|
|
|
def test_enter_json_requires_yes():
|
|
rpc = _RpcRecorder(_ADMITTED)
|
|
rc = cli.main(["enter", "PF-1", "--json"], rpc=rpc)
|
|
assert rc == 2 and rpc.calls == [] # no auto-submit (FR-7)
|
|
|
|
|
|
# --- exit (sell intent RPC) -------------------------------------------------
|
|
def test_exit_with_yes_submits_sell_with_limit_price():
|
|
rpc = _RpcRecorder(_ADMITTED)
|
|
cli.main(["exit", "S-1", "--price", "70000", "--yes"], rpc=rpc)
|
|
p = rpc.calls[0]["payload"]
|
|
assert p["intent_type"] == "sell" and p["slot_id"] == "S-1"
|
|
assert p["limit_price"] == 70000
|
|
|
|
|
|
def test_exit_market_when_no_price():
|
|
rpc = _RpcRecorder(_ADMITTED)
|
|
cli.main(["exit", "S-1", "--yes"], rpc=rpc)
|
|
assert rpc.calls[0]["payload"]["limit_price"] is None
|
|
|
|
|
|
# --- live guard (OQ-2 double ack) -------------------------------------------
|
|
def test_live_requires_live_flag(monkeypatch):
|
|
monkeypatch.setattr(cli, "_broker_mode", lambda cfg: ("kis", True))
|
|
rpc = _RpcRecorder(_ADMITTED)
|
|
rc = cli.main(["enter", "PF-1", "--yes"], rpc=rpc) # --yes but no --live
|
|
assert rc == 2 and rpc.calls == []
|
|
|
|
|
|
def test_live_proceeds_with_yes_live(monkeypatch):
|
|
monkeypatch.setattr(cli, "_broker_mode", lambda cfg: ("kis", True))
|
|
rpc = _RpcRecorder(_ADMITTED)
|
|
rc = cli.main(["enter", "PF-1", "--yes", "--live"], rpc=rpc)
|
|
assert rc == 0 and len(rpc.calls) == 1
|
|
|
|
|
|
# --- transport failure surfaces as a clear error ----------------------------
|
|
def test_rpc_failure_surfaces_error(capsys):
|
|
def boom(*a, **k):
|
|
raise RuntimeError("daemon unreachable")
|
|
rc = cli.main(["universe"], rpc=boom)
|
|
assert rc == 3
|
|
assert "daemon unreachable" in capsys.readouterr().err
|