import sys
import os
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from fastapi.testclient import TestClient
from main import app


def test_api_contract_core_endpoints():
    client = TestClient(app)

    health = client.get("/health")
    assert health.status_code == 200
    assert health.json()["status"] == "ok"

    reset = client.post("/reset", json={"task_id": "default", "seed": 7})
    assert reset.status_code == 200
    data = reset.json()
    sid = data["session_id"]

    state = client.get(f"/state?session_id={sid}")
    assert state.status_code == 200
    assert "observation" in state.json()

    obs = state.json()["observation"]
    action = {
        "action": {
            "bids": [
                {
                    "agent_id": "renewable_1",
                    "role": "renewable_prosumer",
                    "bid_type": "supply",
                    "quantity_mwh": min(obs["renewable_availability_mwh"], obs["demand_mwh"] * 0.5),
                    "price_usd_per_mwh": 22,
                },
                {
                    "agent_id": "peaker_1",
                    "role": "peaker_plant",
                    "bid_type": "supply",
                    "quantity_mwh": min(obs["peaker_capacity_mwh"], obs["demand_mwh"] * 0.5),
                    "price_usd_per_mwh": 58,
                },
                {
                    "agent_id": "industrial_1",
                    "role": "industrial_load",
                    "bid_type": "demand",
                    "quantity_mwh": obs["demand_mwh"],
                    "price_usd_per_mwh": 85,
                },
            ],
            "ev_charge_mwh": 0,
            "ev_discharge_mwh": 0,
        }
    }

    step = client.post(f"/step?session_id={sid}", json=action)
    assert step.status_code == 200
    step_j = step.json()
    assert "reward" in step_j and "info" in step_j and "observation" in step_j

    info = client.get("/info")
    assert info.status_code == 200
    assert "tasks" in info.json()

    events = client.get(f"/events?session_id={sid}")
    assert events.status_code == 200
    assert "events" in events.json()


def test_demo_mode_endpoint_is_deterministic():
    client = TestClient(app)

    run_a = client.post("/run-demo-mode")
    run_b = client.post("/run-demo-mode")

    assert run_a.status_code == 200
    assert run_b.status_code == 200

    ja = run_a.json()
    jb = run_b.json()

    assert ja["mode"] == "demo"
    assert ja["deterministic"] is True
    assert ja["seed"] == 42
    assert ja["task_id"] == "default"
    assert ja["policy"] == "adaptive"
    assert ja["personality"] == "balanced"
    assert ja["average_reward"] == jb["average_reward"]

    rewards_a = [step["reward"]["score"] for step in ja["trajectory"]]
    rewards_b = [step["reward"]["score"] for step in jb["trajectory"]]
    assert rewards_a == rewards_b