kbt-devops
fe76bc7629
Implements FastAPI backend with ML model support for energy trading, including price prediction models and RL-based battery trading policy. Features dashboard, trading, backtest, and settings API routes with WebSocket support for real-time updates.
47 lines
1.3 KiB
Python
47 lines
1.3 KiB
Python
import pytest
|
|
from fastapi.testclient import TestClient
|
|
|
|
|
|
def test_health_check(client: TestClient):
|
|
response = client.get("/health")
|
|
assert response.status_code == 200
|
|
assert response.json()["status"] == "healthy"
|
|
|
|
|
|
def test_dashboard_summary(client: TestClient):
|
|
response = client.get("/api/v1/dashboard/summary")
|
|
assert response.status_code == 200
|
|
|
|
|
|
def test_latest_prices(client: TestClient):
|
|
response = client.get("/api/v1/dashboard/prices")
|
|
assert response.status_code == 200
|
|
assert "regions" in response.json()
|
|
|
|
|
|
def test_battery_states(client: TestClient):
|
|
response = client.get("/api/v1/dashboard/battery")
|
|
assert response.status_code == 200
|
|
assert "batteries" in response.json()
|
|
|
|
|
|
def test_arbitrage_opportunities(client: TestClient):
|
|
response = client.get("/api/v1/dashboard/arbitrage")
|
|
assert response.status_code == 200
|
|
assert "opportunities" in response.json()
|
|
|
|
|
|
def test_get_settings(client: TestClient):
|
|
response = client.get("/api/v1/settings")
|
|
assert response.status_code == 200
|
|
|
|
|
|
def test_list_models(client: TestClient):
|
|
response = client.get("/api/v1/models")
|
|
assert response.status_code == 200
|
|
|
|
|
|
def test_get_strategies(client: TestClient):
|
|
response = client.get("/api/v1/trading/strategies")
|
|
assert response.status_code == 200
|