"""Cards, map popups and place rows label total_pupils "pupils".

fact_performance's total_pupils is the cohort a year's results were measured
on. For a secondary that is the GCSE year group alone: Burntwood showed 245 in
search against 1,462 on roll. The list and place payloads therefore carry the
register's whole-school count, and nothing when the register has none.
"""

import numpy as np
import pandas as pd
import pytest
from fastapi.testclient import TestClient


def _schools_df() -> pd.DataFrame:
    base = {
        "local_authority": "Essex", "school_type": "Academy converter",
        "year": 202425, "ofsted_grade": 2.0, "ofsted_date": None,
        "town": "Brentwood", "postcode": "CM13 1AA", "status": "Open",
        "address": "1 Test Street", "latitude": 51.6, "longitude": 0.3,
        "gender": "Mixed", "rwm_expected_pct": np.nan, "attainment_8_score": 50.0,
    }
    rows = [
        # Secondary: results cohort 245, register 1,462.
        {**base, "urn": 100001, "school_name": "Alpha High", "phase": "Secondary",
         "total_pupils": 245, "gias_total_pupils": 1462},
        # Register count missing: no count, never the cohort.
        {**base, "urn": 100002, "school_name": "Beta High", "phase": "Secondary",
         "total_pupils": 180, "gias_total_pupils": np.nan},
    ]
    # Enough schools in one town for it to have a place page.
    rows += [
        {**base, "urn": 100010 + i, "school_name": f"Gamma High {i}", "phase": "Secondary",
         "total_pupils": 200, "gias_total_pupils": 1000 + i}
        for i in range(5)
    ]
    return pd.DataFrame(rows)


@pytest.fixture()
def client(monkeypatch):
    from backend import app as app_module

    monkeypatch.setattr(app_module, "load_latest_school_data", _schools_df)
    monkeypatch.setattr(app_module, "load_school_data", _schools_df)
    monkeypatch.setattr(app_module, "_place_registry", None)
    return TestClient(app_module.app, raise_server_exceptions=False)


def _pupils(schools: list[dict]) -> dict[int, object]:
    return {s["urn"]: s.get("total_pupils") for s in schools}


def test_the_list_carries_the_whole_school_count(client):
    resp = client.get("/api/schools?page_size=50")
    assert resp.status_code == 200, resp.text
    pupils = _pupils(resp.json()["schools"])
    assert pupils[100001] == 1462
    assert pupils[100002] is None


def test_a_place_page_carries_the_whole_school_count(client):
    resp = client.get("/api/places/town/brentwood")
    assert resp.status_code == 200, resp.text
    pupils = _pupils(resp.json()["schools"])
    assert pupils[100001] == 1462
    assert pupils[100002] is None