"""Test WiZ diagnostics."""
from homeassistant.core import HomeAssistant

from . import async_setup_integration

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


async def test_diagnostics(
    hass: HomeAssistant, hass_client: ClientSessionGenerator
) -> None:
    """Test generating diagnostics for a config entry."""
    _, entry = await async_setup_integration(hass)
    diag = await get_diagnostics_for_config_entry(hass, hass_client, entry)

    assert diag == {
        "data": {
            "homeId": "**REDACTED**",
            "mocked": "mocked",
            "roomId": "**REDACTED**",
        },
        "entry": {"data": {"host": "1.1.1.1"}, "title": "Mock Title"},
    }