jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-20 19:57:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Clean up config option tests in UniFi device tracker tests (#119978)

Browse Source
This commit is contained in:
Robert Svensson 2024-06-19 19:42:23 +02:00 committed by GitHub
parent 6fa54229fe
commit 970836da0c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 110 additions and 361 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

471
tests/components/unifi/test_device_tracker.py
View File

@ -2,6 +2,7 @@
from collections.abc import Callable from collections.abc import Callable
from datetime import timedelta from datetime import timedelta
from types import MappingProxyType
from typing import Any from typing import Any
from aiounifi.models.message import MessageKey from aiounifi.models.message import MessageKey
@ -23,12 +24,47 @@ from homeassistant.components.unifi.const import (
) )
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import STATE_HOME, STATE_NOT_HOME, STATE_UNAVAILABLE from homeassistant.const import STATE_HOME, STATE_NOT_HOME, STATE_UNAVAILABLE
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, State
from homeassistant.helpers import entity_registry as er from homeassistant.helpers import entity_registry as er
import homeassistant.util.dt as dt_util import homeassistant.util.dt as dt_util
from tests.common import async_fire_time_changed from tests.common import async_fire_time_changed
WIRED_CLIENT_1 = {
"hostname": "wd_client_1",
"is_wired": True,
"last_seen": 1562600145,
"mac": "00:00:00:00:00:02",
}
WIRELESS_CLIENT_1 = {
"ap_mac": "00:00:00:00:02:01",
"essid": "ssid",
"hostname": "ws_client_1",
"ip": "10.0.0.1",
"is_wired": False,
"last_seen": 1562600145,
"mac": "00:00:00:00:00:01",
}
SWITCH_1 = {
"board_rev": 3,
"device_id": "mock-id-1",
"has_fan": True,
"fan_level": 0,
"ip": "10.0.1.1",
"last_seen": 1562600145,
"mac": "00:00:00:00:01:01",
"model": "US16P150",
"name": "Switch 1",
"next_interval": 20,
"overheating": True,
"state": 1,
"type": "usw",
"upgradable": True,
"version": "4.0.42.10433",
}
@pytest.mark.parametrize( @pytest.mark.parametrize(
"client_payload", "client_payload",
@ -496,213 +532,6 @@ async def test_hub_state_change(hass: HomeAssistant, mock_websocket_state) -> No
assert hass.states.get("device_tracker.device").state == STATE_HOME assert hass.states.get("device_tracker.device").state == STATE_HOME
@pytest.mark.parametrize(
"client_payload",
[
[
{
"essid": "ssid",
"hostname": "wireless_client",
"is_wired": False,
"last_seen": 1562600145,
"mac": "00:00:00:00:00:01",
},
{
"is_wired": True,
"last_seen": 1562600145,
"mac": "00:00:00:00:00:02",
"name": "Wired Client",
},
]
],
)
@pytest.mark.parametrize(
"device_payload",
[
[
{
"board_rev": 3,
"device_id": "mock-id",
"has_fan": True,
"fan_level": 0,
"ip": "10.0.1.1",
"last_seen": 1562600145,
"mac": "00:00:00:00:01:01",
"model": "US16P150",
"name": "Device",
"next_interval": 20,
"overheating": True,
"state": 1,
"type": "usw",
"upgradable": True,
"version": "4.0.42.10433",
}
]
],
)
@pytest.mark.usefixtures("mock_device_registry")
async def test_option_track_clients(
hass: HomeAssistant, config_entry_setup: ConfigEntry
) -> None:
"""Test the tracking of clients can be turned off."""
assert len(hass.states.async_entity_ids(TRACKER_DOMAIN)) == 3
assert hass.states.get("device_tracker.wireless_client")
assert hass.states.get("device_tracker.wired_client")
assert hass.states.get("device_tracker.device")
hass.config_entries.async_update_entry(
config_entry_setup, options={CONF_TRACK_CLIENTS: False}
)
await hass.async_block_till_done()
assert not hass.states.get("device_tracker.wireless_client")
assert not hass.states.get("device_tracker.wired_client")
assert hass.states.get("device_tracker.device")
hass.config_entries.async_update_entry(
config_entry_setup, options={CONF_TRACK_CLIENTS: True}
)
await hass.async_block_till_done()
assert hass.states.get("device_tracker.wireless_client")
assert hass.states.get("device_tracker.wired_client")
assert hass.states.get("device_tracker.device")
@pytest.mark.parametrize(
"client_payload",
[
[
{
"essid": "ssid",
"hostname": "wireless_client",
"is_wired": False,
"last_seen": 1562600145,
"mac": "00:00:00:00:00:01",
},
{
"is_wired": True,
"last_seen": 1562600145,
"mac": "00:00:00:00:00:02",
"name": "Wired Client",
},
]
],
)
@pytest.mark.parametrize(
"device_payload",
[
[
{
"board_rev": 3,
"device_id": "mock-id",
"has_fan": True,
"fan_level": 0,
"ip": "10.0.1.1",
"last_seen": 1562600145,
"mac": "00:00:00:00:01:01",
"model": "US16P150",
"name": "Device",
"next_interval": 20,
"overheating": True,
"state": 1,
"type": "usw",
"upgradable": True,
"version": "4.0.42.10433",
}
]
],
)
@pytest.mark.usefixtures("mock_device_registry")
async def test_option_track_wired_clients(
hass: HomeAssistant, config_entry_setup: ConfigEntry
) -> None:
"""Test the tracking of wired clients can be turned off."""
assert len(hass.states.async_entity_ids(TRACKER_DOMAIN)) == 3
assert hass.states.get("device_tracker.wireless_client")
assert hass.states.get("device_tracker.wired_client")
assert hass.states.get("device_tracker.device")
hass.config_entries.async_update_entry(
config_entry_setup, options={CONF_TRACK_WIRED_CLIENTS: False}
)
await hass.async_block_till_done()
assert hass.states.get("device_tracker.wireless_client")
assert not hass.states.get("device_tracker.wired_client")
assert hass.states.get("device_tracker.device")
hass.config_entries.async_update_entry(
config_entry_setup, options={CONF_TRACK_WIRED_CLIENTS: True}
)
await hass.async_block_till_done()
assert hass.states.get("device_tracker.wireless_client")
assert hass.states.get("device_tracker.wired_client")
assert hass.states.get("device_tracker.device")
@pytest.mark.parametrize(
"client_payload",
[
[
{
"hostname": "client",
"is_wired": True,
"last_seen": 1562600145,
"mac": "00:00:00:00:00:01",
}
]
],
)
@pytest.mark.parametrize(
"device_payload",
[
[
{
"board_rev": 3,
"device_id": "mock-id",
"last_seen": 1562600145,
"ip": "10.0.1.1",
"mac": "00:00:00:00:01:01",
"model": "US16P150",
"name": "Device",
"next_interval": 20,
"overheating": True,
"state": 1,
"type": "usw",
"upgradable": True,
"version": "4.0.42.10433",
}
]
],
)
@pytest.mark.usefixtures("mock_device_registry")
async def test_option_track_devices(
hass: HomeAssistant, config_entry_setup: ConfigEntry
) -> None:
"""Test the tracking of devices can be turned off."""
assert len(hass.states.async_entity_ids(TRACKER_DOMAIN)) == 2
assert hass.states.get("device_tracker.client")
assert hass.states.get("device_tracker.device")
hass.config_entries.async_update_entry(
config_entry_setup, options={CONF_TRACK_DEVICES: False}
)
await hass.async_block_till_done()
assert hass.states.get("device_tracker.client")
assert not hass.states.get("device_tracker.device")
hass.config_entries.async_update_entry(
config_entry_setup, options={CONF_TRACK_DEVICES: True}
)
await hass.async_block_till_done()
assert hass.states.get("device_tracker.client")
assert hass.states.get("device_tracker.device")
@pytest.mark.usefixtures("mock_device_registry") @pytest.mark.usefixtures("mock_device_registry")
async def test_option_ssid_filter( async def test_option_ssid_filter(
hass: HomeAssistant, hass: HomeAssistant,
@ -1031,166 +860,86 @@ async def test_restoring_client(
assert not hass.states.get("device_tracker.not_restored") assert not hass.states.get("device_tracker.not_restored")
@pytest.mark.parametrize("config_entry_options", [{CONF_TRACK_CLIENTS: False}])
@pytest.mark.parametrize( @pytest.mark.parametrize(
"client_payload", ("config_entry_options", "counts", "expected"),
[ [
[ (
{ {CONF_TRACK_CLIENTS: True},
"essid": "ssid", (3, 1),
"hostname": "Wireless client", ((True, True, True), (None, None, True)),
"ip": "10.0.0.1", ),
"is_wired": False, (
"last_seen": 1562600145, {CONF_TRACK_CLIENTS: True, CONF_SSID_FILTER: ["ssid"]},
"mac": "00:00:00:00:00:01", (3, 1),
}, ((True, True, True), (None, None, True)),
{ ),
"hostname": "Wired client", (
"ip": "10.0.0.2", {CONF_TRACK_CLIENTS: True, CONF_SSID_FILTER: ["ssid-2"]},
"is_wired": True, (2, 1),
"last_seen": 1562600145, ((None, True, True), (None, None, True)),
"mac": "00:00:00:00:00:02", ),
}, (
] {CONF_TRACK_CLIENTS: False, CONF_CLIENT_SOURCE: ["00:00:00:00:00:01"]},
], (2, 1),
) ((True, None, True), (None, None, True)),
@pytest.mark.parametrize( ),
"device_payload", (
[ {CONF_TRACK_CLIENTS: False, CONF_CLIENT_SOURCE: ["00:00:00:00:00:02"]},
[ (2, 1),
{ ((None, True, True), (None, None, True)),
"board_rev": 3, ),
"device_id": "mock-id", (
"has_fan": True, {CONF_TRACK_WIRED_CLIENTS: True},
"fan_level": 0, (3, 2),
"ip": "10.0.1.1", ((True, True, True), (True, None, True)),
"last_seen": 1562600145, ),
"mac": "00:00:00:00:01:01", (
"model": "US16P150", {CONF_TRACK_DEVICES: True},
"name": "Device", (3, 2),
"next_interval": 20, ((True, True, True), (True, True, None)),
"overheating": True, ),
"state": 1,
"type": "usw",
"upgradable": True,
"version": "4.0.42.10433",
},
]
], ],
) )
@pytest.mark.parametrize("client_payload", [[WIRELESS_CLIENT_1, WIRED_CLIENT_1]])
@pytest.mark.parametrize("device_payload", [[SWITCH_1]])
@pytest.mark.usefixtures("mock_device_registry") @pytest.mark.usefixtures("mock_device_registry")
async def test_dont_track_clients( async def test_config_entry_options_track(
hass: HomeAssistant, config_entry_setup: ConfigEntry hass: HomeAssistant,
config_entry_setup: ConfigEntry,
config_entry_options: MappingProxyType[str, Any],
counts: tuple[int],
expected: dict[tuple[bool | None]],
) -> None: ) -> None:
"""Test don't track clients config works.""" """Test the different config entry options.
assert len(hass.states.async_entity_ids(TRACKER_DOMAIN)) == 1
assert not hass.states.get("device_tracker.wireless_client")
assert not hass.states.get("device_tracker.wired_client")
assert hass.states.get("device_tracker.device")
hass.config_entries.async_update_entry( Validates how many entities are created
config_entry_setup, options={CONF_TRACK_CLIENTS: True} and that the specific ones exist as expected.
) """
option = next(iter(config_entry_options))
def assert_state(state: State | None, expected: bool | None):
"""Assert if state expected."""
assert state is None if expected is None else state
assert len(hass.states.async_entity_ids(TRACKER_DOMAIN)) == counts[0]
assert_state(hass.states.get("device_tracker.ws_client_1"), expected[0][0])
assert_state(hass.states.get("device_tracker.wd_client_1"), expected[0][1])
assert_state(hass.states.get("device_tracker.switch_1"), expected[0][2])
# Keep only the primary option and turn it off, everything else uses default
hass.config_entries.async_update_entry(config_entry_setup, options={option: False})
await hass.async_block_till_done()
assert len(hass.states.async_entity_ids(TRACKER_DOMAIN)) == counts[1]
assert_state(hass.states.get("device_tracker.ws_client_1"), expected[1][0])
assert_state(hass.states.get("device_tracker.wd_client_1"), expected[1][1])
assert_state(hass.states.get("device_tracker.switch_1"), expected[1][2])
# Turn on the primary option, everything else uses default
hass.config_entries.async_update_entry(config_entry_setup, options={option: True})
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(hass.states.async_entity_ids(TRACKER_DOMAIN)) == 3 assert len(hass.states.async_entity_ids(TRACKER_DOMAIN)) == 3
assert hass.states.get("device_tracker.wireless_client") assert_state(hass.states.get("device_tracker.ws_client_1"), True)
assert hass.states.get("device_tracker.wired_client") assert_state(hass.states.get("device_tracker.wd_client_1"), True)
assert hass.states.get("device_tracker.device") assert_state(hass.states.get("device_tracker.switch_1"), True)
@pytest.mark.parametrize("config_entry_options", [{CONF_TRACK_DEVICES: False}])
@pytest.mark.parametrize(
"client_payload",
[
[
{
"hostname": "client",
"is_wired": True,
"last_seen": 1562600145,
"mac": "00:00:00:00:00:01",
},
]
],
)
@pytest.mark.parametrize(
"device_payload",
[
[
{
"board_rev": 3,
"device_id": "mock-id",
"has_fan": True,
"fan_level": 0,
"ip": "10.0.1.1",
"last_seen": 1562600145,
"mac": "00:00:00:00:01:01",
"model": "US16P150",
"name": "Device",
"next_interval": 20,
"overheating": True,
"state": 1,
"type": "usw",
"upgradable": True,
"version": "4.0.42.10433",
},
]
],
)
@pytest.mark.usefixtures("mock_device_registry")
async def test_dont_track_devices(
hass: HomeAssistant, config_entry_setup: ConfigEntry
) -> None:
"""Test don't track devices config works."""
assert len(hass.states.async_entity_ids(TRACKER_DOMAIN)) == 1
assert hass.states.get("device_tracker.client")
assert not hass.states.get("device_tracker.device")
hass.config_entries.async_update_entry(
config_entry_setup, options={CONF_TRACK_DEVICES: True}
)
await hass.async_block_till_done()
assert len(hass.states.async_entity_ids(TRACKER_DOMAIN)) == 2
assert hass.states.get("device_tracker.client")
assert hass.states.get("device_tracker.device")
@pytest.mark.parametrize("config_entry_options", [{CONF_TRACK_WIRED_CLIENTS: False}])
@pytest.mark.parametrize(
"client_payload",
[
[
{
"essid": "ssid",
"hostname": "Wireless Client",
"is_wired": False,
"last_seen": 1562600145,
"mac": "00:00:00:00:00:01",
},
{
"is_wired": True,
"last_seen": 1562600145,
"mac": "00:00:00:00:00:02",
"name": "Wired Client",
},
]
],
)
@pytest.mark.usefixtures("mock_device_registry")
async def test_dont_track_wired_clients(
hass: HomeAssistant, config_entry_setup: ConfigEntry
) -> None:
"""Test don't track wired clients config works."""
assert len(hass.states.async_entity_ids(TRACKER_DOMAIN)) == 1
assert hass.states.get("device_tracker.wireless_client")
assert not hass.states.get("device_tracker.wired_client")
hass.config_entries.async_update_entry(
config_entry_setup, options={CONF_TRACK_WIRED_CLIENTS: True}
)
await hass.async_block_till_done()
assert len(hass.states.async_entity_ids(TRACKER_DOMAIN)) == 2
assert hass.states.get("device_tracker.wireless_client")
assert hass.states.get("device_tracker.wired_client")