mirror of
https://github.com/home-assistant/core
synced 2024-09-18 19:55:20 +02:00
61 lines
1.9 KiB
Python
61 lines
1.9 KiB
Python
"""Tests for the diagnostics data provided by the TP-Link integration."""
|
|
import json
|
|
|
|
from aiohttp import ClientSession
|
|
from kasa import SmartDevice
|
|
import pytest
|
|
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from . import _mocked_bulb, _mocked_plug, initialize_config_entry_for_device
|
|
|
|
from tests.common import load_fixture
|
|
from tests.components.diagnostics import get_diagnostics_for_config_entry
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"mocked_dev,fixture_file,sysinfo_vars",
|
|
[
|
|
(
|
|
_mocked_bulb(),
|
|
"tplink-diagnostics-data-bulb-kl130.json",
|
|
["mic_mac", "deviceId", "oemId", "hwId", "alias"],
|
|
),
|
|
(
|
|
_mocked_plug(),
|
|
"tplink-diagnostics-data-plug-hs110.json",
|
|
["mac", "deviceId", "oemId", "hwId", "alias", "longitude_i", "latitude_i"],
|
|
),
|
|
],
|
|
)
|
|
async def test_diagnostics(
|
|
hass: HomeAssistant,
|
|
hass_client: ClientSession,
|
|
mocked_dev: SmartDevice,
|
|
fixture_file: str,
|
|
sysinfo_vars: list[str],
|
|
):
|
|
"""Test diagnostics for config entry."""
|
|
diagnostics_data = json.loads(load_fixture(fixture_file, "tplink"))
|
|
|
|
mocked_dev.internal_state = diagnostics_data["device_last_response"]
|
|
|
|
config_entry = await initialize_config_entry_for_device(hass, mocked_dev)
|
|
result = await get_diagnostics_for_config_entry(hass, hass_client, config_entry)
|
|
|
|
assert isinstance(result, dict)
|
|
assert "device_last_response" in result
|
|
|
|
# There must be some redactions in place, so the raw data must not match
|
|
assert result["device_last_response"] != diagnostics_data["device_last_response"]
|
|
|
|
last_response = result["device_last_response"]
|
|
|
|
# We should always have sysinfo available
|
|
assert "system" in last_response
|
|
assert "get_sysinfo" in last_response["system"]
|
|
|
|
sysinfo = last_response["system"]["get_sysinfo"]
|
|
for var in sysinfo_vars:
|
|
assert sysinfo[var] == "**REDACTED**"
|