"""Test label registry API."""

import pytest

from homeassistant.components.config import label_registry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import label_registry as lr

from tests.typing import MockHAClientWebSocket, WebSocketGenerator


@pytest.fixture(name="client")
async def client_fixture(
    hass: HomeAssistant, hass_ws_client: WebSocketGenerator
) -> MockHAClientWebSocket:
    """Fixture that can interact with the config manager API."""
    label_registry.async_setup(hass)
    return await hass_ws_client(hass)


async def test_list_labels(
    client: MockHAClientWebSocket,
    label_registry: lr.LabelRegistry,
) -> None:
    """Test list entries."""
    label_registry.async_create("mock 1")
    label_registry.async_create(
        name="mock 2",
        color="#00FF00",
        icon="mdi:two",
        description="This is the second label",
    )

    assert len(label_registry.labels) == 2

    await client.send_json_auto_id({"type": "config/label_registry/list"})

    msg = await client.receive_json()

    assert len(msg["result"]) == len(label_registry.labels)
    assert msg["result"][0] == {
        "color": None,
        "description": None,
        "icon": None,
        "label_id": "mock_1",
        "name": "mock 1",
    }
    assert msg["result"][1] == {
        "color": "#00FF00",
        "description": "This is the second label",
        "icon": "mdi:two",
        "label_id": "mock_2",
        "name": "mock 2",
    }


async def test_create_label(
    client: MockHAClientWebSocket,
    label_registry: lr.LabelRegistry,
) -> None:
    """Test create entry."""
    await client.send_json_auto_id(
        {
            "name": "MOCK",
            "type": "config/label_registry/create",
        }
    )

    msg = await client.receive_json()

    assert len(label_registry.labels) == 1
    assert msg["result"] == {
        "color": None,
        "description": None,
        "icon": None,
        "label_id": "mock",
        "name": "MOCK",
    }

    await client.send_json_auto_id(
        {
            "id": 2,
            "name": "MOCKERY",
            "type": "config/label_registry/create",
            "color": "#00FF00",
            "description": "This is the second label",
            "icon": "mdi:two",
        }
    )

    msg = await client.receive_json()

    assert len(label_registry.labels) == 2
    assert msg["result"] == {
        "color": "#00FF00",
        "description": "This is the second label",
        "icon": "mdi:two",
        "label_id": "mockery",
        "name": "MOCKERY",
    }

    await client.send_json_auto_id(
        {
            "name": "MAGIC",
            "type": "config/label_registry/create",
            "color": "indigo",
            "description": "This is the third label",
            "icon": "mdi:three",
        }
    )

    msg = await client.receive_json()

    assert len(label_registry.labels) == 3
    assert msg["result"] == {
        "color": "indigo",
        "description": "This is the third label",
        "icon": "mdi:three",
        "label_id": "magic",
        "name": "MAGIC",
    }


async def test_create_label_with_name_already_in_use(
    client: MockHAClientWebSocket,
    label_registry: lr.LabelRegistry,
) -> None:
    """Test create entry that should fail."""
    label_registry.async_create("mock")
    assert len(label_registry.labels) == 1

    await client.send_json_auto_id(
        {"name": "mock", "type": "config/label_registry/create"}
    )

    msg = await client.receive_json()

    assert not msg["success"]
    assert msg["error"]["code"] == "invalid_info"
    assert msg["error"]["message"] == "The name mock (mock) is already in use"
    assert len(label_registry.labels) == 1


async def test_delete_label(
    client: MockHAClientWebSocket,
    label_registry: lr.LabelRegistry,
) -> None:
    """Test delete entry."""
    label = label_registry.async_create("mock")
    assert len(label_registry.labels) == 1

    await client.send_json_auto_id(
        {"label_id": label.label_id, "type": "config/label_registry/delete"}
    )

    msg = await client.receive_json()

    assert msg["success"]
    assert not label_registry.labels


async def test_delete_non_existing_label(
    client: MockHAClientWebSocket,
    label_registry: lr.LabelRegistry,
) -> None:
    """Test delete entry that should fail."""
    label_registry.async_create("mock")
    assert len(label_registry.labels) == 1

    await client.send_json_auto_id(
        {"label_id": "omg_puppies", "type": "config/label_registry/delete"}
    )

    msg = await client.receive_json()

    assert not msg["success"]
    assert msg["error"]["code"] == "invalid_info"
    assert msg["error"]["message"] == "Label ID doesn't exist"
    assert len(label_registry.labels) == 1


async def test_update_label(
    client: MockHAClientWebSocket,
    label_registry: lr.LabelRegistry,
) -> None:
    """Test update entry."""
    label = label_registry.async_create("mock")
    assert len(label_registry.labels) == 1

    await client.send_json_auto_id(
        {
            "label_id": label.label_id,
            "name": "UPDATED",
            "icon": "mdi:test",
            "color": "#00FF00",
            "description": "This is a label description",
            "type": "config/label_registry/update",
        }
    )

    msg = await client.receive_json()

    assert len(label_registry.labels) == 1
    assert msg["result"] == {
        "color": "#00FF00",
        "description": "This is a label description",
        "icon": "mdi:test",
        "label_id": "mock",
        "name": "UPDATED",
    }

    await client.send_json_auto_id(
        {
            "label_id": label.label_id,
            "name": "UPDATED AGAIN",
            "icon": None,
            "color": None,
            "description": None,
            "type": "config/label_registry/update",
        }
    )

    msg = await client.receive_json()

    assert len(label_registry.labels) == 1
    assert msg["result"] == {
        "color": None,
        "description": None,
        "icon": None,
        "label_id": "mock",
        "name": "UPDATED AGAIN",
    }

    await client.send_json_auto_id(
        {
            "label_id": label.label_id,
            "name": "UPDATED YET AGAIN",
            "icon": None,
            "color": "primary",
            "description": None,
            "type": "config/label_registry/update",
        }
    )

    msg = await client.receive_json()

    assert len(label_registry.labels) == 1
    assert msg["result"] == {
        "color": "primary",
        "description": None,
        "icon": None,
        "label_id": "mock",
        "name": "UPDATED YET AGAIN",
    }


async def test_update_with_name_already_in_use(
    client: MockHAClientWebSocket,
    label_registry: lr.LabelRegistry,
) -> None:
    """Test update entry."""
    label = label_registry.async_create("mock 1")
    label_registry.async_create("mock 2")
    assert len(label_registry.labels) == 2

    await client.send_json_auto_id(
        {
            "label_id": label.label_id,
            "name": "mock 2",
            "type": "config/label_registry/update",
        }
    )

    msg = await client.receive_json()

    assert not msg["success"]
    assert msg["error"]["code"] == "invalid_info"
    assert msg["error"]["message"] == "The name mock 2 (mock2) is already in use"
    assert len(label_registry.labels) == 2