# backend/tests/test_structure.py
import pytest
from httpx import AsyncClient

pytestmark = pytest.mark.anyio


async def test_manage_teams(
    client: AsyncClient, auth_headers, valid_tournament_payload
):
    # Setup
    res = await client.post(
        "/tournaments", json=valid_tournament_payload, headers=auth_headers
    )
    t_id = res.json()["id"]

    # 1. Add a Team
    new_team = {"name": "Team E"}
    post_res = await client.post(
        f"/tournaments/{t_id}/teams", json=new_team, headers=auth_headers
    )
    assert post_res.status_code == 200

    # 2. Verify Bracket Regenerated
    detail_res = await client.get(f"/tournaments/{t_id}")
    matches = detail_res.json()["matches"]

    # With 5 teams, bracket size increases
    assert len(matches) > 3

    # 3. Bulk Update
    new_team_list = ["Team X", "Team Y"]
    patch_res = await client.patch(
        f"/tournaments/{t_id}/teams", json=new_team_list, headers=auth_headers
    )
    assert patch_res.status_code == 200

    updated_detail = await client.get(f"/tournaments/{t_id}")
    final_matches = updated_detail.json()["matches"]

    # Fix: For Double Elim with 2 teams, we might get 2 matches (WB Final + Grand Final).
    # Just ensure we have at least 1 match.
    assert len(final_matches) >= 1

    # Verify the teams are actually in the first match
    first_match = next(
        m
        for m in final_matches
        if m["name"] == "Winners Final"
        or m["name"] == "Grand Final"
        or m["name"].startswith("WB")
    )
    p1_name = first_match["p1_team"]["name"] if first_match["p1_team"] else None
    p2_name = first_match["p2_team"]["name"] if first_match["p2_team"] else None

    assert "Team X" in [p1_name, p2_name]
    assert "Team Y" in [p1_name, p2_name]


async def test_manage_courts(
    client: AsyncClient, auth_headers, valid_tournament_payload
):
    res = await client.post(
        "/tournaments", json=valid_tournament_payload, headers=auth_headers
    )
    t_id = res.json()["id"]

    # Get initial courts
    courts_res = await client.get(f"/tournaments/{t_id}")
    initial_courts = courts_res.json()["courts"]
    assert len(initial_courts) == 2

    # Delete
    court_id = initial_courts[0]["id"]
    del_res = await client.delete(
        f"/tournaments/{t_id}/courts/{court_id}", headers=auth_headers
    )
    assert del_res.status_code == 200

    # Create
    create_res = await client.post(
        f"/tournaments/{t_id}/courts", json={"name": "New Court"}, headers=auth_headers
    )
    assert create_res.status_code == 200
    assert create_res.json()["name"] == "New Court"