def _make_set_with_subclass(client, name="Enemy", sub_name="Bat"):
    set_id = client.post("/sets", json={"name": name}).json()["id"]
    main_class_id = client.post(f"/sets/{set_id}/main-classes", json={"name": name}).json()["id"]
    sub_class_id = client.post(f"/main-classes/{main_class_id}/sub-classes", json={"name": sub_name}).json()["id"]
    return set_id, sub_class_id


def _make_frame(client, session_name="run01", frame_index=0):
    payload = {"session_name": session_name, "frame_index": frame_index, "image_path": "x.png", "width": 1280, "height": 720}
    return client.post("/frames", json=payload).json()["id"]


def test_create_list_update_delete_label(client):
    set_id, sub_class_id = _make_set_with_subclass(client)
    frame_id = _make_frame(client)

    resp = client.post(
        f"/frames/{frame_id}/sets/{set_id}/labels",
        json={"sub_class_id": sub_class_id, "x": 10, "y": 20, "width": 30, "height": 40, "created_by": "jonas"},
    )
    assert resp.status_code == 201
    label_id = resp.json()["id"]

    resp = client.get(f"/frames/{frame_id}/sets/{set_id}/labels")
    assert len(resp.json()) == 1

    resp = client.patch(f"/labels/{label_id}", json={"x": 15})
    assert resp.status_code == 200
    assert resp.json()["x"] == 15

    resp = client.delete(f"/labels/{label_id}")
    assert resp.status_code == 204

    resp = client.get(f"/frames/{frame_id}/sets/{set_id}/labels")
    assert resp.json() == []


def test_label_rejects_sub_class_from_another_set(client):
    _, sub_class_id = _make_set_with_subclass(client, name="Enemy", sub_name="Bat")
    other_set_id, _ = _make_set_with_subclass(client, name="Items", sub_name="Gold")
    frame_id = _make_frame(client)

    resp = client.post(
        f"/frames/{frame_id}/sets/{other_set_id}/labels",
        json={"sub_class_id": sub_class_id, "x": 0, "y": 0, "width": 1, "height": 1},
    )
    assert resp.status_code == 422


def test_labels_require_existing_frame_and_set(client):
    _, sub_class_id = _make_set_with_subclass(client)
    resp = client.post(
        "/frames/999/sets/999/labels",
        json={"sub_class_id": sub_class_id, "x": 0, "y": 0, "width": 1, "height": 1},
    )
    assert resp.status_code == 404


def test_frame_set_status_defaults_and_updates(client):
    set_id, _ = _make_set_with_subclass(client)
    frame_id = _make_frame(client)

    resp = client.get(f"/frames/{frame_id}/sets/{set_id}/status")
    assert resp.status_code == 200
    assert resp.json()["status"] == "unlabeled"

    resp = client.put(
        f"/frames/{frame_id}/sets/{set_id}/status",
        json={"status": "reviewed", "updated_by": "jonas"},
    )
    assert resp.status_code == 200
    assert resp.json()["status"] == "reviewed"

    resp = client.get(f"/frames/{frame_id}/sets/{set_id}/status")
    assert resp.json()["status"] == "reviewed"