diff options
Diffstat (limited to 'services/api')
| -rw-r--r-- | services/api/src/trading_api/main.py | 1 | ||||
| -rw-r--r-- | services/api/src/trading_api/routers/orders.py | 1 | ||||
| -rw-r--r-- | services/api/src/trading_api/routers/portfolio.py | 3 | ||||
| -rw-r--r-- | services/api/src/trading_api/routers/strategies.py | 2 | ||||
| -rw-r--r-- | services/api/tests/test_api.py | 5 |
5 files changed, 9 insertions, 3 deletions
diff --git a/services/api/src/trading_api/main.py b/services/api/src/trading_api/main.py index 61cfe36..39f7b43 100644 --- a/services/api/src/trading_api/main.py +++ b/services/api/src/trading_api/main.py @@ -1,4 +1,5 @@ """Trading Platform REST API.""" + from contextlib import asynccontextmanager from fastapi import FastAPI diff --git a/services/api/src/trading_api/routers/orders.py b/services/api/src/trading_api/routers/orders.py index 989694f..d0b9fa6 100644 --- a/services/api/src/trading_api/routers/orders.py +++ b/services/api/src/trading_api/routers/orders.py @@ -1,4 +1,5 @@ """Order endpoints.""" + from fastapi import APIRouter, Request from shared.sa_models import OrderRow, SignalRow from sqlalchemy import select diff --git a/services/api/src/trading_api/routers/portfolio.py b/services/api/src/trading_api/routers/portfolio.py index f4169cb..3b30e1d 100644 --- a/services/api/src/trading_api/routers/portfolio.py +++ b/services/api/src/trading_api/routers/portfolio.py @@ -1,6 +1,7 @@ """Portfolio endpoints.""" + from fastapi import APIRouter, Request -from shared.sa_models import PositionRow, PortfolioSnapshotRow +from shared.sa_models import PositionRow from sqlalchemy import select router = APIRouter() diff --git a/services/api/src/trading_api/routers/strategies.py b/services/api/src/trading_api/routers/strategies.py index a8d778d..2861eec 100644 --- a/services/api/src/trading_api/routers/strategies.py +++ b/services/api/src/trading_api/routers/strategies.py @@ -1,4 +1,5 @@ """Strategy endpoints.""" + import sys from pathlib import Path @@ -16,6 +17,7 @@ router = APIRouter() async def list_strategies(): """List available strategies.""" from strategy_engine.plugin_loader import load_strategies + strategies_dir = _STRATEGY_DIR / "strategies" strategies = load_strategies(strategies_dir) return [ diff --git a/services/api/tests/test_api.py b/services/api/tests/test_api.py index 99cf9e3..669143b 100644 --- a/services/api/tests/test_api.py +++ b/services/api/tests/test_api.py @@ -1,12 +1,13 @@ """Tests for the REST API.""" -import pytest -from unittest.mock import AsyncMock, MagicMock, patch + +from unittest.mock import AsyncMock, patch from fastapi.testclient import TestClient def test_health_endpoint(): """Health endpoint returns ok.""" from trading_api.main import app + # Override lifespan to skip DB with patch("trading_api.main.lifespan") as mock_lifespan: mock_lifespan.return_value.__aenter__ = AsyncMock() |
