"""Test the Lutron Caseta diagnostics."""

from unittest.mock import patch

from homeassistant.components.lutron_caseta import DOMAIN
from homeassistant.components.lutron_caseta.const import (
    CONF_CA_CERTS,
    CONF_CERTFILE,
    CONF_KEYFILE,
)
from homeassistant.const import CONF_HOST

from . import MockBridge

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


async def test_diagnostics(hass, hass_client) -> None:
    """Test generating diagnostics for lutron_caseta."""
    config_entry = MockConfigEntry(
        domain=DOMAIN,
        data={
            CONF_HOST: "1.1.1.1",
            CONF_KEYFILE: "",
            CONF_CERTFILE: "",
            CONF_CA_CERTS: "",
        },
        unique_id="abc",
    )
    config_entry.add_to_hass(hass)

    with patch(
        "homeassistant.components.lutron_caseta.Smartbridge.create_tls",
        return_value=MockBridge(can_connect=True),
    ):
        await hass.config_entries.async_setup(config_entry.entry_id)
        await hass.async_block_till_done()

    diag = await get_diagnostics_for_config_entry(hass, hass_client, config_entry)
    assert diag == {
        "data": {
            "areas": {
                "898": {"id": "898", "name": "Basement", "parent_id": None},
                "822": {"id": "822", "name": "Bedroom", "parent_id": "898"},
                "910": {"id": "910", "name": "Bathroom", "parent_id": "898"},
                "1024": {"id": "1024", "name": "Master Bedroom", "parent_id": None},
                "1025": {"id": "1025", "name": "Kitchen", "parent_id": None},
                "1026": {"id": "1026", "name": "Dining Room", "parent_id": None},
                "1205": {"id": "1205", "name": "Hallway", "parent_id": None},
            },
            "buttons": {
                "111": {
                    "device_id": "111",
                    "current_state": "Release",
                    "button_number": 0,
                    "name": "Dining Room_Pico",
                    "type": "Pico3ButtonRaiseLower",
                    "model": "PJ2-3BRL-GXX-X01",
                    "serial": 68551522,
                    "parent_device": "9",
                },
                "1372": {
                    "device_id": "1372",
                    "current_state": "Release",
                    "button_number": 3,
                    "button_group": "1363",
                    "name": "Hallway_Main Stairs Position 1 Keypad",
                    "type": "SunnataKeypad",
                    "model": "RRST-W3RL-XX",
                    "serial": 66286451,
                    "button_name": "Kitchen Pendants",
                    "button_led": "1362",
                    "device_name": "Kitchen Pendants",
                    "parent_device": "1355",
                },
            },
            "devices": {
                "1": {
                    "model": "model",
                    "name": "bridge",
                    "serial": 1234,
                    "type": "type",
                    "area": "1205",
                },
                "801": {
                    "device_id": "801",
                    "current_state": 100,
                    "fan_speed": None,
                    "zone": "801",
                    "name": "Basement Bedroom_Main Lights",
                    "button_groups": None,
                    "type": "Dimmed",
                    "model": None,
                    "serial": None,
                    "tilt": None,
                    "area": "822",
                },
                "802": {
                    "device_id": "802",
                    "current_state": 100,
                    "fan_speed": None,
                    "zone": "802",
                    "name": "Basement Bedroom_Left Shade",
                    "button_groups": None,
                    "type": "SerenaRollerShade",
                    "model": None,
                    "serial": None,
                    "tilt": None,
                    "area": "822",
                },
                "803": {
                    "device_id": "803",
                    "current_state": 100,
                    "fan_speed": None,
                    "zone": "803",
                    "name": "Basement Bathroom_Exhaust Fan",
                    "button_groups": None,
                    "type": "Switched",
                    "model": None,
                    "serial": None,
                    "tilt": None,
                    "area": "910",
                },
                "804": {
                    "device_id": "804",
                    "current_state": 100,
                    "fan_speed": None,
                    "zone": "804",
                    "name": "Master Bedroom_Ceiling Fan",
                    "button_groups": None,
                    "type": "FanSpeed",
                    "model": None,
                    "serial": None,
                    "tilt": None,
                    "area": "1024",
                },
                "901": {
                    "device_id": "901",
                    "current_state": 100,
                    "fan_speed": None,
                    "zone": "901",
                    "name": "Kitchen_Main Lights",
                    "button_groups": None,
                    "type": "WallDimmer",
                    "model": None,
                    "serial": 5442321,
                    "tilt": None,
                    "area": "1025",
                },
                "9": {
                    "device_id": "9",
                    "current_state": -1,
                    "fan_speed": None,
                    "tilt": None,
                    "zone": None,
                    "name": "Dining Room_Pico",
                    "button_groups": ["4"],
                    "occupancy_sensors": None,
                    "type": "Pico3ButtonRaiseLower",
                    "model": "PJ2-3BRL-GXX-X01",
                    "serial": 68551522,
                    "device_name": "Pico",
                    "area": "1026",
                },
                "1355": {
                    "device_id": "1355",
                    "current_state": -1,
                    "fan_speed": None,
                    "zone": None,
                    "name": "Hallway_Main Stairs Position 1 Keypad",
                    "button_groups": ["1363"],
                    "type": "SunnataKeypad",
                    "model": "RRST-W3RL-XX",
                    "serial": 66286451,
                    "control_station_name": "Main Stairs",
                    "device_name": "Position 1",
                    "area": "1205",
                },
            },
            "occupancy_groups": {},
            "scenes": {},
        },
        "entry": {
            "data": {"ca_certs": "", "certfile": "", "host": "1.1.1.1", "keyfile": ""},
            "title": "Mock Title",
        },
    }