672 lines
20 KiB
Python
672 lines
20 KiB
Python
"""UniFi switch platform tests."""
|
|
from copy import deepcopy
|
|
|
|
from aiounifi.controller import MESSAGE_CLIENT_REMOVED, MESSAGE_EVENT
|
|
from aiounifi.websocket import SIGNAL_DATA
|
|
|
|
from homeassistant import config_entries
|
|
from homeassistant.components.device_tracker import DOMAIN as TRACKER_DOMAIN
|
|
from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN
|
|
from homeassistant.components.unifi.const import (
|
|
CONF_BLOCK_CLIENT,
|
|
CONF_TRACK_CLIENTS,
|
|
CONF_TRACK_DEVICES,
|
|
DOMAIN as UNIFI_DOMAIN,
|
|
)
|
|
from homeassistant.components.unifi.switch import POE_SWITCH
|
|
from homeassistant.helpers import entity_registry
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from .test_controller import (
|
|
CONTROLLER_HOST,
|
|
DESCRIPTION,
|
|
ENTRY_CONFIG,
|
|
setup_unifi_integration,
|
|
)
|
|
|
|
CLIENT_1 = {
|
|
"hostname": "client_1",
|
|
"ip": "10.0.0.1",
|
|
"is_wired": True,
|
|
"last_seen": 1562600145,
|
|
"mac": "00:00:00:00:00:01",
|
|
"name": "POE Client 1",
|
|
"oui": "Producer",
|
|
"sw_mac": "00:00:00:00:01:01",
|
|
"sw_port": 1,
|
|
"wired-rx_bytes": 1234000000,
|
|
"wired-tx_bytes": 5678000000,
|
|
}
|
|
CLIENT_2 = {
|
|
"hostname": "client_2",
|
|
"ip": "10.0.0.2",
|
|
"is_wired": True,
|
|
"last_seen": 1562600145,
|
|
"mac": "00:00:00:00:00:02",
|
|
"name": "POE Client 2",
|
|
"oui": "Producer",
|
|
"sw_mac": "00:00:00:00:01:01",
|
|
"sw_port": 2,
|
|
"wired-rx_bytes": 1234000000,
|
|
"wired-tx_bytes": 5678000000,
|
|
}
|
|
CLIENT_3 = {
|
|
"hostname": "client_3",
|
|
"ip": "10.0.0.3",
|
|
"is_wired": True,
|
|
"last_seen": 1562600145,
|
|
"mac": "00:00:00:00:00:03",
|
|
"name": "Non-POE Client 3",
|
|
"oui": "Producer",
|
|
"sw_mac": "00:00:00:00:01:01",
|
|
"sw_port": 3,
|
|
"wired-rx_bytes": 1234000000,
|
|
"wired-tx_bytes": 5678000000,
|
|
}
|
|
CLIENT_4 = {
|
|
"hostname": "client_4",
|
|
"ip": "10.0.0.4",
|
|
"is_wired": True,
|
|
"last_seen": 1562600145,
|
|
"mac": "00:00:00:00:00:04",
|
|
"name": "Non-POE Client 4",
|
|
"oui": "Producer",
|
|
"sw_mac": "00:00:00:00:01:01",
|
|
"sw_port": 4,
|
|
"wired-rx_bytes": 1234000000,
|
|
"wired-tx_bytes": 5678000000,
|
|
}
|
|
POE_SWITCH_CLIENTS = [
|
|
{
|
|
"hostname": "client_1",
|
|
"ip": "10.0.0.1",
|
|
"is_wired": True,
|
|
"last_seen": 1562600145,
|
|
"mac": "00:00:00:00:00:01",
|
|
"name": "POE Client 1",
|
|
"oui": "Producer",
|
|
"sw_mac": "00:00:00:00:01:01",
|
|
"sw_port": 1,
|
|
"wired-rx_bytes": 1234000000,
|
|
"wired-tx_bytes": 5678000000,
|
|
},
|
|
{
|
|
"hostname": "client_2",
|
|
"ip": "10.0.0.2",
|
|
"is_wired": True,
|
|
"last_seen": 1562600145,
|
|
"mac": "00:00:00:00:00:02",
|
|
"name": "POE Client 2",
|
|
"oui": "Producer",
|
|
"sw_mac": "00:00:00:00:01:01",
|
|
"sw_port": 1,
|
|
"wired-rx_bytes": 1234000000,
|
|
"wired-tx_bytes": 5678000000,
|
|
},
|
|
]
|
|
|
|
DEVICE_1 = {
|
|
"device_id": "mock-id",
|
|
"ip": "10.0.1.1",
|
|
"mac": "00:00:00:00:01:01",
|
|
"last_seen": 1562600145,
|
|
"model": "US16P150",
|
|
"name": "mock-name",
|
|
"port_overrides": [],
|
|
"port_table": [
|
|
{
|
|
"media": "GE",
|
|
"name": "Port 1",
|
|
"port_idx": 1,
|
|
"poe_class": "Class 4",
|
|
"poe_enable": True,
|
|
"poe_mode": "auto",
|
|
"poe_power": "2.56",
|
|
"poe_voltage": "53.40",
|
|
"portconf_id": "1a1",
|
|
"port_poe": True,
|
|
"up": True,
|
|
},
|
|
{
|
|
"media": "GE",
|
|
"name": "Port 2",
|
|
"port_idx": 2,
|
|
"poe_class": "Class 4",
|
|
"poe_enable": True,
|
|
"poe_mode": "auto",
|
|
"poe_power": "2.56",
|
|
"poe_voltage": "53.40",
|
|
"portconf_id": "1a2",
|
|
"port_poe": True,
|
|
"up": True,
|
|
},
|
|
{
|
|
"media": "GE",
|
|
"name": "Port 3",
|
|
"port_idx": 3,
|
|
"poe_class": "Unknown",
|
|
"poe_enable": False,
|
|
"poe_mode": "off",
|
|
"poe_power": "0.00",
|
|
"poe_voltage": "0.00",
|
|
"portconf_id": "1a3",
|
|
"port_poe": False,
|
|
"up": True,
|
|
},
|
|
{
|
|
"media": "GE",
|
|
"name": "Port 4",
|
|
"port_idx": 4,
|
|
"poe_class": "Unknown",
|
|
"poe_enable": False,
|
|
"poe_mode": "auto",
|
|
"poe_power": "0.00",
|
|
"poe_voltage": "0.00",
|
|
"portconf_id": "1a4",
|
|
"port_poe": True,
|
|
"up": True,
|
|
},
|
|
],
|
|
"state": 1,
|
|
"type": "usw",
|
|
"version": "4.0.42.10433",
|
|
}
|
|
|
|
BLOCKED = {
|
|
"blocked": True,
|
|
"hostname": "block_client_1",
|
|
"ip": "10.0.0.1",
|
|
"is_guest": False,
|
|
"is_wired": False,
|
|
"last_seen": 1562600145,
|
|
"mac": "00:00:00:00:01:01",
|
|
"name": "Block Client 1",
|
|
"noted": True,
|
|
"oui": "Producer",
|
|
}
|
|
UNBLOCKED = {
|
|
"blocked": False,
|
|
"hostname": "block_client_2",
|
|
"ip": "10.0.0.2",
|
|
"is_guest": False,
|
|
"is_wired": True,
|
|
"last_seen": 1562600145,
|
|
"mac": "00:00:00:00:01:02",
|
|
"name": "Block Client 2",
|
|
"noted": True,
|
|
"oui": "Producer",
|
|
}
|
|
|
|
EVENT_BLOCKED_CLIENT_CONNECTED = {
|
|
"user": BLOCKED["mac"],
|
|
"radio": "na",
|
|
"channel": "44",
|
|
"hostname": BLOCKED["hostname"],
|
|
"key": "EVT_WU_Connected",
|
|
"subsystem": "wlan",
|
|
"site_id": "name",
|
|
"time": 1587753456179,
|
|
"datetime": "2020-04-24T18:37:36Z",
|
|
"msg": f'User{[BLOCKED["mac"]]} has connected."',
|
|
"_id": "5ea331fa30c49e00f90ddc1a",
|
|
}
|
|
|
|
EVENT_BLOCKED_CLIENT_BLOCKED = {
|
|
"user": BLOCKED["mac"],
|
|
"hostname": BLOCKED["hostname"],
|
|
"key": "EVT_WC_Blocked",
|
|
"subsystem": "wlan",
|
|
"site_id": "name",
|
|
"time": 1587753456179,
|
|
"datetime": "2020-04-24T18:37:36Z",
|
|
"msg": f'User{[BLOCKED["mac"]]} has been blocked."',
|
|
"_id": "5ea331fa30c49e00f90ddc1a",
|
|
}
|
|
|
|
EVENT_BLOCKED_CLIENT_UNBLOCKED = {
|
|
"user": BLOCKED["mac"],
|
|
"hostname": BLOCKED["hostname"],
|
|
"key": "EVT_WC_Unblocked",
|
|
"subsystem": "wlan",
|
|
"site_id": "name",
|
|
"time": 1587753456179,
|
|
"datetime": "2020-04-24T18:37:36Z",
|
|
"msg": f'User{[BLOCKED["mac"]]} has been unblocked."',
|
|
"_id": "5ea331fa30c49e00f90ddc1a",
|
|
}
|
|
|
|
|
|
EVENT_CLIENT_2_CONNECTED = {
|
|
"user": CLIENT_2["mac"],
|
|
"radio": "na",
|
|
"channel": "44",
|
|
"hostname": CLIENT_2["hostname"],
|
|
"key": "EVT_WU_Connected",
|
|
"subsystem": "wlan",
|
|
"site_id": "name",
|
|
"time": 1587753456179,
|
|
"datetime": "2020-04-24T18:37:36Z",
|
|
"msg": f'User{[CLIENT_2["mac"]]} has connected."',
|
|
"_id": "5ea331fa30c49e00f90ddc1a",
|
|
}
|
|
|
|
|
|
async def test_platform_manually_configured(hass):
|
|
"""Test that we do not discover anything or try to set up a controller."""
|
|
assert (
|
|
await async_setup_component(
|
|
hass, SWITCH_DOMAIN, {SWITCH_DOMAIN: {"platform": UNIFI_DOMAIN}}
|
|
)
|
|
is True
|
|
)
|
|
assert UNIFI_DOMAIN not in hass.data
|
|
|
|
|
|
async def test_no_clients(hass):
|
|
"""Test the update_clients function when no clients are found."""
|
|
controller = await setup_unifi_integration(
|
|
hass,
|
|
options={CONF_TRACK_CLIENTS: False, CONF_TRACK_DEVICES: False},
|
|
)
|
|
|
|
assert len(controller.mock_requests) == 4
|
|
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 0
|
|
|
|
|
|
async def test_controller_not_client(hass):
|
|
"""Test that the controller doesn't become a switch."""
|
|
controller = await setup_unifi_integration(
|
|
hass,
|
|
options={CONF_TRACK_CLIENTS: False, CONF_TRACK_DEVICES: False},
|
|
clients_response=[CONTROLLER_HOST],
|
|
devices_response=[DEVICE_1],
|
|
)
|
|
|
|
assert len(controller.mock_requests) == 4
|
|
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 0
|
|
cloudkey = hass.states.get("switch.cloud_key")
|
|
assert cloudkey is None
|
|
|
|
|
|
async def test_not_admin(hass):
|
|
"""Test that switch platform only work on an admin account."""
|
|
description = deepcopy(DESCRIPTION)
|
|
description[0]["site_role"] = "not admin"
|
|
controller = await setup_unifi_integration(
|
|
hass,
|
|
options={CONF_TRACK_CLIENTS: False, CONF_TRACK_DEVICES: False},
|
|
site_description=description,
|
|
clients_response=[CLIENT_1],
|
|
devices_response=[DEVICE_1],
|
|
)
|
|
|
|
assert len(controller.mock_requests) == 4
|
|
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 0
|
|
|
|
|
|
async def test_switches(hass):
|
|
"""Test the update_items function with some clients."""
|
|
controller = await setup_unifi_integration(
|
|
hass,
|
|
options={
|
|
CONF_BLOCK_CLIENT: [BLOCKED["mac"], UNBLOCKED["mac"]],
|
|
CONF_TRACK_CLIENTS: False,
|
|
CONF_TRACK_DEVICES: False,
|
|
},
|
|
clients_response=[CLIENT_1, CLIENT_4],
|
|
devices_response=[DEVICE_1],
|
|
clients_all_response=[BLOCKED, UNBLOCKED, CLIENT_1],
|
|
)
|
|
|
|
assert len(controller.mock_requests) == 4
|
|
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 3
|
|
|
|
switch_1 = hass.states.get("switch.poe_client_1")
|
|
assert switch_1 is not None
|
|
assert switch_1.state == "on"
|
|
assert switch_1.attributes["power"] == "2.56"
|
|
assert switch_1.attributes[SWITCH_DOMAIN] == "00:00:00:00:01:01"
|
|
assert switch_1.attributes["port"] == 1
|
|
assert switch_1.attributes["poe_mode"] == "auto"
|
|
|
|
switch_4 = hass.states.get("switch.poe_client_4")
|
|
assert switch_4 is None
|
|
|
|
blocked = hass.states.get("switch.block_client_1")
|
|
assert blocked is not None
|
|
assert blocked.state == "off"
|
|
|
|
unblocked = hass.states.get("switch.block_client_2")
|
|
assert unblocked is not None
|
|
assert unblocked.state == "on"
|
|
|
|
await hass.services.async_call(
|
|
SWITCH_DOMAIN, "turn_off", {"entity_id": "switch.block_client_1"}, blocking=True
|
|
)
|
|
assert len(controller.mock_requests) == 5
|
|
assert controller.mock_requests[4] == {
|
|
"json": {"mac": "00:00:00:00:01:01", "cmd": "block-sta"},
|
|
"method": "post",
|
|
"path": "/cmd/stamgr",
|
|
}
|
|
|
|
await hass.services.async_call(
|
|
SWITCH_DOMAIN, "turn_on", {"entity_id": "switch.block_client_1"}, blocking=True
|
|
)
|
|
assert len(controller.mock_requests) == 6
|
|
assert controller.mock_requests[5] == {
|
|
"json": {"mac": "00:00:00:00:01:01", "cmd": "unblock-sta"},
|
|
"method": "post",
|
|
"path": "/cmd/stamgr",
|
|
}
|
|
|
|
|
|
async def test_remove_switches(hass):
|
|
"""Test the update_items function with some clients."""
|
|
controller = await setup_unifi_integration(
|
|
hass,
|
|
options={CONF_BLOCK_CLIENT: [UNBLOCKED["mac"]]},
|
|
clients_response=[CLIENT_1, UNBLOCKED],
|
|
devices_response=[DEVICE_1],
|
|
)
|
|
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 2
|
|
|
|
poe_switch = hass.states.get("switch.poe_client_1")
|
|
assert poe_switch is not None
|
|
|
|
block_switch = hass.states.get("switch.block_client_2")
|
|
assert block_switch is not None
|
|
|
|
controller.api.websocket._data = {
|
|
"meta": {"message": MESSAGE_CLIENT_REMOVED},
|
|
"data": [CLIENT_1, UNBLOCKED],
|
|
}
|
|
controller.api.session_handler(SIGNAL_DATA)
|
|
await hass.async_block_till_done()
|
|
|
|
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 0
|
|
|
|
poe_switch = hass.states.get("switch.poe_client_1")
|
|
assert poe_switch is None
|
|
|
|
block_switch = hass.states.get("switch.block_client_2")
|
|
assert block_switch is None
|
|
|
|
|
|
async def test_block_switches(hass):
|
|
"""Test the update_items function with some clients."""
|
|
controller = await setup_unifi_integration(
|
|
hass,
|
|
options={
|
|
CONF_BLOCK_CLIENT: [BLOCKED["mac"], UNBLOCKED["mac"]],
|
|
CONF_TRACK_CLIENTS: False,
|
|
CONF_TRACK_DEVICES: False,
|
|
},
|
|
clients_response=[UNBLOCKED],
|
|
clients_all_response=[BLOCKED],
|
|
)
|
|
|
|
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 2
|
|
|
|
blocked = hass.states.get("switch.block_client_1")
|
|
assert blocked is not None
|
|
assert blocked.state == "off"
|
|
|
|
unblocked = hass.states.get("switch.block_client_2")
|
|
assert unblocked is not None
|
|
assert unblocked.state == "on"
|
|
|
|
controller.api.websocket._data = {
|
|
"meta": {"message": MESSAGE_EVENT},
|
|
"data": [EVENT_BLOCKED_CLIENT_UNBLOCKED],
|
|
}
|
|
controller.api.session_handler(SIGNAL_DATA)
|
|
await hass.async_block_till_done()
|
|
|
|
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 2
|
|
blocked = hass.states.get("switch.block_client_1")
|
|
assert blocked is not None
|
|
assert blocked.state == "on"
|
|
|
|
controller.api.websocket._data = {
|
|
"meta": {"message": MESSAGE_EVENT},
|
|
"data": [EVENT_BLOCKED_CLIENT_BLOCKED],
|
|
}
|
|
controller.api.session_handler(SIGNAL_DATA)
|
|
await hass.async_block_till_done()
|
|
|
|
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 2
|
|
blocked = hass.states.get("switch.block_client_1")
|
|
assert blocked is not None
|
|
assert blocked.state == "off"
|
|
|
|
await hass.services.async_call(
|
|
SWITCH_DOMAIN, "turn_off", {"entity_id": "switch.block_client_1"}, blocking=True
|
|
)
|
|
assert len(controller.mock_requests) == 5
|
|
assert controller.mock_requests[4] == {
|
|
"json": {"mac": "00:00:00:00:01:01", "cmd": "block-sta"},
|
|
"method": "post",
|
|
"path": "/cmd/stamgr",
|
|
}
|
|
|
|
await hass.services.async_call(
|
|
SWITCH_DOMAIN, "turn_on", {"entity_id": "switch.block_client_1"}, blocking=True
|
|
)
|
|
assert len(controller.mock_requests) == 6
|
|
assert controller.mock_requests[5] == {
|
|
"json": {"mac": "00:00:00:00:01:01", "cmd": "unblock-sta"},
|
|
"method": "post",
|
|
"path": "/cmd/stamgr",
|
|
}
|
|
|
|
|
|
async def test_new_client_discovered_on_block_control(hass):
|
|
"""Test if 2nd update has a new client."""
|
|
controller = await setup_unifi_integration(
|
|
hass,
|
|
options={
|
|
CONF_BLOCK_CLIENT: [BLOCKED["mac"]],
|
|
CONF_TRACK_CLIENTS: False,
|
|
CONF_TRACK_DEVICES: False,
|
|
},
|
|
)
|
|
|
|
assert len(controller.mock_requests) == 4
|
|
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 0
|
|
|
|
blocked = hass.states.get("switch.block_client_1")
|
|
assert blocked is None
|
|
|
|
controller.api.websocket._data = {
|
|
"meta": {"message": "sta:sync"},
|
|
"data": [BLOCKED],
|
|
}
|
|
controller.api.session_handler(SIGNAL_DATA)
|
|
await hass.async_block_till_done()
|
|
|
|
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 0
|
|
|
|
controller.api.websocket._data = {
|
|
"meta": {"message": MESSAGE_EVENT},
|
|
"data": [EVENT_BLOCKED_CLIENT_CONNECTED],
|
|
}
|
|
controller.api.session_handler(SIGNAL_DATA)
|
|
await hass.async_block_till_done()
|
|
|
|
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 1
|
|
blocked = hass.states.get("switch.block_client_1")
|
|
assert blocked is not None
|
|
|
|
|
|
async def test_option_block_clients(hass):
|
|
"""Test the changes to option reflects accordingly."""
|
|
controller = await setup_unifi_integration(
|
|
hass,
|
|
options={CONF_BLOCK_CLIENT: [BLOCKED["mac"]]},
|
|
clients_all_response=[BLOCKED, UNBLOCKED],
|
|
)
|
|
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 1
|
|
|
|
# Add a second switch
|
|
hass.config_entries.async_update_entry(
|
|
controller.config_entry,
|
|
options={CONF_BLOCK_CLIENT: [BLOCKED["mac"], UNBLOCKED["mac"]]},
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 1
|
|
|
|
# Remove the second switch again
|
|
hass.config_entries.async_update_entry(
|
|
controller.config_entry,
|
|
options={CONF_BLOCK_CLIENT: [BLOCKED["mac"]]},
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 1
|
|
|
|
# Enable one and remove another one
|
|
hass.config_entries.async_update_entry(
|
|
controller.config_entry,
|
|
options={CONF_BLOCK_CLIENT: [UNBLOCKED["mac"]]},
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 0
|
|
|
|
# Remove one
|
|
hass.config_entries.async_update_entry(
|
|
controller.config_entry,
|
|
options={CONF_BLOCK_CLIENT: []},
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 0
|
|
|
|
|
|
async def test_new_client_discovered_on_poe_control(hass):
|
|
"""Test if 2nd update has a new client."""
|
|
controller = await setup_unifi_integration(
|
|
hass,
|
|
options={CONF_TRACK_CLIENTS: False, CONF_TRACK_DEVICES: False},
|
|
clients_response=[CLIENT_1],
|
|
devices_response=[DEVICE_1],
|
|
)
|
|
|
|
assert len(controller.mock_requests) == 4
|
|
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 1
|
|
|
|
controller.api.websocket._data = {
|
|
"meta": {"message": "sta:sync"},
|
|
"data": [CLIENT_2],
|
|
}
|
|
controller.api.session_handler(SIGNAL_DATA)
|
|
await hass.async_block_till_done()
|
|
|
|
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 1
|
|
|
|
controller.api.websocket._data = {
|
|
"meta": {"message": MESSAGE_EVENT},
|
|
"data": [EVENT_CLIENT_2_CONNECTED],
|
|
}
|
|
controller.api.session_handler(SIGNAL_DATA)
|
|
await hass.async_block_till_done()
|
|
|
|
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 2
|
|
switch_2 = hass.states.get("switch.poe_client_2")
|
|
assert switch_2 is not None
|
|
|
|
await hass.services.async_call(
|
|
SWITCH_DOMAIN, "turn_off", {"entity_id": "switch.poe_client_1"}, blocking=True
|
|
)
|
|
assert len(controller.mock_requests) == 5
|
|
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 2
|
|
assert controller.mock_requests[4] == {
|
|
"json": {
|
|
"port_overrides": [{"port_idx": 1, "portconf_id": "1a1", "poe_mode": "off"}]
|
|
},
|
|
"method": "put",
|
|
"path": "/rest/device/mock-id",
|
|
}
|
|
|
|
await hass.services.async_call(
|
|
SWITCH_DOMAIN, "turn_on", {"entity_id": "switch.poe_client_1"}, blocking=True
|
|
)
|
|
assert len(controller.mock_requests) == 6
|
|
assert controller.mock_requests[4] == {
|
|
"json": {
|
|
"port_overrides": [
|
|
{"port_idx": 1, "portconf_id": "1a1", "poe_mode": "auto"}
|
|
]
|
|
},
|
|
"method": "put",
|
|
"path": "/rest/device/mock-id",
|
|
}
|
|
|
|
|
|
async def test_ignore_multiple_poe_clients_on_same_port(hass):
|
|
"""Ignore when there are multiple POE driven clients on same port.
|
|
|
|
If there is a non-UniFi switch powered by POE,
|
|
clients will be transparently marked as having POE as well.
|
|
"""
|
|
controller = await setup_unifi_integration(
|
|
hass,
|
|
clients_response=POE_SWITCH_CLIENTS,
|
|
devices_response=[DEVICE_1],
|
|
)
|
|
|
|
assert len(controller.mock_requests) == 4
|
|
assert len(hass.states.async_entity_ids(TRACKER_DOMAIN)) == 3
|
|
|
|
switch_1 = hass.states.get("switch.poe_client_1")
|
|
switch_2 = hass.states.get("switch.poe_client_2")
|
|
assert switch_1 is None
|
|
assert switch_2 is None
|
|
|
|
|
|
async def test_restoring_client(hass):
|
|
"""Test the update_items function with some clients."""
|
|
config_entry = config_entries.ConfigEntry(
|
|
version=1,
|
|
domain=UNIFI_DOMAIN,
|
|
title="Mock Title",
|
|
data=ENTRY_CONFIG,
|
|
source="test",
|
|
connection_class=config_entries.CONN_CLASS_LOCAL_POLL,
|
|
system_options={},
|
|
options={},
|
|
entry_id=1,
|
|
)
|
|
|
|
registry = await entity_registry.async_get_registry(hass)
|
|
registry.async_get_or_create(
|
|
SWITCH_DOMAIN,
|
|
UNIFI_DOMAIN,
|
|
f'{POE_SWITCH}-{CLIENT_1["mac"]}',
|
|
suggested_object_id=CLIENT_1["hostname"],
|
|
config_entry=config_entry,
|
|
)
|
|
registry.async_get_or_create(
|
|
SWITCH_DOMAIN,
|
|
UNIFI_DOMAIN,
|
|
f'{POE_SWITCH}-{CLIENT_2["mac"]}',
|
|
suggested_object_id=CLIENT_2["hostname"],
|
|
config_entry=config_entry,
|
|
)
|
|
|
|
controller = await setup_unifi_integration(
|
|
hass,
|
|
options={
|
|
CONF_BLOCK_CLIENT: ["random mac"],
|
|
CONF_TRACK_CLIENTS: False,
|
|
CONF_TRACK_DEVICES: False,
|
|
},
|
|
clients_response=[CLIENT_2],
|
|
devices_response=[DEVICE_1],
|
|
clients_all_response=[CLIENT_1],
|
|
)
|
|
|
|
assert len(controller.mock_requests) == 4
|
|
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 2
|
|
|
|
device_1 = hass.states.get("switch.client_1")
|
|
assert device_1 is not None
|