"""Test LIFX diagnostics."""
from homeassistant.components import lifx
from homeassistant.const import CONF_HOST
from homeassistant.core import HomeAssistant
from homeassistant.setup import async_setup_component

from . import (
    DEFAULT_ENTRY_TITLE,
    IP_ADDRESS,
    SERIAL,
    _mocked_bulb,
    _mocked_clean_bulb,
    _mocked_infrared_bulb,
    _mocked_light_strip,
    _patch_config_flow_try_connect,
    _patch_device,
    _patch_discovery,
)

from tests.common import MockConfigEntry
from tests.components.diagnostics import get_diagnostics_for_config_entry
from tests.typing import ClientSessionGenerator


async def test_bulb_diagnostics(
    hass: HomeAssistant, hass_client: ClientSessionGenerator
) -> None:
    """Test diagnostics for a standard bulb."""
    config_entry = MockConfigEntry(
        domain=lifx.DOMAIN,
        title=DEFAULT_ENTRY_TITLE,
        data={CONF_HOST: IP_ADDRESS},
        unique_id=SERIAL,
    )
    config_entry.add_to_hass(hass)
    bulb = _mocked_bulb()
    with _patch_discovery(device=bulb), _patch_config_flow_try_connect(
        device=bulb
    ), _patch_device(device=bulb):
        await async_setup_component(hass, lifx.DOMAIN, {lifx.DOMAIN: {}})
        await hass.async_block_till_done()

    diag = await get_diagnostics_for_config_entry(hass, hass_client, config_entry)
    assert diag == {
        "data": {
            "brightness": 3,
            "features": {
                "buttons": False,
                "chain": False,
                "color": True,
                "extended_multizone": False,
                "hev": False,
                "infrared": False,
                "matrix": False,
                "max_kelvin": 9000,
                "min_kelvin": 2500,
                "multizone": False,
                "relays": False,
            },
            "firmware": "3.00",
            "hue": 1,
            "kelvin": 4,
            "power": 0,
            "product_id": 1,
            "saturation": 2,
            "vendor": None,
        },
        "entry": {"data": {"host": "**REDACTED**"}, "title": "My Bulb"},
    }


async def test_clean_bulb_diagnostics(
    hass: HomeAssistant, hass_client: ClientSessionGenerator
) -> None:
    """Test diagnostics for a standard bulb."""
    config_entry = MockConfigEntry(
        domain=lifx.DOMAIN,
        title=DEFAULT_ENTRY_TITLE,
        data={CONF_HOST: IP_ADDRESS},
        unique_id=SERIAL,
    )
    config_entry.add_to_hass(hass)
    bulb = _mocked_clean_bulb()
    with _patch_discovery(device=bulb), _patch_config_flow_try_connect(
        device=bulb
    ), _patch_device(device=bulb):
        await async_setup_component(hass, lifx.DOMAIN, {lifx.DOMAIN: {}})
        await hass.async_block_till_done()

    diag = await get_diagnostics_for_config_entry(hass, hass_client, config_entry)
    assert diag == {
        "data": {
            "brightness": 3,
            "features": {
                "buttons": False,
                "chain": False,
                "color": True,
                "extended_multizone": False,
                "hev": True,
                "infrared": False,
                "matrix": False,
                "max_kelvin": 9000,
                "min_kelvin": 1500,
                "multizone": False,
                "relays": False,
            },
            "firmware": "3.00",
            "hev": {
                "hev_config": {"duration": 7200, "indication": False},
                "hev_cycle": {"duration": 7200, "last_power": False, "remaining": 30},
                "last_result": 0,
            },
            "hue": 1,
            "kelvin": 4,
            "power": 0,
            "product_id": 90,
            "saturation": 2,
            "vendor": None,
        },
        "entry": {"data": {"host": "**REDACTED**"}, "title": "My Bulb"},
    }


async def test_infrared_bulb_diagnostics(
    hass: HomeAssistant, hass_client: ClientSessionGenerator
) -> None:
    """Test diagnostics for a standard bulb."""
    config_entry = MockConfigEntry(
        domain=lifx.DOMAIN,
        title=DEFAULT_ENTRY_TITLE,
        data={CONF_HOST: IP_ADDRESS},
        unique_id=SERIAL,
    )
    config_entry.add_to_hass(hass)
    bulb = _mocked_infrared_bulb()
    with _patch_discovery(device=bulb), _patch_config_flow_try_connect(
        device=bulb
    ), _patch_device(device=bulb):
        await async_setup_component(hass, lifx.DOMAIN, {lifx.DOMAIN: {}})
        await hass.async_block_till_done()

    diag = await get_diagnostics_for_config_entry(hass, hass_client, config_entry)
    assert diag == {
        "data": {
            "brightness": 3,
            "features": {
                "buttons": False,
                "chain": False,
                "color": True,
                "extended_multizone": False,
                "hev": False,
                "infrared": True,
                "matrix": False,
                "max_kelvin": 9000,
                "min_kelvin": 1500,
                "multizone": False,
                "relays": False,
            },
            "firmware": "3.00",
            "hue": 1,
            "infrared": {"brightness": 65535},
            "kelvin": 4,
            "power": 0,
            "product_id": 29,
            "saturation": 2,
            "vendor": None,
        },
        "entry": {"data": {"host": "**REDACTED**"}, "title": "My Bulb"},
    }


async def test_legacy_multizone_bulb_diagnostics(
    hass: HomeAssistant, hass_client: ClientSessionGenerator
) -> None:
    """Test diagnostics for a standard bulb."""
    config_entry = MockConfigEntry(
        domain=lifx.DOMAIN,
        title=DEFAULT_ENTRY_TITLE,
        data={CONF_HOST: IP_ADDRESS},
        unique_id=SERIAL,
    )
    config_entry.add_to_hass(hass)
    bulb = _mocked_light_strip()
    bulb.zones_count = 8
    bulb.color_zones = [
        (54612, 65535, 65535, 3500),
        (54612, 65535, 65535, 3500),
        (54612, 65535, 65535, 3500),
        (54612, 65535, 65535, 3500),
        (46420, 65535, 65535, 3500),
        (46420, 65535, 65535, 3500),
        (46420, 65535, 65535, 3500),
        (46420, 65535, 65535, 3500),
    ]
    with _patch_discovery(device=bulb), _patch_config_flow_try_connect(
        device=bulb
    ), _patch_device(device=bulb):
        await async_setup_component(hass, lifx.DOMAIN, {lifx.DOMAIN: {}})
        await hass.async_block_till_done()

    diag = await get_diagnostics_for_config_entry(hass, hass_client, config_entry)
    assert diag == {
        "data": {
            "brightness": 3,
            "features": {
                "buttons": False,
                "chain": False,
                "color": True,
                "extended_multizone": False,
                "hev": False,
                "infrared": False,
                "matrix": False,
                "max_kelvin": 9000,
                "min_kelvin": 2500,
                "multizone": True,
                "relays": False,
            },
            "firmware": "3.00",
            "hue": 1,
            "kelvin": 4,
            "power": 0,
            "product_id": 31,
            "saturation": 2,
            "vendor": None,
            "zones": {
                "count": 8,
                "state": {
                    "0": {
                        "brightness": 65535,
                        "hue": 54612,
                        "kelvin": 3500,
                        "saturation": 65535,
                    },
                    "1": {
                        "brightness": 65535,
                        "hue": 54612,
                        "kelvin": 3500,
                        "saturation": 65535,
                    },
                    "2": {
                        "brightness": 65535,
                        "hue": 54612,
                        "kelvin": 3500,
                        "saturation": 65535,
                    },
                    "3": {
                        "brightness": 65535,
                        "hue": 54612,
                        "kelvin": 3500,
                        "saturation": 65535,
                    },
                    "4": {
                        "brightness": 65535,
                        "hue": 46420,
                        "kelvin": 3500,
                        "saturation": 65535,
                    },
                    "5": {
                        "brightness": 65535,
                        "hue": 46420,
                        "kelvin": 3500,
                        "saturation": 65535,
                    },
                    "6": {
                        "brightness": 65535,
                        "hue": 46420,
                        "kelvin": 3500,
                        "saturation": 65535,
                    },
                    "7": {
                        "brightness": 65535,
                        "hue": 46420,
                        "kelvin": 3500,
                        "saturation": 65535,
                    },
                },
            },
        },
        "entry": {"data": {"host": "**REDACTED**"}, "title": "My Bulb"},
    }


async def test_multizone_bulb_diagnostics(
    hass: HomeAssistant, hass_client: ClientSessionGenerator
) -> None:
    """Test diagnostics for a standard bulb."""
    config_entry = MockConfigEntry(
        domain=lifx.DOMAIN,
        title=DEFAULT_ENTRY_TITLE,
        data={CONF_HOST: IP_ADDRESS},
        unique_id=SERIAL,
    )
    config_entry.add_to_hass(hass)
    bulb = _mocked_light_strip()
    bulb.product = 38
    bulb.zones_count = 8
    bulb.color_zones = [
        (54612, 65535, 65535, 3500),
        (54612, 65535, 65535, 3500),
        (54612, 65535, 65535, 3500),
        (54612, 65535, 65535, 3500),
        (46420, 65535, 65535, 3500),
        (46420, 65535, 65535, 3500),
        (46420, 65535, 65535, 3500),
        (46420, 65535, 65535, 3500),
    ]
    with _patch_discovery(device=bulb), _patch_config_flow_try_connect(
        device=bulb
    ), _patch_device(device=bulb):
        await async_setup_component(hass, lifx.DOMAIN, {lifx.DOMAIN: {}})
        await hass.async_block_till_done()

    diag = await get_diagnostics_for_config_entry(hass, hass_client, config_entry)
    assert diag == {
        "data": {
            "brightness": 3,
            "features": {
                "buttons": False,
                "chain": False,
                "color": True,
                "extended_multizone": True,
                "hev": False,
                "infrared": False,
                "matrix": False,
                "max_kelvin": 9000,
                "min_ext_mz_firmware": 1532997580,
                "min_ext_mz_firmware_components": [2, 77],
                "min_kelvin": 1500,
                "multizone": True,
                "relays": False,
            },
            "firmware": "3.00",
            "hue": 1,
            "kelvin": 4,
            "power": 0,
            "product_id": 38,
            "saturation": 2,
            "vendor": None,
            "zones": {
                "count": 8,
                "state": {
                    "0": {
                        "brightness": 65535,
                        "hue": 54612,
                        "kelvin": 3500,
                        "saturation": 65535,
                    },
                    "1": {
                        "brightness": 65535,
                        "hue": 54612,
                        "kelvin": 3500,
                        "saturation": 65535,
                    },
                    "2": {
                        "brightness": 65535,
                        "hue": 54612,
                        "kelvin": 3500,
                        "saturation": 65535,
                    },
                    "3": {
                        "brightness": 65535,
                        "hue": 54612,
                        "kelvin": 3500,
                        "saturation": 65535,
                    },
                    "4": {
                        "brightness": 65535,
                        "hue": 46420,
                        "kelvin": 3500,
                        "saturation": 65535,
                    },
                    "5": {
                        "brightness": 65535,
                        "hue": 46420,
                        "kelvin": 3500,
                        "saturation": 65535,
                    },
                    "6": {
                        "brightness": 65535,
                        "hue": 46420,
                        "kelvin": 3500,
                        "saturation": 65535,
                    },
                    "7": {
                        "brightness": 65535,
                        "hue": 46420,
                        "kelvin": 3500,
                        "saturation": 65535,
                    },
                },
            },
        },
        "entry": {"data": {"host": "**REDACTED**"}, "title": "My Bulb"},
    }