"""Tests for the diagnostics data provided by the VeSync integration."""

from syrupy.assertion import SnapshotAssertion
from syrupy.matchers import path_type

from homeassistant.components.vesync.const import DOMAIN
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.typing import ConfigType
from homeassistant.setup import async_setup_component

from tests.components.diagnostics import (
    get_diagnostics_for_config_entry,
    get_diagnostics_for_device,
)
from tests.typing import ClientSessionGenerator


async def test_async_get_config_entry_diagnostics__no_devices(
    hass: HomeAssistant,
    hass_client: ClientSessionGenerator,
    config_entry: ConfigEntry,
    config: ConfigType,
    snapshot: SnapshotAssertion,
    manager,
) -> None:
    """Test diagnostics for config entry."""
    assert await async_setup_component(hass, DOMAIN, config)
    await hass.async_block_till_done()

    diag = await get_diagnostics_for_config_entry(hass, hass_client, config_entry)

    assert isinstance(diag, dict)
    assert diag == snapshot


async def test_async_get_config_entry_diagnostics__single_humidifier(
    hass: HomeAssistant,
    hass_client: ClientSessionGenerator,
    config_entry: ConfigEntry,
    config: ConfigType,
    snapshot: SnapshotAssertion,
    manager,
    humidifier,
) -> None:
    """Test diagnostics for config entry."""
    manager._dev_list["humidifiers"].append(humidifier)

    assert await async_setup_component(hass, DOMAIN, config)
    await hass.async_block_till_done()

    diag = await get_diagnostics_for_config_entry(hass, hass_client, config_entry)

    assert isinstance(diag, dict)
    assert diag == snapshot


async def test_async_get_device_diagnostics__single_fan(
    hass: HomeAssistant,
    device_registry: dr.DeviceRegistry,
    hass_client: ClientSessionGenerator,
    config_entry: ConfigEntry,
    config: ConfigType,
    snapshot: SnapshotAssertion,
    manager,
    fan,
) -> None:
    """Test diagnostics for config entry."""
    manager._dev_list["fans"].append(fan)

    assert await async_setup_component(hass, DOMAIN, config)
    await hass.async_block_till_done()

    device = device_registry.async_get_device(
        identifiers={(DOMAIN, "fan")},
    )
    assert device is not None

    diag = await get_diagnostics_for_device(hass, hass_client, config_entry, device)

    assert isinstance(diag, dict)
    diag["home_assistant"]["entities"] = sorted(
        diag["home_assistant"]["entities"], key=lambda ent: ent["entity_id"]
    )
    assert diag == snapshot(
        matcher=path_type(
            {
                "home_assistant.entities.0.state.last_changed": (str,),
                "home_assistant.entities.0.state.last_reported": (str,),
                "home_assistant.entities.0.state.last_updated": (str,),
                "home_assistant.entities.1.state.last_changed": (str,),
                "home_assistant.entities.1.state.last_reported": (str,),
                "home_assistant.entities.1.state.last_updated": (str,),
                "home_assistant.entities.2.state.last_changed": (str,),
                "home_assistant.entities.2.state.last_reported": (str,),
                "home_assistant.entities.2.state.last_updated": (str,),
                "home_assistant.entities.3.state.last_changed": (str,),
                "home_assistant.entities.3.state.last_reported": (str,),
                "home_assistant.entities.3.state.last_updated": (str,),
                "home_assistant.entities.4.state.last_changed": (str,),
                "home_assistant.entities.4.state.last_reported": (str,),
                "home_assistant.entities.4.state.last_updated": (str,),
                "home_assistant.entities.5.state.last_changed": (str,),
                "home_assistant.entities.5.state.last_reported": (str,),
                "home_assistant.entities.5.state.last_updated": (str,),
                "home_assistant.entities.6.state.last_changed": (str,),
                "home_assistant.entities.6.state.last_reported": (str,),
                "home_assistant.entities.6.state.last_updated": (str,),
            }
        )
    )