core/tests/components/fully_kiosk/test_diagnostics.py

43 lines
1.4 KiB
Python

"""Test the Fully Kiosk Browser diagnostics."""
from unittest.mock import MagicMock
from homeassistant.components.diagnostics import REDACTED
from homeassistant.components.fully_kiosk.const import DOMAIN
from homeassistant.components.fully_kiosk.diagnostics import (
DEVICE_INFO_TO_REDACT,
SETTINGS_TO_REDACT,
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr
from tests.common import MockConfigEntry
from tests.components.diagnostics import get_diagnostics_for_device
from tests.typing import ClientSessionGenerator
async def test_diagnostics(
hass: HomeAssistant,
device_registry: dr.DeviceRegistry,
hass_client: ClientSessionGenerator,
mock_fully_kiosk: MagicMock,
init_integration: MockConfigEntry,
) -> None:
"""Test Fully Kiosk diagnostics."""
device = device_registry.async_get_device(identifiers={(DOMAIN, "abcdef-123456")})
diagnostics = await get_diagnostics_for_device(
hass, hass_client, init_integration, device
)
assert diagnostics
for key in DEVICE_INFO_TO_REDACT:
if hasattr(diagnostics, key):
assert diagnostics[key] == REDACTED
for key in SETTINGS_TO_REDACT:
if hasattr(diagnostics["settings"], key):
assert (
diagnostics["settings"][key] == REDACTED
or diagnostics["settings"][key] == ""
)