39 lines
1.0 KiB
Python
39 lines
1.0 KiB
Python
"""Tests for Tradfri diagnostics."""
|
|
from unittest.mock import MagicMock, Mock
|
|
|
|
from aiohttp import ClientSession
|
|
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from .common import setup_integration
|
|
from .test_fan import mock_fan
|
|
|
|
from tests.components.diagnostics import get_diagnostics_for_config_entry
|
|
|
|
|
|
async def test_diagnostics(
|
|
hass: HomeAssistant,
|
|
hass_client: ClientSession,
|
|
mock_gateway: Mock,
|
|
mock_api_factory: MagicMock,
|
|
) -> None:
|
|
"""Test diagnostics for config entry."""
|
|
mock_gateway.mock_devices.append(
|
|
# Add a fan
|
|
mock_fan(
|
|
test_state={
|
|
"fan_speed": 10,
|
|
"air_quality": 42,
|
|
"filter_lifetime_remaining": 120,
|
|
}
|
|
)
|
|
)
|
|
|
|
init_integration = await setup_integration(hass)
|
|
|
|
result = await get_diagnostics_for_config_entry(hass, hass_client, init_integration)
|
|
|
|
assert isinstance(result, dict)
|
|
assert result["gateway_version"] == "1.2.1234"
|
|
assert len(result["device_data"]) == 1
|