from __future__ import annotations

import pytest
from fastapi import HTTPException
from mcp_bybit.leverage_cap import enforce_leverage, get_max_leverage


def test_get_max_leverage_returns_creds_value():
    creds = {"max_leverage": 5}
    assert get_max_leverage(creds) == 5


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_under_cap():
    creds = {"max_leverage": 3}
    enforce_leverage(2, creds=creds, exchange="bybit")  # no raise


def test_enforce_leverage_pass_at_cap():
    creds = {"max_leverage": 3}
    enforce_leverage(3, creds=creds, exchange="bybit")  # no raise


def test_enforce_leverage_reject_over_cap():
    creds = {"max_leverage": 3}
    with pytest.raises(HTTPException) as exc:
        enforce_leverage(10, creds=creds, exchange="bybit")
    assert exc.value.status_code == 403
    assert exc.value.detail["error"] == "LEVERAGE_CAP_EXCEEDED"
    assert exc.value.detail["exchange"] == "bybit"
    assert exc.value.detail["requested"] == 10
    assert exc.value.detail["max"] == 3


def test_enforce_leverage_reject_when_below_one():
    creds = {"max_leverage": 3}
    with pytest.raises(HTTPException) as exc:
        enforce_leverage(0, creds=creds, exchange="bybit")
    assert exc.value.status_code == 403


def test_enforce_leverage_default_when_none():
    """Se requested è None, applica il cap come default."""
    creds = {"max_leverage": 3}
    result = enforce_leverage(None, creds=creds, exchange="bybit")
    assert result == 3