import pytest
from fastapi.testclient import TestClient
from app.main import app


@pytest.fixture
def client():
    return TestClient(app)


@pytest.fixture
def sample_price_data():
    return {
        "timestamp": "2024-01-01T00:00:00",
        "region": "FR",
        "day_ahead_price": 50.0,
        "real_time_price": 55.0,
        "volume_mw": 1000.0,
    }


@pytest.fixture
def sample_battery_state():
    return {
        "timestamp": "2024-01-01T00:00:00",
        "battery_id": "battery_1",
        "capacity_mwh": 100.0,
        "charge_level_mwh": 50.0,
        "charge_rate_mw": 50.0,
        "discharge_rate_mw": 50.0,
        "efficiency": 0.9,
    }