root
29 lines
845 B
Python
29 lines
845 B
Python
from __future__ import annotations
|
|
|
|
from unittest.mock import AsyncMock, MagicMock
|
|
|
|
import pytest
|
|
from cerbero_mcp.exchanges.ibkr import tools as t
|
|
|
|
|
|
def test_place_order_req_schema():
|
|
req = t.PlaceOrderReq(symbol="AAPL", side="buy", qty=1)
|
|
assert req.order_type == "market"
|
|
assert req.tif == "day"
|
|
assert req.exchange == "SMART"
|
|
|
|
|
|
def test_place_order_req_options_validates_occ():
|
|
req = t.PlaceOrderReq(
|
|
symbol="AAPL 240119C00190000", side="buy", qty=1, asset_class="options",
|
|
)
|
|
assert req.asset_class == "options"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_account_tool_calls_client():
|
|
client = MagicMock()
|
|
client.get_account = AsyncMock(return_value={"netliquidation": {"amount": 10000}})
|
|
res = await t.get_account(client, t.GetAccountReq())
|
|
assert res["netliquidation"]["amount"] == 10000
|