core/tests/components/asuswrt/test_diagnostics.py

42 lines
1.4 KiB
Python

"""Tests for the diagnostics data provided by the AsusWRT integration."""
from homeassistant.components.asuswrt.const import DOMAIN
from homeassistant.components.asuswrt.diagnostics import TO_REDACT
from homeassistant.components.device_tracker import CONF_CONSIDER_HOME
from homeassistant.components.diagnostics import async_redact_data
from homeassistant.config_entries import ConfigEntryState
from homeassistant.core import HomeAssistant
from .common import CONFIG_DATA_TELNET, ROUTER_MAC_ADDR
from tests.common import MockConfigEntry
from tests.components.diagnostics import get_diagnostics_for_config_entry
from tests.typing import ClientSessionGenerator
async def test_diagnostics(
hass: HomeAssistant,
hass_client: ClientSessionGenerator,
connect_legacy,
) -> None:
"""Test diagnostics."""
mock_config_entry = MockConfigEntry(
domain=DOMAIN,
data=CONFIG_DATA_TELNET,
options={CONF_CONSIDER_HOME: 60},
unique_id=ROUTER_MAC_ADDR,
)
mock_config_entry.add_to_hass(hass)
await hass.config_entries.async_setup(mock_config_entry.entry_id)
await hass.async_block_till_done()
assert mock_config_entry.state is ConfigEntryState.LOADED
entry_dict = async_redact_data(mock_config_entry.as_dict(), TO_REDACT)
result = await get_diagnostics_for_config_entry(
hass, hass_client, mock_config_entry
)
assert result["entry"] == entry_dict | {"discovery_keys": {}}