"""Test Sensibo diagnostics."""

from __future__ import annotations

from syrupy.assertion import SnapshotAssertion

from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant

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

EXCLUDE_ATTRIBUTES = {"full_features"}


async def test_diagnostics(
    hass: HomeAssistant,
    hass_client: ClientSessionGenerator,
    load_int: ConfigEntry,
    snapshot: SnapshotAssertion,
) -> None:
    """Test generating diagnostics for a config entry."""
    entry = load_int

    diag = await get_diagnostics_for_config_entry(hass, hass_client, entry)

    assert diag["ABC999111"]["full_capabilities"] == snapshot
    assert diag["ABC999111"]["fan_modes_translated"] == snapshot
    assert diag["ABC999111"]["swing_modes_translated"] == snapshot
    assert diag["ABC999111"]["horizontal_swing_modes_translated"] == snapshot
    assert diag["ABC999111"]["smart_low_state"] == snapshot
    assert diag["ABC999111"]["smart_high_state"] == snapshot
    assert diag["ABC999111"]["pure_conf"] == snapshot

    def limit_attrs(prop, path):
        exclude_attrs = EXCLUDE_ATTRIBUTES
        return prop in exclude_attrs

    assert diag == snapshot(name="full_snapshot", exclude=limit_attrs)