Make zone dependency of device tracker an after dep (#32880)

* Make zone dependency of device tracker an after dep

* Fix test
pull/32932/head
Paulus Schoutsen 2020-03-17 03:59:39 -07:00
parent 912409ed0c
commit 67d728fc50
5 changed files with 35 additions and 33 deletions

View File

@ -3,8 +3,8 @@
"name": "Device Tracker", "name": "Device Tracker",
"documentation": "https://www.home-assistant.io/integrations/device_tracker", "documentation": "https://www.home-assistant.io/integrations/device_tracker",
"requirements": [], "requirements": [],
"dependencies": ["zone"], "dependencies": [],
"after_dependencies": [], "after_dependencies": ["zone"],
"codeowners": [], "codeowners": [],
"quality_scale": "internal" "quality_scale": "internal"
} }

View File

@ -160,8 +160,10 @@ async def test_webhook_handle_get_zones(hass, create_registrations, webhook_clie
assert resp.status == 200 assert resp.status == 200
json = await resp.json() json = await resp.json()
assert len(json) == 1 assert len(json) == 2
assert json[0]["entity_id"] == "zone.home" zones = sorted(json, key=lambda entry: entry["entity_id"])
assert zones[0]["entity_id"] == "zone.home"
assert zones[1]["entity_id"] == "zone.test"
async def test_webhook_handle_get_config(hass, create_registrations, webhook_client): async def test_webhook_handle_get_config(hass, create_registrations, webhook_client):

View File

@ -108,7 +108,7 @@ async def test_no_clients(hass):
"""Test the update_clients function when no clients are found.""" """Test the update_clients function when no clients are found."""
await setup_unifi_integration(hass) await setup_unifi_integration(hass)
assert len(hass.states.async_all()) == 1 assert len(hass.states.async_all()) == 0
async def test_tracked_devices(hass): async def test_tracked_devices(hass):
@ -123,7 +123,7 @@ async def test_tracked_devices(hass):
devices_response=[DEVICE_1, DEVICE_2], devices_response=[DEVICE_1, DEVICE_2],
known_wireless_clients=(CLIENT_4["mac"],), known_wireless_clients=(CLIENT_4["mac"],),
) )
assert len(hass.states.async_all()) == 7 assert len(hass.states.async_all()) == 6
client_1 = hass.states.get("device_tracker.client_1") client_1 = hass.states.get("device_tracker.client_1")
assert client_1 is not None assert client_1 is not None
@ -184,7 +184,7 @@ async def test_controller_state_change(hass):
controller = await setup_unifi_integration( controller = await setup_unifi_integration(
hass, clients_response=[CLIENT_1], devices_response=[DEVICE_1], hass, clients_response=[CLIENT_1], devices_response=[DEVICE_1],
) )
assert len(hass.states.async_all()) == 3 assert len(hass.states.async_all()) == 2
# Controller unavailable # Controller unavailable
controller.async_unifi_signalling_callback( controller.async_unifi_signalling_callback(
@ -214,7 +214,7 @@ async def test_option_track_clients(hass):
controller = await setup_unifi_integration( controller = await setup_unifi_integration(
hass, clients_response=[CLIENT_1, CLIENT_2], devices_response=[DEVICE_1], hass, clients_response=[CLIENT_1, CLIENT_2], devices_response=[DEVICE_1],
) )
assert len(hass.states.async_all()) == 4 assert len(hass.states.async_all()) == 3
client_1 = hass.states.get("device_tracker.client_1") client_1 = hass.states.get("device_tracker.client_1")
assert client_1 is not None assert client_1 is not None
@ -259,7 +259,7 @@ async def test_option_track_wired_clients(hass):
controller = await setup_unifi_integration( controller = await setup_unifi_integration(
hass, clients_response=[CLIENT_1, CLIENT_2], devices_response=[DEVICE_1], hass, clients_response=[CLIENT_1, CLIENT_2], devices_response=[DEVICE_1],
) )
assert len(hass.states.async_all()) == 4 assert len(hass.states.async_all()) == 3
client_1 = hass.states.get("device_tracker.client_1") client_1 = hass.states.get("device_tracker.client_1")
assert client_1 is not None assert client_1 is not None
@ -304,7 +304,7 @@ async def test_option_track_devices(hass):
controller = await setup_unifi_integration( controller = await setup_unifi_integration(
hass, clients_response=[CLIENT_1, CLIENT_2], devices_response=[DEVICE_1], hass, clients_response=[CLIENT_1, CLIENT_2], devices_response=[DEVICE_1],
) )
assert len(hass.states.async_all()) == 4 assert len(hass.states.async_all()) == 3
client_1 = hass.states.get("device_tracker.client_1") client_1 = hass.states.get("device_tracker.client_1")
assert client_1 is not None assert client_1 is not None
@ -349,7 +349,7 @@ async def test_option_ssid_filter(hass):
controller = await setup_unifi_integration( controller = await setup_unifi_integration(
hass, options={CONF_SSID_FILTER: ["ssid"]}, clients_response=[CLIENT_3], hass, options={CONF_SSID_FILTER: ["ssid"]}, clients_response=[CLIENT_3],
) )
assert len(hass.states.async_all()) == 2 assert len(hass.states.async_all()) == 1
# SSID filter active # SSID filter active
client_3 = hass.states.get("device_tracker.client_3") client_3 = hass.states.get("device_tracker.client_3")
@ -387,7 +387,7 @@ async def test_wireless_client_go_wired_issue(hass):
client_1_client["last_seen"] = dt_util.as_timestamp(dt_util.utcnow()) client_1_client["last_seen"] = dt_util.as_timestamp(dt_util.utcnow())
controller = await setup_unifi_integration(hass, clients_response=[client_1_client]) controller = await setup_unifi_integration(hass, clients_response=[client_1_client])
assert len(hass.states.async_all()) == 2 assert len(hass.states.async_all()) == 1
client_1 = hass.states.get("device_tracker.client_1") client_1 = hass.states.get("device_tracker.client_1")
assert client_1 is not None assert client_1 is not None
@ -460,7 +460,7 @@ async def test_restoring_client(hass):
clients_response=[CLIENT_2], clients_response=[CLIENT_2],
clients_all_response=[CLIENT_1], clients_all_response=[CLIENT_1],
) )
assert len(hass.states.async_all()) == 3 assert len(hass.states.async_all()) == 2
device_1 = hass.states.get("device_tracker.client_1") device_1 = hass.states.get("device_tracker.client_1")
assert device_1 is not None assert device_1 is not None
@ -474,7 +474,7 @@ async def test_dont_track_clients(hass):
clients_response=[CLIENT_1], clients_response=[CLIENT_1],
devices_response=[DEVICE_1], devices_response=[DEVICE_1],
) )
assert len(hass.states.async_all()) == 2 assert len(hass.states.async_all()) == 1
client_1 = hass.states.get("device_tracker.client_1") client_1 = hass.states.get("device_tracker.client_1")
assert client_1 is None assert client_1 is None
@ -492,7 +492,7 @@ async def test_dont_track_devices(hass):
clients_response=[CLIENT_1], clients_response=[CLIENT_1],
devices_response=[DEVICE_1], devices_response=[DEVICE_1],
) )
assert len(hass.states.async_all()) == 2 assert len(hass.states.async_all()) == 1
client_1 = hass.states.get("device_tracker.client_1") client_1 = hass.states.get("device_tracker.client_1")
assert client_1 is not None assert client_1 is not None
@ -509,7 +509,7 @@ async def test_dont_track_wired_clients(hass):
options={unifi.controller.CONF_TRACK_WIRED_CLIENTS: False}, options={unifi.controller.CONF_TRACK_WIRED_CLIENTS: False},
clients_response=[CLIENT_1, CLIENT_2], clients_response=[CLIENT_1, CLIENT_2],
) )
assert len(hass.states.async_all()) == 2 assert len(hass.states.async_all()) == 1
client_1 = hass.states.get("device_tracker.client_1") client_1 = hass.states.get("device_tracker.client_1")
assert client_1 is not None assert client_1 is not None

View File

@ -55,7 +55,7 @@ async def test_no_clients(hass):
) )
assert len(controller.mock_requests) == 4 assert len(controller.mock_requests) == 4
assert len(hass.states.async_all()) == 1 assert len(hass.states.async_all()) == 0
async def test_sensors(hass): async def test_sensors(hass):
@ -71,7 +71,7 @@ async def test_sensors(hass):
) )
assert len(controller.mock_requests) == 4 assert len(controller.mock_requests) == 4
assert len(hass.states.async_all()) == 5 assert len(hass.states.async_all()) == 4
wired_client_rx = hass.states.get("sensor.wired_client_name_rx") wired_client_rx = hass.states.get("sensor.wired_client_name_rx")
assert wired_client_rx.state == "1234.0" assert wired_client_rx.state == "1234.0"

View File

@ -209,7 +209,7 @@ async def test_no_clients(hass):
) )
assert len(controller.mock_requests) == 4 assert len(controller.mock_requests) == 4
assert len(hass.states.async_all()) == 1 assert len(hass.states.async_all()) == 0
async def test_controller_not_client(hass): async def test_controller_not_client(hass):
@ -222,7 +222,7 @@ async def test_controller_not_client(hass):
) )
assert len(controller.mock_requests) == 4 assert len(controller.mock_requests) == 4
assert len(hass.states.async_all()) == 1 assert len(hass.states.async_all()) == 0
cloudkey = hass.states.get("switch.cloud_key") cloudkey = hass.states.get("switch.cloud_key")
assert cloudkey is None assert cloudkey is None
@ -240,7 +240,7 @@ async def test_not_admin(hass):
) )
assert len(controller.mock_requests) == 4 assert len(controller.mock_requests) == 4
assert len(hass.states.async_all()) == 1 assert len(hass.states.async_all()) == 0
async def test_switches(hass): async def test_switches(hass):
@ -258,7 +258,7 @@ async def test_switches(hass):
) )
assert len(controller.mock_requests) == 4 assert len(controller.mock_requests) == 4
assert len(hass.states.async_all()) == 4 assert len(hass.states.async_all()) == 3
switch_1 = hass.states.get("switch.poe_client_1") switch_1 = hass.states.get("switch.poe_client_1")
assert switch_1 is not None assert switch_1 is not None
@ -312,7 +312,7 @@ async def test_new_client_discovered_on_block_control(hass):
) )
assert len(controller.mock_requests) == 4 assert len(controller.mock_requests) == 4
assert len(hass.states.async_all()) == 1 assert len(hass.states.async_all()) == 0
blocked = hass.states.get("switch.block_client_1") blocked = hass.states.get("switch.block_client_1")
assert blocked is None assert blocked is None
@ -324,7 +324,7 @@ async def test_new_client_discovered_on_block_control(hass):
controller.api.session_handler("data") controller.api.session_handler("data")
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(hass.states.async_all()) == 2 assert len(hass.states.async_all()) == 1
blocked = hass.states.get("switch.block_client_1") blocked = hass.states.get("switch.block_client_1")
assert blocked is not None assert blocked is not None
@ -336,7 +336,7 @@ async def test_option_block_clients(hass):
options={CONF_BLOCK_CLIENT: [BLOCKED["mac"]]}, options={CONF_BLOCK_CLIENT: [BLOCKED["mac"]]},
clients_all_response=[BLOCKED, UNBLOCKED], clients_all_response=[BLOCKED, UNBLOCKED],
) )
assert len(hass.states.async_all()) == 2 assert len(hass.states.async_all()) == 1
# Add a second switch # Add a second switch
hass.config_entries.async_update_entry( hass.config_entries.async_update_entry(
@ -344,28 +344,28 @@ async def test_option_block_clients(hass):
options={CONF_BLOCK_CLIENT: [BLOCKED["mac"], UNBLOCKED["mac"]]}, options={CONF_BLOCK_CLIENT: [BLOCKED["mac"], UNBLOCKED["mac"]]},
) )
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(hass.states.async_all()) == 3 assert len(hass.states.async_all()) == 2
# Remove the second switch again # Remove the second switch again
hass.config_entries.async_update_entry( hass.config_entries.async_update_entry(
controller.config_entry, options={CONF_BLOCK_CLIENT: [BLOCKED["mac"]]}, controller.config_entry, options={CONF_BLOCK_CLIENT: [BLOCKED["mac"]]},
) )
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(hass.states.async_all()) == 2 assert len(hass.states.async_all()) == 1
# Enable one and remove another one # Enable one and remove another one
hass.config_entries.async_update_entry( hass.config_entries.async_update_entry(
controller.config_entry, options={CONF_BLOCK_CLIENT: [UNBLOCKED["mac"]]}, controller.config_entry, options={CONF_BLOCK_CLIENT: [UNBLOCKED["mac"]]},
) )
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(hass.states.async_all()) == 2 assert len(hass.states.async_all()) == 1
# Remove one # Remove one
hass.config_entries.async_update_entry( hass.config_entries.async_update_entry(
controller.config_entry, options={CONF_BLOCK_CLIENT: []}, controller.config_entry, options={CONF_BLOCK_CLIENT: []},
) )
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(hass.states.async_all()) == 1 assert len(hass.states.async_all()) == 0
async def test_new_client_discovered_on_poe_control(hass): async def test_new_client_discovered_on_poe_control(hass):
@ -378,7 +378,7 @@ async def test_new_client_discovered_on_poe_control(hass):
) )
assert len(controller.mock_requests) == 4 assert len(controller.mock_requests) == 4
assert len(hass.states.async_all()) == 2 assert len(hass.states.async_all()) == 1
controller.api.websocket._data = { controller.api.websocket._data = {
"meta": {"message": "sta:sync"}, "meta": {"message": "sta:sync"},
@ -391,7 +391,7 @@ async def test_new_client_discovered_on_poe_control(hass):
"switch", "turn_off", {"entity_id": "switch.poe_client_1"}, blocking=True "switch", "turn_off", {"entity_id": "switch.poe_client_1"}, blocking=True
) )
assert len(controller.mock_requests) == 5 assert len(controller.mock_requests) == 5
assert len(hass.states.async_all()) == 3 assert len(hass.states.async_all()) == 2
assert controller.mock_requests[4] == { assert controller.mock_requests[4] == {
"json": { "json": {
"port_overrides": [{"port_idx": 1, "portconf_id": "1a1", "poe_mode": "off"}] "port_overrides": [{"port_idx": 1, "portconf_id": "1a1", "poe_mode": "off"}]
@ -430,7 +430,7 @@ async def test_ignore_multiple_poe_clients_on_same_port(hass):
) )
assert len(controller.mock_requests) == 4 assert len(controller.mock_requests) == 4
assert len(hass.states.async_all()) == 4 assert len(hass.states.async_all()) == 3
switch_1 = hass.states.get("switch.poe_client_1") switch_1 = hass.states.get("switch.poe_client_1")
switch_2 = hass.states.get("switch.poe_client_2") switch_2 = hass.states.get("switch.poe_client_2")
@ -481,7 +481,7 @@ async def test_restoring_client(hass):
) )
assert len(controller.mock_requests) == 4 assert len(controller.mock_requests) == 4
assert len(hass.states.async_all()) == 3 assert len(hass.states.async_all()) == 2
device_1 = hass.states.get("switch.client_1") device_1 = hass.states.get("switch.client_1")
assert device_1 is not None assert device_1 is not None