evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve UniFi device tracker tests Pt2 (#120796)

Browse source
This commit is contained in:
Robert Svensson 2024-06-30 15:25:38 +02:00 • committed by GitHub
parent bb62a8a7dc
commit 289a630578
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 35 additions and 63 deletions
Download patch file Download diff file Expand all files Collapse all files

98
tests/components/unifi/test_device_tracker.py
View file

@ -500,64 +500,51 @@ async def test_wireless_client_go_wired_issue(
UniFi Network has a known issue that when a wireless device goes away it sometimes gets marked as wired. UniFi Network has a known issue that when a wireless device goes away it sometimes gets marked as wired.
""" """
client_payload.append( client_payload.append(
{ WIRELESS_CLIENT_1 | {"last_seen": dt_util.as_timestamp(dt_util.utcnow())}
"essid": "ssid",
"hostname": "client",
"ip": "10.0.0.1",
"is_wired": False,
"last_seen": dt_util.as_timestamp(dt_util.utcnow()),
"mac": "00:00:00:00:00:01",
}
) )
config_entry = await config_entry_factory() await config_entry_factory()
assert len(hass.states.async_entity_ids(TRACKER_DOMAIN)) == 1 assert len(hass.states.async_entity_ids(TRACKER_DOMAIN)) == 1
# Client is wireless # Client is wireless
client_state = hass.states.get("device_tracker.client") assert hass.states.get("device_tracker.ws_client_1").state == STATE_HOME
assert client_state.state == STATE_HOME
# Trigger wired bug # Trigger wired bug
client = client_payload[0] ws_client_1 = client_payload[0] | {
client["last_seen"] = dt_util.as_timestamp(dt_util.utcnow()) "last_seen": dt_util.as_timestamp(dt_util.utcnow()),
client["is_wired"] = True "is_wired": True,
mock_websocket_message(message=MessageKey.CLIENT, data=client) }
mock_websocket_message(message=MessageKey.CLIENT, data=ws_client_1)
await hass.async_block_till_done() await hass.async_block_till_done()
# Wired bug fix keeps client marked as wireless # Wired bug fix keeps client marked as wireless
client_state = hass.states.get("device_tracker.client") assert hass.states.get("device_tracker.ws_client_1").state == STATE_HOME
assert client_state.state == STATE_HOME
# Pass time # Pass time
new_time = dt_util.utcnow() + timedelta( new_time = dt_util.utcnow() + timedelta(seconds=DEFAULT_DETECTION_TIME)
seconds=(config_entry.options.get(CONF_DETECTION_TIME, DEFAULT_DETECTION_TIME))
)
with freeze_time(new_time): with freeze_time(new_time):
async_fire_time_changed(hass, new_time) async_fire_time_changed(hass, new_time)
await hass.async_block_till_done() await hass.async_block_till_done()
# Marked as home according to the timer # Marked as home according to the timer
client_state = hass.states.get("device_tracker.client") assert hass.states.get("device_tracker.ws_client_1").state == STATE_NOT_HOME
assert client_state.state == STATE_NOT_HOME
# Try to mark client as connected # Try to mark client as connected
client["last_seen"] += 1 ws_client_1["last_seen"] += 1
mock_websocket_message(message=MessageKey.CLIENT, data=client) mock_websocket_message(message=MessageKey.CLIENT, data=ws_client_1)
await hass.async_block_till_done() await hass.async_block_till_done()
# Make sure it don't go online again until wired bug disappears # Make sure it don't go online again until wired bug disappears
client_state = hass.states.get("device_tracker.client") assert hass.states.get("device_tracker.ws_client_1").state == STATE_NOT_HOME
assert client_state.state == STATE_NOT_HOME
# Make client wireless # Make client wireless
client["last_seen"] += 1 ws_client_1["last_seen"] += 1
client["is_wired"] = False ws_client_1["is_wired"] = False
mock_websocket_message(message=MessageKey.CLIENT, data=client) mock_websocket_message(message=MessageKey.CLIENT, data=ws_client_1)
await hass.async_block_till_done() await hass.async_block_till_done()
# Client is no longer affected by wired bug and can be marked online # Client is no longer affected by wired bug and can be marked online
client_state = hass.states.get("device_tracker.client") assert hass.states.get("device_tracker.ws_client_1").state == STATE_HOME
assert client_state.state == STATE_HOME
@pytest.mark.parametrize("config_entry_options", [{CONF_IGNORE_WIRED_BUG: True}]) @pytest.mark.parametrize("config_entry_options", [{CONF_IGNORE_WIRED_BUG: True}])
@ -570,64 +557,49 @@ async def test_option_ignore_wired_bug(
) -> None: ) -> None:
"""Test option to ignore wired bug.""" """Test option to ignore wired bug."""
client_payload.append( client_payload.append(
{ WIRELESS_CLIENT_1 | {"last_seen": dt_util.as_timestamp(dt_util.utcnow())}
"ap_mac": "00:00:00:00:02:01",
"essid": "ssid",
"hostname": "client",
"ip": "10.0.0.1",
"is_wired": False,
"last_seen": dt_util.as_timestamp(dt_util.utcnow()),
"mac": "00:00:00:00:00:01",
}
) )
config_entry = await config_entry_factory() await config_entry_factory()
assert len(hass.states.async_entity_ids(TRACKER_DOMAIN)) == 1 assert len(hass.states.async_entity_ids(TRACKER_DOMAIN)) == 1
# Client is wireless # Client is wireless
client_state = hass.states.get("device_tracker.client") assert hass.states.get("device_tracker.ws_client_1").state == STATE_HOME
assert client_state.state == STATE_HOME
# Trigger wired bug # Trigger wired bug
client = client_payload[0] ws_client_1 = client_payload[0]
client["is_wired"] = True ws_client_1["is_wired"] = True
mock_websocket_message(message=MessageKey.CLIENT, data=client) mock_websocket_message(message=MessageKey.CLIENT, data=ws_client_1)
await hass.async_block_till_done() await hass.async_block_till_done()
# Wired bug in effect # Wired bug in effect
client_state = hass.states.get("device_tracker.client") assert hass.states.get("device_tracker.ws_client_1").state == STATE_HOME
assert client_state.state == STATE_HOME
# pass time # Pass time
new_time = dt_util.utcnow() + timedelta( new_time = dt_util.utcnow() + timedelta(seconds=DEFAULT_DETECTION_TIME)
seconds=config_entry.options.get(CONF_DETECTION_TIME, DEFAULT_DETECTION_TIME)
)
with freeze_time(new_time): with freeze_time(new_time):
async_fire_time_changed(hass, new_time) async_fire_time_changed(hass, new_time)
await hass.async_block_till_done() await hass.async_block_till_done()
# Timer marks client as away # Timer marks client as away
client_state = hass.states.get("device_tracker.client") assert hass.states.get("device_tracker.ws_client_1").state == STATE_NOT_HOME
assert client_state.state == STATE_NOT_HOME
# Mark client as connected again # Mark client as connected again
client["last_seen"] += 1 ws_client_1["last_seen"] += 1
mock_websocket_message(message=MessageKey.CLIENT, data=client) mock_websocket_message(message=MessageKey.CLIENT, data=ws_client_1)
await hass.async_block_till_done() await hass.async_block_till_done()
# Ignoring wired bug allows client to go home again even while affected # Ignoring wired bug allows client to go home again even while affected
client_state = hass.states.get("device_tracker.client") assert hass.states.get("device_tracker.ws_client_1").state == STATE_HOME
assert client_state.state == STATE_HOME
# Make client wireless # Make client wireless
client["last_seen"] += 1 ws_client_1["last_seen"] += 1
client["is_wired"] = False ws_client_1["is_wired"] = False
mock_websocket_message(message=MessageKey.CLIENT, data=client) mock_websocket_message(message=MessageKey.CLIENT, data=ws_client_1)
await hass.async_block_till_done() await hass.async_block_till_done()
# Client is wireless and still connected # Client is wireless and still connected
client_state = hass.states.get("device_tracker.client") assert hass.states.get("device_tracker.ws_client_1").state == STATE_HOME
assert client_state.state == STATE_HOME
@pytest.mark.parametrize( @pytest.mark.parametrize(