AdrianoDev
48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
from fastapi import HTTPException
|
|
|
|
from mcp_alpaca.leverage_cap import enforce_leverage, get_max_leverage
|
|
|
|
|
|
def test_get_max_leverage_returns_creds_value():
|
|
creds = {"max_leverage": 4}
|
|
assert get_max_leverage(creds) == 4
|
|
|
|
|
|
def test_get_max_leverage_default_when_missing():
|
|
"""Default 1 (cash) se il secret non ha max_leverage."""
|
|
assert get_max_leverage({}) == 1
|
|
|
|
|
|
def test_enforce_leverage_pass_at_cap_one():
|
|
"""Alpaca cash account: cap 1, leverage 1 OK."""
|
|
creds = {"max_leverage": 1}
|
|
enforce_leverage(1, creds=creds, exchange="alpaca") # no raise
|
|
|
|
|
|
def test_enforce_leverage_reject_over_cap_one():
|
|
creds = {"max_leverage": 1}
|
|
with pytest.raises(HTTPException) as exc:
|
|
enforce_leverage(2, creds=creds, exchange="alpaca")
|
|
assert exc.value.status_code == 403
|
|
assert exc.value.detail["error"] == "LEVERAGE_CAP_EXCEEDED"
|
|
assert exc.value.detail["exchange"] == "alpaca"
|
|
assert exc.value.detail["requested"] == 2
|
|
assert exc.value.detail["max"] == 1
|
|
|
|
|
|
def test_enforce_leverage_reject_when_below_one():
|
|
creds = {"max_leverage": 1}
|
|
with pytest.raises(HTTPException) as exc:
|
|
enforce_leverage(0, creds=creds, exchange="alpaca")
|
|
assert exc.value.status_code == 403
|
|
|
|
|
|
def test_enforce_leverage_default_when_none():
|
|
"""Se requested è None, applica il cap come default."""
|
|
creds = {"max_leverage": 1}
|
|
result = enforce_leverage(None, creds=creds, exchange="alpaca")
|
|
assert result == 1
|