mirror of
https://github.com/home-assistant/core.git
synced 2025-05-22 23:07:06 +00:00
9bff88ad3e
* Add diagnostics to fibaro * Enhance diagnostic test --------- Co-authored-by: Josef Zweck <josef@zweck.dev>
97 lines
2.8 KiB
Python
97 lines
2.8 KiB
Python
"""Tests for the diagnostics data provided by the fibaro integration."""
|
|
|
|
from unittest.mock import Mock
|
|
|
|
from syrupy import SnapshotAssertion
|
|
|
|
from homeassistant.components.fibaro import DOMAIN
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import device_registry as dr, entity_registry as er
|
|
|
|
from .conftest import TEST_SERIALNUMBER, init_integration
|
|
|
|
from tests.common import MockConfigEntry
|
|
from tests.components.diagnostics import (
|
|
get_diagnostics_for_config_entry,
|
|
get_diagnostics_for_device,
|
|
)
|
|
from tests.typing import ClientSessionGenerator
|
|
|
|
|
|
async def test_config_entry_diagnostics(
|
|
hass: HomeAssistant,
|
|
hass_client: ClientSessionGenerator,
|
|
mock_fibaro_client: Mock,
|
|
mock_config_entry: MockConfigEntry,
|
|
mock_light: Mock,
|
|
mock_room: Mock,
|
|
snapshot: SnapshotAssertion,
|
|
) -> None:
|
|
"""Test diagnostics."""
|
|
|
|
# Arrange
|
|
mock_fibaro_client.read_rooms.return_value = [mock_room]
|
|
mock_fibaro_client.read_devices.return_value = [mock_light]
|
|
# Act
|
|
await init_integration(hass, mock_config_entry)
|
|
# Assert
|
|
assert (
|
|
await get_diagnostics_for_config_entry(hass, hass_client, mock_config_entry)
|
|
== snapshot
|
|
)
|
|
|
|
|
|
async def test_device_diagnostics(
|
|
hass: HomeAssistant,
|
|
hass_client: ClientSessionGenerator,
|
|
mock_fibaro_client: Mock,
|
|
mock_config_entry: MockConfigEntry,
|
|
mock_light: Mock,
|
|
mock_room: Mock,
|
|
entity_registry: er.EntityRegistry,
|
|
device_registry: dr.DeviceRegistry,
|
|
snapshot: SnapshotAssertion,
|
|
) -> None:
|
|
"""Test diagnostics."""
|
|
|
|
# Arrange
|
|
mock_fibaro_client.read_rooms.return_value = [mock_room]
|
|
mock_fibaro_client.read_devices.return_value = [mock_light]
|
|
# Act
|
|
await init_integration(hass, mock_config_entry)
|
|
entry = entity_registry.async_get("light.room_1_test_light_3")
|
|
device = device_registry.async_get(entry.device_id)
|
|
# Assert
|
|
assert device
|
|
assert (
|
|
await get_diagnostics_for_device(hass, hass_client, mock_config_entry, device)
|
|
== snapshot
|
|
)
|
|
|
|
|
|
async def test_device_diagnostics_for_hub(
|
|
hass: HomeAssistant,
|
|
hass_client: ClientSessionGenerator,
|
|
mock_fibaro_client: Mock,
|
|
mock_config_entry: MockConfigEntry,
|
|
mock_light: Mock,
|
|
mock_power_sensor: Mock,
|
|
mock_room: Mock,
|
|
device_registry: dr.DeviceRegistry,
|
|
snapshot: SnapshotAssertion,
|
|
) -> None:
|
|
"""Test diagnostics for the hub."""
|
|
|
|
# Arrange
|
|
mock_fibaro_client.read_rooms.return_value = [mock_room]
|
|
mock_fibaro_client.read_devices.return_value = [mock_light, mock_power_sensor]
|
|
# Act
|
|
await init_integration(hass, mock_config_entry)
|
|
device = device_registry.async_get_device({(DOMAIN, TEST_SERIALNUMBER)})
|
|
# Assert
|
|
assert device
|
|
assert (
|
|
await get_diagnostics_for_device(hass, hass_client, mock_config_entry, device)
|
|
== snapshot
|
|
)
|