import pytest


@pytest.mark.asyncio
async def test_favorite_groups_crud(client, host_factory, db_session):
    # Create group
    resp = await client.post(
        "/api/favorites/groups",
        json={"name": "Production", "sort_order": 1, "color": "purple"},
    )
    assert resp.status_code == 201
    group = resp.json()
    assert group["name"] == "Production"

    # List groups
    resp = await client.get("/api/favorites/groups")
    assert resp.status_code == 200
    data = resp.json()
    assert any(g["name"] == "Production" for g in data["groups"])

    # Update group
    resp = await client.patch(
        f"/api/favorites/groups/{group['id']}",
        json={"name": "Prod", "sort_order": 2},
    )
    assert resp.status_code == 200
    updated = resp.json()
    assert updated["name"] == "Prod"
    assert updated["sort_order"] == 2

    # Delete group
    resp = await client.delete(f"/api/favorites/groups/{group['id']}")
    assert resp.status_code == 204


@pytest.mark.asyncio
async def test_favorite_containers_add_list_delete(client, host_factory, db_session):
    # Create a host + docker container in DB
    host = await host_factory.create(db_session, id="host-0001", name="docker1", ip_address="10.0.0.10")

    from app.crud.docker_container import DockerContainerRepository

    repo = DockerContainerRepository(db_session)
    c = await repo.upsert(
        host_id=host.id,
        container_id="abcdef123456",
        name="portainer",
        image="portainer/portainer-ce:latest",
        state="running",
        status="Up 1 hour",
        health=None,
        ports={"raw": "0.0.0.0:9000->9000/tcp"},
        labels=None,
        compose_project=None,
    )
    await db_session.commit()

    # Create favorite
    resp = await client.post(
        "/api/favorites/containers",
        json={"host_id": host.id, "container_id": c.container_id, "group_id": None},
    )
    assert resp.status_code == 201
    fav = resp.json()
    assert fav["docker_container"]["name"] == "portainer"
    fav_id = fav["id"]

    # List favorites
    resp = await client.get("/api/favorites/containers")
    assert resp.status_code == 200
    data = resp.json()
    assert any(x["id"] == fav_id for x in data["containers"])

    # Idempotent add should return 201? We return 201 always, but route returns 201; if exists we return 200? current impl returns 201 for POST but returns existing directly.
    resp2 = await client.post(
        "/api/favorites/containers",
        json={"host_id": host.id, "container_id": c.container_id, "group_id": None},
    )
    assert resp2.status_code in (200, 201)
    fav2 = resp2.json()
    assert fav2["id"] == fav_id

    # Delete favorite
    resp = await client.delete(f"/api/favorites/containers/{fav_id}")
    assert resp.status_code == 204

    # List favorites -> empty
    resp = await client.get("/api/favorites/containers")
    assert resp.status_code == 200
    assert resp.json()["containers"] == []