diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/integration/test_order_execution_flow.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/integration/test_order_execution_flow.py b/tests/integration/test_order_execution_flow.py index 819b7db..d842d29 100644 --- a/tests/integration/test_order_execution_flow.py +++ b/tests/integration/test_order_execution_flow.py @@ -27,7 +27,7 @@ async def test_signal_to_order_flow(): ) exchange = AsyncMock() - exchange.fetch_balance = AsyncMock(return_value={"free": {"USDT": 10000}}) + exchange.get_buying_power = AsyncMock(return_value=Decimal("10000")) risk_manager = RiskManager( max_position_size=Decimal("0.5"), @@ -74,7 +74,7 @@ async def test_signal_rejected_by_risk_manager(): ) exchange = AsyncMock() - exchange.fetch_balance = AsyncMock(return_value={"free": {"USDT": 1000}}) + exchange.get_buying_power = AsyncMock(return_value=Decimal("1000")) risk_manager = RiskManager( max_position_size=Decimal("0.1"), |