0cc80a9d29
* Add OUI to tplink diagnostics The main reason discovery does not work for new devices is we are missing the OUI. Since we redact the whole mac address in the diagnostics, this makes it difficult to fix. We now include the OUI in the diagnostics * fix: use cached mac * fix: tests
65 lines
2.1 KiB
Python
65 lines
2.1 KiB
Python
"""Tests for the diagnostics data provided by the TP-Link integration."""
|
|
import json
|
|
|
|
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
|
|
from tests.typing import ClientSessionGenerator
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("mocked_dev", "fixture_file", "sysinfo_vars", "expected_oui"),
|
|
[
|
|
(
|
|
_mocked_bulb(),
|
|
"tplink-diagnostics-data-bulb-kl130.json",
|
|
["mic_mac", "deviceId", "oemId", "hwId", "alias"],
|
|
"AA:BB:CC",
|
|
),
|
|
(
|
|
_mocked_plug(),
|
|
"tplink-diagnostics-data-plug-hs110.json",
|
|
["mac", "deviceId", "oemId", "hwId", "alias", "longitude_i", "latitude_i"],
|
|
"AA:BB:CC",
|
|
),
|
|
],
|
|
)
|
|
async def test_diagnostics(
|
|
hass: HomeAssistant,
|
|
hass_client: ClientSessionGenerator,
|
|
mocked_dev: SmartDevice,
|
|
fixture_file: str,
|
|
sysinfo_vars: list[str],
|
|
expected_oui: str | None,
|
|
):
|
|
"""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**"
|
|
|
|
assert result["oui"] == expected_oui
|