feat(phase2): BRACKET orders + Kevin risk caps (Tasks 18, 19)
Some checks failed
ci/woodpecker/push/woodpecker Pipeline was canceled

Task 18 — OrderRequest + AlpacaBroker BRACKET support:
- OrderRequest gains order_class ("simple" | "bracket"),
  take_profit_price, stop_loss_price + model_validator that requires
  both legs when order_class == "bracket".
- AlpacaBroker._build_order_request branches to a MarketOrderRequest
  with OrderClass.BRACKET + TakeProfitRequest + StopLossRequest legs,
  TimeInForce.GTC so the bracket survives day boundaries.

Task 19 — RiskManager Kevin caps + circuit-breaker:
- TradeExecutorConfig gains 4 fields: kevin_daily_trade_cap,
  kevin_daily_alloc_cap_usd, kevin_equity_drawdown_halt_pct,
  kevin_daily_loss_circuit_pct.
- check_risk() applies the caps only when
  signal.strategy_id == KEVIN_STRATEGY_UUID; non-Kevin signals pass
  through the existing path unchanged.
- 4 new checks in order: drawdown halt (sets permanent
  trading:paused), daily-loss circuit (setex 24h), daily trade-count
  cap, daily allocation cap (rolling today's $ + this trade's
  notional).
- Counter keys: kevin:daily_trades:YYYY-MM-DD,
  kevin:daily_alloc_usd:YYYY-MM-DD, kevin:daily_pnl_usd:YYYY-MM-DD,
  kevin:starting_equity_usd. All read-only here; bridge + executor
  write them.

Tests: 5 bracket + 9 kevin-caps + 28 regression-safe. Total 67 + 14
new = 81 passing (excluding -m integration). No DB needed.
This commit is contained in:
Viktor Barzin 2026-05-26 21:03:59 +00:00
parent fdc2a60257
commit f7ca671bf3
8 changed files with 458 additions and 2 deletions

View file

@ -0,0 +1,248 @@
"""Tests for the Kevin-specific risk caps (Task 19).
Daily trade-count cap, daily allocation cap, drawdown halt, daily-loss
circuit breaker. The Kevin caps are namespaced inside `check_risk` so
they only apply when the incoming `TradeSignal.strategy_id ==
KEVIN_STRATEGY_UUID`.
"""
from __future__ import annotations
from datetime import datetime, timezone
from decimal import Decimal
from unittest.mock import AsyncMock, patch
import pytest
from services.trade_executor.config import TradeExecutorConfig
from services.trade_executor.risk_manager import RiskManager
from shared.constants.kevin import KEVIN_STRATEGY_UUID
from shared.schemas.trading import (
AccountInfo,
PositionInfo,
SignalDirection,
TradeSignal,
)
# ---------------------------------------------------------------------------
# Helpers
# ---------------------------------------------------------------------------
def _kevin_config(**overrides) -> TradeExecutorConfig:
defaults = dict(
max_position_pct=0.05,
max_total_exposure_pct=0.80,
max_positions=20,
default_stop_loss_pct=0.03,
cooldown_minutes=30,
alpaca_api_key="test",
alpaca_secret_key="test",
paper_trading=True,
kevin_daily_trade_cap=5,
kevin_daily_alloc_cap_usd=10000.0,
kevin_equity_drawdown_halt_pct=0.20,
kevin_daily_loss_circuit_pct=0.05,
)
defaults.update(overrides)
return TradeExecutorConfig(**defaults)
def _kevin_signal(target_dollars: float = 2000.0) -> TradeSignal:
return TradeSignal(
ticker="NVDA",
direction=SignalDirection.LONG,
strength=0.8,
strategy_sources=["meet_kevin"],
sentiment_context={"current_price": 150.0},
timestamp=datetime.now(timezone.utc),
strategy_id=KEVIN_STRATEGY_UUID,
target_dollars=Decimal(str(target_dollars)),
)
def _non_kevin_signal() -> TradeSignal:
return TradeSignal(
ticker="AAPL",
direction=SignalDirection.LONG,
strength=0.8,
strategy_sources=["news_sentiment"],
sentiment_context={"current_price": 150.0},
timestamp=datetime.now(timezone.utc),
# no strategy_id → not Kevin
)
def _mock_broker(equity: float = 100_000.0) -> AsyncMock:
broker = AsyncMock()
broker.get_positions = AsyncMock(return_value=[])
broker.get_account = AsyncMock(
return_value=AccountInfo(
equity=equity, cash=equity, buying_power=equity * 2, portfolio_value=equity
)
)
return broker
def _redis_mock(
daily_trades: int = 0,
daily_alloc: float = 0.0,
starting_equity: float | None = 100_000.0,
daily_pnl_usd: float = 0.0,
paused: bool = False,
) -> AsyncMock:
"""Returns an AsyncMock Redis that answers the specific keys the
RiskManager queries in the Kevin caps path.
"""
redis = AsyncMock()
async def _get(key):
if isinstance(key, bytes):
key = key.decode()
# global pause
if key == "trading:paused":
return b"1" if paused else None
# kevin counters (today)
if key.startswith("kevin:daily_trades:"):
return str(daily_trades).encode()
if key.startswith("kevin:daily_alloc_usd:"):
return str(daily_alloc).encode()
if key == "kevin:starting_equity_usd":
return None if starting_equity is None else str(starting_equity).encode()
if key.startswith("kevin:daily_pnl_usd:"):
return str(daily_pnl_usd).encode()
return None
redis.get = AsyncMock(side_effect=_get)
redis.set = AsyncMock()
redis.setex = AsyncMock()
return redis
# ---------------------------------------------------------------------------
# Tests — daily trade-count cap
# ---------------------------------------------------------------------------
class TestKevinDailyTradeCap:
@pytest.mark.asyncio
async def test_below_cap_passes(self):
config = _kevin_config(kevin_daily_trade_cap=5)
broker = _mock_broker()
rm = RiskManager(config, broker, redis=_redis_mock(daily_trades=3))
with patch.object(RiskManager, "_is_market_hours", return_value=True):
approved, _ = await rm.check_risk(_kevin_signal())
assert approved is True
@pytest.mark.asyncio
async def test_at_cap_rejects(self):
config = _kevin_config(kevin_daily_trade_cap=5)
broker = _mock_broker()
rm = RiskManager(config, broker, redis=_redis_mock(daily_trades=5))
with patch.object(RiskManager, "_is_market_hours", return_value=True):
approved, reason = await rm.check_risk(_kevin_signal())
assert approved is False
assert reason == "kevin_daily_trade_cap"
@pytest.mark.asyncio
async def test_cap_does_not_apply_to_non_kevin_signal(self):
config = _kevin_config(kevin_daily_trade_cap=5)
broker = _mock_broker()
rm = RiskManager(config, broker, redis=_redis_mock(daily_trades=999))
with patch.object(RiskManager, "_is_market_hours", return_value=True):
approved, _ = await rm.check_risk(_non_kevin_signal())
assert approved is True
# ---------------------------------------------------------------------------
# Tests — daily allocation cap
# ---------------------------------------------------------------------------
class TestKevinDailyAllocCap:
@pytest.mark.asyncio
async def test_within_alloc_cap_passes(self):
config = _kevin_config(kevin_daily_alloc_cap_usd=10_000)
broker = _mock_broker()
rm = RiskManager(config, broker, redis=_redis_mock(daily_alloc=6000))
with patch.object(RiskManager, "_is_market_hours", return_value=True):
approved, _ = await rm.check_risk(_kevin_signal(target_dollars=2000))
assert approved is True
@pytest.mark.asyncio
async def test_alloc_cap_rejects(self):
config = _kevin_config(kevin_daily_alloc_cap_usd=10_000)
broker = _mock_broker()
rm = RiskManager(config, broker, redis=_redis_mock(daily_alloc=9000))
with patch.object(RiskManager, "_is_market_hours", return_value=True):
# 9000 + 2000 = 11000 > 10000
approved, reason = await rm.check_risk(_kevin_signal(target_dollars=2000))
assert approved is False
assert reason == "kevin_daily_alloc_cap"
# ---------------------------------------------------------------------------
# Tests — drawdown halt (permanent pause)
# ---------------------------------------------------------------------------
class TestKevinDrawdownHalt:
@pytest.mark.asyncio
async def test_drawdown_below_threshold_sets_permanent_pause(self):
config = _kevin_config(kevin_equity_drawdown_halt_pct=0.20)
broker = _mock_broker(equity=70_000) # 30% drawdown from 100k
redis = _redis_mock(starting_equity=100_000)
rm = RiskManager(config, broker, redis=redis)
with patch.object(RiskManager, "_is_market_hours", return_value=True):
approved, reason = await rm.check_risk(_kevin_signal())
assert approved is False
assert reason == "kevin_drawdown_halt"
# permanent pause was set (no TTL)
redis.set.assert_any_call("trading:paused", "1")
@pytest.mark.asyncio
async def test_drawdown_above_threshold_passes(self):
config = _kevin_config(kevin_equity_drawdown_halt_pct=0.20)
broker = _mock_broker(equity=85_000) # 15% drawdown → above threshold
rm = RiskManager(
config, broker, redis=_redis_mock(starting_equity=100_000)
)
with patch.object(RiskManager, "_is_market_hours", return_value=True):
approved, _ = await rm.check_risk(_kevin_signal())
assert approved is True
# ---------------------------------------------------------------------------
# Tests — daily-loss circuit breaker (24h pause)
# ---------------------------------------------------------------------------
class TestKevinDailyLossCircuit:
@pytest.mark.asyncio
async def test_daily_loss_exceeds_circuit_sets_24h_pause(self):
config = _kevin_config(kevin_daily_loss_circuit_pct=0.05)
broker = _mock_broker(equity=100_000)
# -6% daily loss on 100k equity = -6000
redis = _redis_mock(daily_pnl_usd=-6000)
rm = RiskManager(config, broker, redis=redis)
with patch.object(RiskManager, "_is_market_hours", return_value=True):
approved, reason = await rm.check_risk(_kevin_signal())
assert approved is False
assert reason == "kevin_daily_loss_circuit"
# 24h pause set via setex
assert redis.setex.called
# called with ttl ~= 86400
args, _kw = redis.setex.call_args
assert args[0] == "trading:paused"
assert int(args[1]) == 86400
@pytest.mark.asyncio
async def test_daily_loss_within_circuit_passes(self):
config = _kevin_config(kevin_daily_loss_circuit_pct=0.05)
broker = _mock_broker(equity=100_000)
# -3% daily loss → below circuit
rm = RiskManager(config, broker, redis=_redis_mock(daily_pnl_usd=-3000))
with patch.object(RiskManager, "_is_market_hours", return_value=True):
approved, _ = await rm.check_risk(_kevin_signal())
assert approved is True

View file

View file

@ -0,0 +1,89 @@
"""Unit tests for BRACKET order support in OrderRequest + AlpacaBroker.
No network exercises the request-building path only.
"""
from __future__ import annotations
import pytest
from alpaca.trading.enums import OrderClass as AlpacaOrderClass
from alpaca.trading.requests import MarketOrderRequest
from pydantic import ValidationError
from shared.broker.alpaca_broker import AlpacaBroker
from shared.schemas.trading import OrderRequest, OrderSide, OrderType
def _broker() -> AlpacaBroker:
return AlpacaBroker(api_key="test", secret_key="test", paper=True)
def test_order_request_defaults_to_simple_class():
o = OrderRequest(ticker="NVDA", side=OrderSide.BUY, qty=10)
assert o.order_class == "simple"
assert o.take_profit_price is None
assert o.stop_loss_price is None
def test_order_request_bracket_requires_both_legs():
with pytest.raises(ValidationError, match="bracket orders require"):
OrderRequest(
ticker="NVDA",
side=OrderSide.BUY,
qty=10,
order_class="bracket",
take_profit_price=200.0,
# stop_loss_price missing
)
with pytest.raises(ValidationError, match="bracket orders require"):
OrderRequest(
ticker="NVDA",
side=OrderSide.BUY,
qty=10,
order_class="bracket",
stop_loss_price=150.0,
# take_profit_price missing
)
def test_order_request_bracket_with_both_legs_validates():
o = OrderRequest(
ticker="NVDA",
side=OrderSide.BUY,
qty=10,
order_class="bracket",
take_profit_price=200.0,
stop_loss_price=150.0,
)
assert o.order_class == "bracket"
assert o.take_profit_price == 200.0
assert o.stop_loss_price == 150.0
def test_build_order_request_simple_market():
o = OrderRequest(ticker="NVDA", side=OrderSide.BUY, qty=10)
req = _broker()._build_order_request(o)
assert isinstance(req, MarketOrderRequest)
assert req.symbol == "NVDA"
assert req.qty == 10
# default MarketOrderRequest has order_class=None or SIMPLE
assert getattr(req, "order_class", None) in (None, AlpacaOrderClass.SIMPLE)
def test_build_order_request_bracket_attaches_legs():
o = OrderRequest(
ticker="NVDA",
side=OrderSide.BUY,
qty=10,
order_class="bracket",
take_profit_price=200.0,
stop_loss_price=150.0,
)
req = _broker()._build_order_request(o)
assert isinstance(req, MarketOrderRequest)
assert req.order_class == AlpacaOrderClass.BRACKET
assert req.take_profit is not None
assert float(req.take_profit.limit_price) == 200.0
assert req.stop_loss is not None
assert float(req.stop_loss.stop_price) == 150.0