| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917 |
- from __future__ import annotations
- from pathlib import Path
- from tempfile import TemporaryDirectory
- from fastapi.testclient import TestClient
- from src.trader_mcp import strategy_registry, strategy_store
- from src.trader_mcp.server import app
- from src.trader_mcp.strategy_context import StrategyContext
- from src.trader_mcp.strategy_sdk import Strategy as BaseStrategy
- from strategies.exposure_protector import Strategy as ExposureStrategy
- from strategies.grid_trader import Strategy as GridStrategy
- from strategies.trend_follower import Strategy as TrendStrategy
- STRATEGY_CODE = '''
- from src.trader_mcp.strategy_sdk import Strategy
- class Strategy(Strategy):
- def init(self):
- return {"started": True, "config_copy": dict(self.config)}
- '''
- def test_strategies_endpoints_roundtrip():
- with TemporaryDirectory() as tmpdir:
- strategy_store.DB_PATH = Path(tmpdir) / "trader_mcp.sqlite3"
- from src.trader_mcp import strategy_registry
- strategy_registry.STRATEGIES_DIR = Path(tmpdir) / "strategies"
- strategy_registry.STRATEGIES_DIR.mkdir()
- (strategy_registry.STRATEGIES_DIR / "demo.py").write_text(STRATEGY_CODE)
- client = TestClient(app)
- r = client.get("/strategies")
- assert r.status_code == 200
- body = r.json()
- assert "available" in body
- assert "configured" in body
- r = client.post(
- "/strategies",
- json={
- "id": "demo-1",
- "strategy_type": "demo",
- "account_id": "acct-1",
- "client_id": "strategy:test",
- "mode": "observe",
- "config": {"risk": 0.01},
- },
- )
- assert r.status_code == 200
- assert r.json()["id"] == "demo-1"
- r = client.get("/strategies")
- assert any(item["id"] == "demo-1" for item in r.json()["configured"])
- r = client.delete("/strategies/demo-1")
- assert r.status_code == 200
- assert r.json()["ok"] is True
- def test_strategy_context_binds_identity(monkeypatch):
- calls = {}
- def fake_place_order(arguments):
- calls["place_order"] = arguments
- return {"ok": True}
- def fake_open_orders(account_id, client_id=None):
- calls["open_orders"] = {"account_id": account_id, "client_id": client_id}
- return {"ok": True}
- def fake_cancel_all(account_id, client_id=None):
- calls["cancel_all"] = {"account_id": account_id, "client_id": client_id}
- return {"ok": True}
- monkeypatch.setattr("src.trader_mcp.strategy_context.place_order", fake_place_order)
- monkeypatch.setattr("src.trader_mcp.strategy_context.list_open_orders", fake_open_orders)
- monkeypatch.setattr("src.trader_mcp.strategy_context.cancel_all_orders", fake_cancel_all)
- ctx = StrategyContext(id="inst-1", account_id="acct-1", client_id="client-1", mode="active")
- ctx.place_order(side="sell", market="xrpusd", order_type="limit", amount="10", price="2")
- ctx.get_open_orders()
- ctx.cancel_all_orders()
- assert calls["place_order"]["account_id"] == "acct-1"
- assert calls["place_order"]["client_id"] == "client-1"
- assert calls["open_orders"] == {"account_id": "acct-1", "client_id": "client-1"}
- assert calls["cancel_all"] == {"account_id": "acct-1", "client_id": "client-1"}
- def test_stop_loss_strategy_loads_with_aligned_regime_config(tmp_path):
- original_db = strategy_store.DB_PATH
- original_dir = strategy_registry.STRATEGIES_DIR
- try:
- strategy_store.DB_PATH = tmp_path / "trader_mcp.sqlite3"
- strategy_registry.STRATEGIES_DIR = tmp_path / "strategies"
- strategy_registry.STRATEGIES_DIR.mkdir()
- (strategy_registry.STRATEGIES_DIR / "grid_trader.py").write_text((Path(__file__).resolve().parents[1] / "strategies" / "grid_trader.py").read_text())
- (strategy_registry.STRATEGIES_DIR / "exposure_protector.py").write_text((Path(__file__).resolve().parents[1] / "strategies" / "exposure_protector.py").read_text())
- grid_defaults = strategy_registry.get_strategy_default_config("grid_trader")
- stop_defaults = strategy_registry.get_strategy_default_config("exposure_protector")
- assert grid_defaults["trade_sides"] == "both"
- assert grid_defaults["grid_step_pct"] == 0.012
- assert stop_defaults["trail_distance_pct"] == 0.03
- assert stop_defaults["rebalance_target_ratio"] == 0.5
- assert stop_defaults["min_rebalance_seconds"] == 180
- assert stop_defaults["min_price_move_pct"] == 0.005
- finally:
- strategy_store.DB_PATH = original_db
- strategy_registry.STRATEGIES_DIR = original_dir
- def test_grid_supervision_reports_factual_capacity_not_handoff_commands():
- class FakeContext:
- account_id = "acct-1"
- market_symbol = "xrpusd"
- base_currency = "XRP"
- counter_currency = "USD"
- mode = "active"
- strategy = GridStrategy(FakeContext(), {})
- strategy.state.update({
- "last_price": 1.45,
- "base_available": 50.0,
- "counter_available": 38.7,
- "regimes": {"1h": {"trend": {"state": "bull"}}},
- })
- supervision = strategy._supervision()
- assert supervision["inventory_pressure"] == "base_heavy"
- assert supervision["capacity_available"] is False
- assert supervision["side_capacity"] == {"buy": True, "sell": True}
- strategy.state.update({
- "base_available": 88.0,
- "counter_available": 4.0,
- })
- supervision = strategy._supervision()
- assert supervision["inventory_pressure"] == "base_side_depleted"
- assert supervision["side_capacity"] == {"buy": True, "sell": False}
- def test_grid_supervision_exposes_adverse_side_open_orders():
- class FakeContext:
- account_id = "acct-1"
- market_symbol = "xrpusd"
- base_currency = "XRP"
- counter_currency = "USD"
- mode = "active"
- strategy = GridStrategy(FakeContext(), {})
- strategy.state.update({
- "last_price": 1.60,
- "center_price": 1.45,
- "orders": [
- {"side": "sell", "price": "1.62", "amount": "10", "status": "open"},
- {"side": "sell", "price": "1.66", "amount": "5", "status": "open"},
- {"side": "buy", "price": "1.38", "amount": "7", "status": "open"},
- ],
- })
- supervision = strategy._supervision()
- assert supervision["market_bias"] == "bullish"
- assert supervision["adverse_side"] == "sell"
- assert supervision["adverse_side_open_order_count"] == 2
- assert supervision["adverse_side_open_order_notional_quote"] > 0
- assert "sell ladder exposed" in " ".join(supervision["concerns"])
- def test_trend_and_protector_supervision_reports_facts_only():
- class FakeContext:
- account_id = "acct-1"
- market_symbol = "xrpusd"
- base_currency = "XRP"
- counter_currency = "USD"
- mode = "active"
- trend = TrendStrategy(FakeContext(), {"trade_side": "buy", "order_notional_quote": 1.0})
- trend.state.update({"last_price": 1.45, "base_available": 20.0, "counter_available": 20.0, "last_order_at": 0.0})
- trend_supervision = trend._supervision()
- assert trend_supervision["trade_side"] == "buy"
- assert trend_supervision["capacity_available"] is True
- assert trend_supervision["entry_offset_pct"] == 0.003
- assert trend_supervision["chasing_risk"] in {"low", "moderate", "elevated"}
- assert "switch_readiness" not in trend_supervision
- assert "desired_companion" not in trend_supervision
- protector = ExposureStrategy(FakeContext(), {})
- protector.state.update({"last_price": 1.45, "base_available": 40.0, "counter_available": 10.0})
- protector_supervision = protector._supervision()
- assert protector_supervision["rebalance_needed"] is True
- assert protector_supervision["repair_progress"] <= 1.0
- assert "switch_readiness" not in protector_supervision
- assert "desired_companion" not in protector_supervision
- def test_exposure_protector_holds_inside_hysteresis_band(monkeypatch):
- class FakeContext:
- account_id = "acct-1"
- market_symbol = "xrpusd"
- base_currency = "XRP"
- counter_currency = "USD"
- mode = "active"
- def __init__(self):
- self.placed_orders = []
- def get_account_info(self):
- return {"balances": [{"asset_code": "XRP", "available": 9.2}, {"asset_code": "USD", "available": 10.0}]}
- def get_price(self, market):
- return {"price": 1.0}
- def get_fee_rates(self, market):
- return {"maker": 0.0, "taker": 0.004}
- def place_order(self, **kwargs):
- self.placed_orders.append(kwargs)
- return {"status": "ok", "id": f"oid-{len(self.placed_orders)}"}
- ctx = FakeContext()
- strategy = ExposureStrategy(ctx, {"rebalance_target_ratio": 0.5, "rebalance_step_ratio": 0.15, "balance_tolerance": 0.05, "cooldown_ticks": 0, "min_rebalance_seconds": 0, "trail_distance_pct": 0.03})
- strategy.state["last_rebalance_side"] = "sell"
- strategy.state["last_order_at"] = 0
- monkeypatch.setattr(strategy, "_refresh_regimes", lambda: None)
- result = strategy.on_tick({})
- assert result["action"] == "hold"
- assert result["reason"] == "within rebalance hysteresis"
- assert ctx.placed_orders == []
- def test_grid_apply_policy_keeps_explicit_grid_levels():
- class FakeContext:
- account_id = "acct-1"
- market_symbol = "xrpusd"
- base_currency = "XRP"
- counter_currency = "USD"
- mode = "active"
- strategy = GridStrategy(FakeContext(), {"grid_levels": 5, "policy": {"risk_posture": "normal"}})
- strategy.apply_policy()
- assert strategy.config["grid_levels"] == 5
- assert strategy.state["policy_derived"]["grid_levels"] == 5
- def test_grid_seed_keeps_other_side_when_one_side_fails(monkeypatch):
- class FakeContext:
- base_currency = "XRP"
- counter_currency = "USD"
- market_symbol = "xrpusd"
- minimum_order_value = 10.0
- mode = "active"
- def __init__(self):
- self.attempts = []
- self.buy_attempts = 0
- self.sell_attempts = 0
- def get_fee_rates(self, market):
- return {"maker": 0.0, "taker": 0.0}
- def suggest_order_amount(self, **kwargs):
- return 10.0
- def place_order(self, **kwargs):
- self.attempts.append(kwargs)
- if kwargs["side"] == "buy":
- self.buy_attempts += 1
- if self.buy_attempts == 3:
- raise RuntimeError("insufficient USD")
- elif kwargs["side"] == "sell":
- self.sell_attempts += 1
- return {"status": "ok", "id": f"{kwargs['side']}-{len(self.attempts)}"}
- ctx = FakeContext()
- strategy = GridStrategy(ctx, {"grid_levels": 5, "order_call_delay_ms": 0, "enable_trend_guard": False, "fee_rate": 0.0})
- strategy.state["center_price"] = 100.0
- monkeypatch.setattr(strategy, "_supported_levels", lambda side, center, min_notional: 5)
- monkeypatch.setattr(strategy, "_refresh_balance_snapshot", lambda: None)
- strategy._place_grid(100.0)
- orders = strategy.state["orders"]
- assert ctx.buy_attempts == 5
- assert ctx.sell_attempts == 5
- assert len([o for o in orders if o["side"] == "buy"]) == 4
- assert len([o for o in orders if o["side"] == "sell"]) == 5
- assert any("partial success" in line for line in (strategy.state.get("debug_log") or [])) or strategy.state.get("last_error") == "insufficient USD"
- def test_grid_skips_rebuild_when_balance_refresh_fails(monkeypatch):
- class FakeContext:
- base_currency = "XRP"
- counter_currency = "USD"
- market_symbol = "xrpusd"
- minimum_order_value = 10.0
- mode = "active"
- def __init__(self):
- self.cancelled_all = 0
- self.placed_orders = []
- def get_fee_rates(self, market):
- return {"maker": 0.0, "taker": 0.004}
- def get_account_info(self):
- raise RuntimeError("Bitstamp auth breaker active, retry later")
- def cancel_all_orders(self):
- self.cancelled_all += 1
- return {"ok": True}
- def suggest_order_amount(self, **kwargs):
- return 10.0
- def place_order(self, **kwargs):
- self.placed_orders.append(kwargs)
- return {"status": "ok", "id": f"oid-{len(self.placed_orders)}"}
- ctx = FakeContext()
- strategy = GridStrategy(ctx, {"grid_levels": 5, "order_call_delay_ms": 0, "enable_trend_guard": False, "fee_rate": 0.004})
- strategy.state["center_price"] = 1.4397
- strategy.state["seeded"] = True
- strategy.state["orders"] = [{"side": "buy", "price": 1.43, "amount": 10.0, "id": "o1"}]
- strategy.state["order_ids"] = ["o1"]
- monkeypatch.setattr(strategy, "_sync_open_orders_state", lambda: [{"side": "buy", "price": 1.43, "amount": 10.0, "id": "o1"}])
- monkeypatch.setattr(strategy, "_price", lambda: 1.4397)
- monkeypatch.setattr(strategy, "_refresh_regimes", lambda: None)
- result = strategy.on_tick({})
- assert result["action"] == "hold"
- assert result["reason"] == "balance refresh unavailable"
- assert ctx.cancelled_all == 0
- assert ctx.placed_orders == []
- def test_grid_skips_shape_rebuild_when_balance_reads_turn_inconclusive(monkeypatch):
- class FakeContext:
- base_currency = "XRP"
- counter_currency = "USD"
- market_symbol = "xrpusd"
- minimum_order_value = 10.0
- mode = "active"
- def __init__(self):
- self.cancelled_all = 0
- self.placed_orders = []
- self.calls = 0
- def get_fee_rates(self, market):
- return {"maker": 0.0, "taker": 0.004}
- def get_account_info(self):
- self.calls += 1
- if self.calls == 1:
- return {
- "balances": [
- {"asset_code": "USD", "available": 41.29},
- {"asset_code": "XRP", "available": 9.98954},
- ]
- }
- raise RuntimeError("Bitstamp auth breaker active, retry later")
- def cancel_all_orders(self):
- self.cancelled_all += 1
- return {"ok": True}
- def suggest_order_amount(self, **kwargs):
- return 10.0
- def place_order(self, **kwargs):
- self.placed_orders.append(kwargs)
- return {"status": "ok", "id": f"oid-{len(self.placed_orders)}"}
- ctx = FakeContext()
- strategy = GridStrategy(ctx, {"grid_levels": 2, "order_call_delay_ms": 0, "enable_trend_guard": False, "fee_rate": 0.004})
- strategy.state["center_price"] = 1.3285
- strategy.state["seeded"] = True
- strategy.state["orders"] = [
- {"side": "buy", "price": 1.3243993, "amount": 7.63, "id": "existing-buy"},
- {"side": "sell", "price": 1.3326007, "amount": 9.0, "id": "sell-1"},
- {"side": "sell", "price": 1.3367011, "amount": 9.0, "id": "sell-2"},
- ]
- strategy.state["order_ids"] = ["existing-buy", "sell-1", "sell-2"]
- def fake_sync_open_orders_state():
- live = [{"side": "sell", "price": 1.3326007, "amount": 9.0, "id": "sell-1"}]
- strategy.state["orders"] = live
- strategy.state["order_ids"] = ["sell-1"]
- strategy.state["open_order_count"] = 1
- return live
- monkeypatch.setattr(strategy, "_sync_open_orders_state", fake_sync_open_orders_state)
- monkeypatch.setattr(strategy, "_price", lambda: 1.3285)
- monkeypatch.setattr(strategy, "_refresh_regimes", lambda: None)
- monkeypatch.setattr(strategy, "_grid_refresh_paused", lambda: False)
- monkeypatch.setattr(strategy, "_recenter_threshold_pct", lambda: 0.5)
- result = strategy.on_tick({})
- assert result["action"] == "hold"
- assert ctx.cancelled_all == 0
- assert ctx.placed_orders == []
- def test_grid_missing_order_triggers_full_rebuild(monkeypatch):
- class FakeContext:
- base_currency = "XRP"
- counter_currency = "USD"
- market_symbol = "xrpusd"
- minimum_order_value = 10.0
- mode = "active"
- def __init__(self):
- self.cancelled_all = 0
- self.placed_orders = []
- def get_fee_rates(self, market):
- return {"maker": 0.0, "taker": 0.004}
- def get_account_info(self):
- return {
- "balances": [
- {"asset_code": "USD", "available": 13.55},
- {"asset_code": "XRP", "available": 22.0103},
- ]
- }
- def suggest_order_amount(
- self,
- *,
- side,
- price,
- levels,
- min_notional,
- fee_rate,
- max_notional_per_order=0.0,
- dust_collect=False,
- order_size=0.0,
- safety=0.995,
- ):
- if side == "buy":
- quote_available = 13.55
- spendable_quote = quote_available * safety
- quote_cap = min(spendable_quote, max_notional_per_order) if max_notional_per_order > 0 else spendable_quote
- if quote_cap < min_notional * (1 + fee_rate):
- return 0.0
- return quote_cap / (price * (1 + fee_rate))
- return 0.0
- def cancel_all_orders(self):
- self.cancelled_all += 1
- return {"ok": True}
- def place_order(self, **kwargs):
- self.placed_orders.append(kwargs)
- return {"status": "ok", "id": f"oid-{len(self.placed_orders)}"}
- ctx = FakeContext()
- strategy = GridStrategy(
- ctx,
- {
- "grid_levels": 2,
- "grid_step_pct": 0.0062,
- "grid_step_min_pct": 0.0033,
- "grid_step_max_pct": 0.012,
- "max_notional_per_order": 12,
- "order_call_delay_ms": 0,
- "trade_sides": "both",
- "debug_orders": True,
- "dust_collect": True,
- "enable_trend_guard": False,
- "fee_rate": 0.004,
- },
- )
- strategy.state["center_price"] = 1.3285
- strategy.state["seeded"] = True
- strategy.state["base_available"] = 22.0103
- strategy.state["counter_available"] = 13.55
- strategy.state["orders"] = [
- {"side": "buy", "price": 1.3243993, "amount": 7.63, "id": "existing-buy"},
- {"side": "sell", "price": 1.3326007, "amount": 9.0, "id": "sell-1"},
- {"side": "sell", "price": 1.3367011, "amount": 9.0, "id": "sell-2"},
- ]
- strategy.state["order_ids"] = ["existing-buy", "sell-1", "sell-2"]
- def fake_sync_open_orders_state():
- live = [{"side": "sell", "price": 1.3326007, "amount": 9.0, "id": "sell-1"}]
- strategy.state["orders"] = live
- strategy.state["order_ids"] = ["sell-1"]
- strategy.state["open_order_count"] = 1
- return live
- monkeypatch.setattr(strategy, "_sync_open_orders_state", fake_sync_open_orders_state)
- monkeypatch.setattr(strategy, "_price", lambda: 1.3285)
- monkeypatch.setattr(strategy, "_refresh_regimes", lambda: None)
- monkeypatch.setattr(strategy, "_grid_refresh_paused", lambda: False)
- monkeypatch.setattr(strategy, "_recenter_threshold_pct", lambda: 0.5)
- result = strategy.on_tick({})
- assert result["action"] in {"seed", "reseed"}
- assert ctx.cancelled_all == 1
- assert len(ctx.placed_orders) > 0
- assert strategy.state["last_action"] == "reseeded"
- def test_grid_side_imbalance_triggers_full_rebuild(monkeypatch):
- class FakeContext:
- base_currency = "XRP"
- counter_currency = "USD"
- market_symbol = "xrpusd"
- minimum_order_value = 10.0
- mode = "active"
- def __init__(self):
- self.cancelled_all = 0
- self.placed_orders = []
- def get_fee_rates(self, market):
- return {"maker": 0.0, "taker": 0.004}
- def get_account_info(self):
- return {"balances": [{"asset_code": "USD", "available": 41.29}, {"asset_code": "XRP", "available": 9.98954}]}
- def cancel_all_orders(self):
- self.cancelled_all += 1
- return {"ok": True}
- def suggest_order_amount(self, **kwargs):
- return 10.0
- def place_order(self, **kwargs):
- self.placed_orders.append(kwargs)
- return {"status": "ok", "id": f"oid-{len(self.placed_orders)}"}
- ctx = FakeContext()
- strategy = GridStrategy(ctx, {"grid_levels": 2, "order_call_delay_ms": 0, "enable_trend_guard": False, "fee_rate": 0.004})
- strategy.state["center_price"] = 1.3907
- strategy.state["seeded"] = True
- strategy.state["orders"] = [{"side": "buy", "price": 1.3800, "amount": 10.0, "id": "o1"} for _ in range(5)]
- strategy.state["order_ids"] = [f"o{i}" for i in range(5)]
- def fake_sync_open_orders_state():
- live = [{"side": "buy", "price": 1.3800, "amount": 10.0, "id": f"o{i}"} for i in range(5)]
- strategy.state["orders"] = live
- strategy.state["order_ids"] = [f"o{i}" for i in range(5)]
- strategy.state["open_order_count"] = 5
- return live
- monkeypatch.setattr(strategy, "_sync_open_orders_state", fake_sync_open_orders_state)
- monkeypatch.setattr(strategy, "_refresh_balance_snapshot", lambda: True)
- monkeypatch.setattr(strategy, "_price", lambda: 1.3915)
- monkeypatch.setattr(strategy, "_refresh_regimes", lambda: None)
- monkeypatch.setattr(strategy, "_grid_refresh_paused", lambda: False)
- monkeypatch.setattr(strategy, "_recenter_threshold_pct", lambda: 0.5)
- result = strategy.on_tick({})
- assert result["action"] in {"seed", "reseed"}
- assert ctx.cancelled_all == 1
- assert len(ctx.placed_orders) > 0
- def test_grid_recenters_exactly_on_live_price():
- class FakeContext:
- base_currency = "XRP"
- counter_currency = "USD"
- market_symbol = "xrpusd"
- minimum_order_value = 10.0
- mode = "active"
- def cancel_all_orders(self):
- return {"ok": True}
- def get_fee_rates(self, market):
- return {"maker": 0.0, "taker": 0.0}
- def suggest_order_amount(self, **kwargs):
- return 0.1
- def place_order(self, **kwargs):
- return {"status": "ok", "id": "oid-1"}
- strategy = GridStrategy(FakeContext(), {})
- strategy.state["center_price"] = 100.0
- strategy._recenter_and_rebuild_from_price(160.0, "test recenter")
- assert strategy.state["center_price"] == 160.0
- def test_grid_stop_cancels_all_open_orders():
- class FakeContext:
- base_currency = "XRP"
- counter_currency = "USD"
- market_symbol = "xrpusd"
- minimum_order_value = 10.0
- mode = "active"
- def __init__(self):
- self.cancelled = False
- def cancel_all_orders(self):
- self.cancelled = True
- return {"ok": True}
- def get_fee_rates(self, market):
- return {"maker": 0.0, "taker": 0.0}
- strategy = GridStrategy(FakeContext(), {})
- strategy.state["orders"] = [{"id": "o1"}]
- strategy.state["order_ids"] = ["o1"]
- strategy.state["open_order_count"] = 1
- strategy.on_stop()
- assert strategy.context.cancelled is True
- assert strategy.state["open_order_count"] == 0
- assert strategy.state["last_action"] == "stopped"
- def test_base_strategy_report_uses_context_snapshot():
- class FakeContext:
- id = "s-1"
- account_id = "acct-1"
- market_symbol = "xrpusd"
- base_currency = "XRP"
- counter_currency = "USD"
- mode = "active"
- def get_strategy_snapshot(self):
- return {
- "identity": {"strategy_id": "s-1", "strategy_name": "Demo", "account_id": "acct-1", "market": "xrpusd", "base_currency": "XRP", "quote_currency": "USD"},
- "control": {"enabled_state": "on", "mode": "active"},
- "position": {"balances": [{"asset_code": "XRP", "available": 1.0}]},
- "orders": {"open_orders": [{"id": "o1"}]},
- "execution": {"execution_quality": "good"},
- }
- class DemoStrategy(BaseStrategy):
- LABEL = "Demo"
- report = DemoStrategy(FakeContext(), {}).report()
- assert report["identity"]["strategy_id"] == "s-1"
- assert report["control"]["mode"] == "active"
- assert report["position"]["open_orders"][0]["id"] == "o1"
- def test_trend_follower_uses_policy_and_reports_fit():
- class FakeContext:
- id = "s-2"
- account_id = "acct-2"
- client_id = "cid-2"
- mode = "active"
- market_symbol = "xrpusd"
- base_currency = "XRP"
- counter_currency = "USD"
- def get_price(self, symbol):
- return {"price": 1.2}
- def place_order(self, **kwargs):
- return {"ok": True, "order": kwargs}
- def get_strategy_snapshot(self):
- return {"identity": {}, "control": {}, "position": {}, "orders": {}, "execution": {}}
- strat = TrendStrategy(FakeContext(), {"trade_side": "buy", "order_notional_quote": 1.5})
- strat.apply_policy()
- report = strat.report()
- assert report["fit"]["risk_profile"] == "growth"
- assert strat.state["policy_derived"]["order_notional_quote"] > 0
- def test_trend_follower_buys_from_bull_regime_without_explicit_strength():
- class FakeContext:
- id = "s-bull"
- account_id = "acct-1"
- client_id = "cid-1"
- mode = "active"
- market_symbol = "xrpusd"
- base_currency = "XRP"
- counter_currency = "USD"
- def __init__(self):
- self.orders = []
- def get_price(self, symbol):
- return {"price": 1.2}
- def place_order(self, **kwargs):
- self.orders.append(kwargs)
- return {"ok": True, "order": kwargs}
- def get_account_info(self):
- return {"balances": [{"asset_code": "USD", "available": 1000}, {"asset_code": "XRP", "available": 0}]}
- minimum_order_value = 10.0
- def suggest_order_amount(self, **kwargs):
- return float(kwargs.get("quote_notional") or 0.0) / float(kwargs.get("price") or 1.0)
- def get_strategy_snapshot(self):
- return {"identity": {}, "control": {}, "position": {}, "orders": {}, "execution": {}}
- ctx = FakeContext()
- strat = TrendStrategy(ctx, {"trade_side": "buy", "order_notional_quote": 2.0})
- result = strat.on_tick({})
- assert result["action"] == "buy"
- assert ctx.orders[-1]["side"] == "buy"
- assert ctx.orders[-1]["amount"] == 2.0 / 1.2
- assert strat.state["last_action"] == "buy_trend"
- def test_trend_follower_sells_from_bear_regime_without_explicit_strength():
- class FakeContext:
- id = "s-bear"
- account_id = "acct-1"
- client_id = "cid-1"
- mode = "active"
- market_symbol = "xrpusd"
- base_currency = "XRP"
- counter_currency = "USD"
- def __init__(self):
- self.orders = []
- def get_price(self, symbol):
- return {"price": 1.2}
- def place_order(self, **kwargs):
- self.orders.append(kwargs)
- return {"ok": True, "order": kwargs}
- def get_account_info(self):
- return {"balances": [{"asset_code": "USD", "available": 1000}, {"asset_code": "XRP", "available": 10}]}
- minimum_order_value = 10.0
- def suggest_order_amount(self, **kwargs):
- return float(kwargs.get("quote_notional") or 0.0) / float(kwargs.get("price") or 1.0)
- def get_strategy_snapshot(self):
- return {"identity": {}, "control": {}, "position": {}, "orders": {}, "execution": {}}
- ctx = FakeContext()
- strat = TrendStrategy(ctx, {"trade_side": "sell", "order_notional_quote": 2.0})
- result = strat.on_tick({})
- assert result["action"] == "sell"
- assert ctx.orders[-1]["side"] == "sell"
- assert ctx.orders[-1]["amount"] == 2.0 / 1.2
- assert strat.state["last_action"] == "sell_trend"
- def test_trend_follower_buy_only_ignores_bear_regime():
- class FakeContext:
- id = "s-buy-only"
- account_id = "acct-1"
- client_id = "cid-1"
- mode = "active"
- market_symbol = "xrpusd"
- base_currency = "XRP"
- counter_currency = "USD"
- def __init__(self):
- self.orders = []
- def get_price(self, symbol):
- return {"price": 1.2}
- def get_regime(self, symbol, timeframe="1h"):
- return {
- "trend": {"state": "bear", "ema_fast": 1.17, "ema_slow": 1.2},
- "momentum": {"state": "bear", "rsi": 36, "macd_histogram": -0.002},
- }
- def place_order(self, **kwargs):
- self.orders.append(kwargs)
- return {"ok": True, "order": kwargs}
- def get_account_info(self):
- return {"balances": [{"asset_code": "USD", "available": 1000}, {"asset_code": "XRP", "available": 10}]}
- minimum_order_value = 10.0
- def suggest_order_amount(self, **kwargs):
- return 6.0
- def get_strategy_snapshot(self):
- return {"identity": {}, "control": {}, "position": {}, "orders": {}, "execution": {}}
- ctx = FakeContext()
- strat = TrendStrategy(ctx, {"trade_side": "buy", "order_notional_quote": 2.0})
- result = strat.on_tick({})
- assert result["action"] == "buy"
- assert ctx.orders[-1]["side"] == "buy"
- assert strat.state["last_action"] == "buy_trend"
- def test_trend_follower_sell_only_ignores_bull_regime():
- class FakeContext:
- id = "s-sell-only"
- account_id = "acct-1"
- client_id = "cid-1"
- mode = "active"
- market_symbol = "xrpusd"
- base_currency = "XRP"
- counter_currency = "USD"
- def __init__(self):
- self.orders = []
- def get_price(self, symbol):
- return {"price": 1.2}
- def get_regime(self, symbol, timeframe="1h"):
- return {
- "trend": {"state": "bull", "ema_fast": 1.21, "ema_slow": 1.18},
- "momentum": {"state": "bull", "rsi": 64, "macd_histogram": 0.002},
- }
- def place_order(self, **kwargs):
- self.orders.append(kwargs)
- return {"ok": True, "order": kwargs}
- def get_account_info(self):
- return {"balances": [{"asset_code": "USD", "available": 1000}, {"asset_code": "XRP", "available": 0}]}
- minimum_order_value = 10.0
- def suggest_order_amount(self, **kwargs):
- return 10.0
- def get_strategy_snapshot(self):
- return {"identity": {}, "control": {}, "position": {}, "orders": {}, "execution": {}}
- ctx = FakeContext()
- strat = TrendStrategy(ctx, {"trade_side": "sell", "order_notional_quote": 2.0})
- result = strat.on_tick({})
- assert result["action"] == "sell"
- assert ctx.orders[-1]["side"] == "sell"
- assert strat.state["last_action"] == "sell_trend"
- def test_trend_follower_policy_does_not_override_explicit_order_notional_quote():
- class FakeContext:
- id = "s-explicit"
- account_id = "acct-1"
- client_id = "cid-1"
- mode = "active"
- market_symbol = "xrpusd"
- base_currency = "XRP"
- counter_currency = "USD"
- def get_price(self, symbol):
- return {"price": 1.2}
- def get_strategy_snapshot(self):
- return {"identity": {}, "control": {}, "position": {}, "orders": {}, "execution": {}}
- strat = TrendStrategy(FakeContext(), {"trade_side": "buy", "order_notional_quote": 10.5})
- strat.apply_policy()
- assert strat.config["order_notional_quote"] == 10.5
- assert strat.state["policy_derived"]["order_notional_quote"] == 10.5
- def test_trend_follower_passes_live_fee_rate_into_sizing_helper():
- class FakeContext:
- id = "s-fee"
- account_id = "acct-1"
- client_id = "cid-1"
- mode = "active"
- market_symbol = "xrpusd"
- base_currency = "XRP"
- counter_currency = "USD"
- minimum_order_value = 10.0
- def __init__(self):
- self.fee_calls = []
- self.suggest_calls = []
- def get_price(self, symbol):
- return {"price": 1.2}
- def get_fee_rates(self, market_symbol=None):
- self.fee_calls.append(market_symbol)
- return {"maker": 0.0025, "taker": 0.004}
- def suggest_order_amount(self, **kwargs):
- self.suggest_calls.append(kwargs)
- return 8.0
- def place_order(self, **kwargs):
- return {"ok": True, "order": kwargs}
- def get_account_info(self):
- return {"balances": [{"asset_code": "USD", "available": 1000}, {"asset_code": "XRP", "available": 0}]}
- def get_strategy_snapshot(self):
- return {"identity": {}, "control": {}, "position": {}, "orders": {}, "execution": {}}
- ctx = FakeContext()
- strat = TrendStrategy(ctx, {"trade_side": "buy", "order_notional_quote": 10.5})
- strat.on_tick({})
- assert ctx.fee_calls == ["xrpusd"]
- assert ctx.suggest_calls[-1]["fee_rate"] == 0.0025
|