Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
core/tests/components/unifi/test_switch.py

1208 lines
36 KiB
Python
Raw Normal View History

Rename various usages of UniFi to better identify only UniFi Network (#59568) * Renames various usages of UniFi to better indentify only UniFi Network * Apply suggestions from code review Co-authored-by: Robert Svensson <Kane610@users.noreply.github.com> * Missed renames * Updates more locations * Removes instances of application/controller * Missed a spot * Updates all UniFi Controller instances * Fixes typo * Reverts changes to translations Co-authored-by: Robert Svensson <Kane610@users.noreply.github.com>
2021-11-26 21:44:49 +00:00
"""UniFi Network switch platform tests."""
Expose UniFi PoE ports as individual switches (#80566) * Add simple PoE control switches * Add basic tests * Complete testing * Dont use port.up as part of available * Bump aiounifi to v40
2022-10-19 17:54:40 +00:00
UniFi - Improve switch tests (#27200) * Continue rewriting tests for UniFi
2019-10-05 14:16:08 +00:00
from copy import deepcopy
Expose UniFi PoE ports as individual switches (#80566) * Add simple PoE control switches * Add basic tests * Complete testing * Dont use port.up as part of available * Bump aiounifi to v40
2022-10-19 17:54:40 +00:00
from datetime import timedelta
UniFi POE control (#17011) * First commit * Feature complete? * Add dependency * Move setting poe mode logic to library * Use guard clauses * Bump requirement to 2 * Simplify saving switches with poe off * Store and use poe mode * Fix indentation * Fix flake8 * Configuration future proofing * Bump dependency to v3 * Add first test * Proper use of defaults with config flow (thanks helto) * Appease hound * Make sure there can't be duplicate entries of combination host+site * More tests * More tests * 98% coverage of controller * Fix hound comments * Config flow step init not necessary * Use async_current_entries to check if host and site for controller is used * Remove storing/restoring poe off devices to slim PR * First batch of switch tests * More switch tests. * Small improvements and clean up * Make tests pass Don't name device in device registry * Dont process clients that belong to non-UniFi POE switches * Allow selection of site from a list in config flow * Fix double blank lines in method * Update codeowners
2018-10-16 08:35:35 +00:00
Expose UniFi PoE ports as individual switches (#80566) * Add simple PoE control switches * Add basic tests * Complete testing * Dont use port.up as part of available * Bump aiounifi to v40
2022-10-19 17:54:40 +00:00
from aiounifi.models.message import MessageKey
from aiounifi.websocket import WebsocketState
UniFi - Support automatic removal of clients (#34307)
2020-04-16 22:08:53 +00:00
Remove old UniFi POE client implementation (#81749) Remove all references to POE client implementation
2022-11-08 06:48:54 +00:00
from homeassistant import config_entries
Add outlets to UniFi switch platform (#64859) * Add outlets to UniFi switch platform * Outlet shouldn't have entity category set
2022-01-25 06:49:02 +00:00
from homeassistant.components.switch import (
DOMAIN as SWITCH_DOMAIN,
SERVICE_TURN_OFF,
SERVICE_TURN_ON,
Expose UniFi PoE ports as individual switches (#80566) * Add simple PoE control switches * Add basic tests * Complete testing * Dont use port.up as part of available * Bump aiounifi to v40
2022-10-19 17:54:40 +00:00
SwitchDeviceClass,
Add outlets to UniFi switch platform (#64859) * Add outlets to UniFi switch platform * Outlet shouldn't have entity category set
2022-01-25 06:49:02 +00:00
)
UniFi - Add block network access control to config option (#32004) * Add block network access control to config option * Clean up
2020-03-05 05:55:56 +00:00
from homeassistant.components.unifi.const import (
CONF_BLOCK_CLIENT,
Add DPI Restriction switch to UniFi integration (#42499) * initial implementation for controlling DPI restrictions * address PR review comments and add DataUpdateCoordinator * fix existing tests against new lib version * add tests for DPI switches * bump aiounifi * listen to events instead of polling * fix tests * remove useless test * bump aiounifi * rename device to UniFi Controller per PR feedback
2020-11-03 07:36:37 +00:00
CONF_DPI_RESTRICTIONS,
UniFi - Add block network access control to config option (#32004) * Add block network access control to config option * Clean up
2020-03-05 05:55:56 +00:00
CONF_TRACK_CLIENTS,
CONF_TRACK_DEVICES,
UniFi - Store controller in config_entry.entry_id (#34553) * Store controller in config_entry.entry_id * Clean up imports
2020-04-23 14:48:24 +00:00
DOMAIN as UNIFI_DOMAIN,
UniFi - Add block network access control to config option (#32004) * Add block network access control to config option * Clean up
2020-03-05 05:55:56 +00:00
)
Expose UniFi PoE ports as individual switches (#80566) * Add simple PoE control switches * Add basic tests * Complete testing * Dont use port.up as part of available * Bump aiounifi to v40
2022-10-19 17:54:40 +00:00
from homeassistant.config_entries import RELOAD_AFTER_UPDATE_DELAY
from homeassistant.const import (
ATTR_DEVICE_CLASS,
ATTR_ENTITY_ID,
STATE_OFF,
STATE_ON,
STATE_UNAVAILABLE,
)
Update tests t-z to use async_get() instead of async_get_registry() (#47655)
2021-03-09 13:25:03 +00:00
from homeassistant.helpers import entity_registry as er
Increase test coverage of UniFi integration (#46347) * Increase coverage of init * Increase coverage of config_flow * Improve coverage of controller * Minor improvement to switch test * Fix review comment * Mock websocket class * Replace the rest of the old websocket event tests * Improve websocket fixture for cleaner tests * Fix typing * Improve connection state signalling based on Martins feedback * Improve tests of reconnection_mechanisms based on Martins review comments * Fix unload entry * Fix isort issue after rebase * Fix martins comment on not using caplog * Fix wireless clients test * Fix martins comments on wireless clients test
2021-03-05 20:28:41 +00:00
from homeassistant.helpers.dispatcher import async_dispatcher_send
Use enums in unifi tests (#62151)
2021-12-17 10:12:43 +00:00
from homeassistant.helpers.entity import EntityCategory
Expose UniFi PoE ports as individual switches (#80566) * Add simple PoE control switches * Add basic tests * Complete testing * Dont use port.up as part of available * Bump aiounifi to v40
2022-10-19 17:54:40 +00:00
from homeassistant.helpers.entity_registry import RegistryEntryDisabler
from homeassistant.util import dt
UniFi POE control (#17011) * First commit * Feature complete? * Add dependency * Move setting poe mode logic to library * Use guard clauses * Bump requirement to 2 * Simplify saving switches with poe off * Store and use poe mode * Fix indentation * Fix flake8 * Configuration future proofing * Bump dependency to v3 * Add first test * Proper use of defaults with config flow (thanks helto) * Appease hound * Make sure there can't be duplicate entries of combination host+site * More tests * More tests * 98% coverage of controller * Fix hound comments * Config flow step init not necessary * Use async_current_entries to check if host and site for controller is used * Remove storing/restoring poe off devices to slim PR * First batch of switch tests * More switch tests. * Small improvements and clean up * Make tests pass Don't name device in device registry * Dont process clients that belong to non-UniFi POE switches * Allow selection of site from a list in config flow * Fix double blank lines in method * Update codeowners
2018-10-16 08:35:35 +00:00
UniFi - Improve controller tests (#27261) * Improve controller tests and harmonize setup_unifi_integration to one * Store listeners to dispatchers to be used during reset
2019-10-07 19:55:35 +00:00
from .test_controller import (
CONTROLLER_HOST,
UniFi - Sites don't declare role on UniFiOS 1.7.0 beta (#35555) * New way to identify role, compatible with standalone controller or as part of unifios * Remove error * Bump dependency to v22 * Remove unused import
2020-05-12 22:37:16 +00:00
DESCRIPTION,
UniFi - Improve controller tests (#27261) * Improve controller tests and harmonize setup_unifi_integration to one * Store listeners to dispatchers to be used during reset
2019-10-07 19:55:35 +00:00
ENTRY_CONFIG,
setup_unifi_integration,
)
Remove old UniFi POE client implementation (#81749) Remove all references to POE client implementation
2022-11-08 06:48:54 +00:00
from tests.common import async_fire_time_changed
Use mock_restore_cache in tests (#77298)
2022-08-25 07:28:53 +00:00
UniFi POE control (#17011) * First commit * Feature complete? * Add dependency * Move setting poe mode logic to library * Use guard clauses * Bump requirement to 2 * Simplify saving switches with poe off * Store and use poe mode * Fix indentation * Fix flake8 * Configuration future proofing * Bump dependency to v3 * Add first test * Proper use of defaults with config flow (thanks helto) * Appease hound * Make sure there can't be duplicate entries of combination host+site * More tests * More tests * 98% coverage of controller * Fix hound comments * Config flow step init not necessary * Use async_current_entries to check if host and site for controller is used * Remove storing/restoring poe off devices to slim PR * First batch of switch tests * More switch tests. * Small improvements and clean up * Make tests pass Don't name device in device registry * Dont process clients that belong to non-UniFi POE switches * Allow selection of site from a list in config flow * Fix double blank lines in method * Update codeowners
2018-10-16 08:35:35 +00:00
CLIENT_1 = {
Black
2019-07-31 19:25:30 +00:00
"hostname": "client_1",
"ip": "10.0.0.1",
"is_wired": True,
UniFi - Improve switch tests (#27200) * Continue rewriting tests for UniFi
2019-10-05 14:16:08 +00:00
"last_seen": 1562600145,
Black
2019-07-31 19:25:30 +00:00
"mac": "00:00:00:00:00:01",
"name": "POE Client 1",
"oui": "Producer",
Rewrite UniFi block client switch (#80969) * Refactor UniFi block client switch entities * Use new switch loader * Rename lambdas * Use is_on rather than _attr_is_on when applicable
2022-10-25 20:36:51 +00:00
"sw_mac": "10:00:00:00:01:01",
Black
2019-07-31 19:25:30 +00:00
"sw_port": 1,
"wired-rx_bytes": 1234000000,
"wired-tx_bytes": 5678000000,
UniFi POE control (#17011) * First commit * Feature complete? * Add dependency * Move setting poe mode logic to library * Use guard clauses * Bump requirement to 2 * Simplify saving switches with poe off * Store and use poe mode * Fix indentation * Fix flake8 * Configuration future proofing * Bump dependency to v3 * Add first test * Proper use of defaults with config flow (thanks helto) * Appease hound * Make sure there can't be duplicate entries of combination host+site * More tests * More tests * 98% coverage of controller * Fix hound comments * Config flow step init not necessary * Use async_current_entries to check if host and site for controller is used * Remove storing/restoring poe off devices to slim PR * First batch of switch tests * More switch tests. * Small improvements and clean up * Make tests pass Don't name device in device registry * Dont process clients that belong to non-UniFi POE switches * Allow selection of site from a list in config flow * Fix double blank lines in method * Update codeowners
2018-10-16 08:35:35 +00:00
}
CLIENT_2 = {
Black
2019-07-31 19:25:30 +00:00
"hostname": "client_2",
"ip": "10.0.0.2",
"is_wired": True,
UniFi - Improve switch tests (#27200) * Continue rewriting tests for UniFi
2019-10-05 14:16:08 +00:00
"last_seen": 1562600145,
Black
2019-07-31 19:25:30 +00:00
"mac": "00:00:00:00:00:02",
"name": "POE Client 2",
"oui": "Producer",
Rewrite UniFi block client switch (#80969) * Refactor UniFi block client switch entities * Use new switch loader * Rename lambdas * Use is_on rather than _attr_is_on when applicable
2022-10-25 20:36:51 +00:00
"sw_mac": "10:00:00:00:01:01",
Black
2019-07-31 19:25:30 +00:00
"sw_port": 2,
"wired-rx_bytes": 1234000000,
"wired-tx_bytes": 5678000000,
UniFi POE control (#17011) * First commit * Feature complete? * Add dependency * Move setting poe mode logic to library * Use guard clauses * Bump requirement to 2 * Simplify saving switches with poe off * Store and use poe mode * Fix indentation * Fix flake8 * Configuration future proofing * Bump dependency to v3 * Add first test * Proper use of defaults with config flow (thanks helto) * Appease hound * Make sure there can't be duplicate entries of combination host+site * More tests * More tests * 98% coverage of controller * Fix hound comments * Config flow step init not necessary * Use async_current_entries to check if host and site for controller is used * Remove storing/restoring poe off devices to slim PR * First batch of switch tests * More switch tests. * Small improvements and clean up * Make tests pass Don't name device in device registry * Dont process clients that belong to non-UniFi POE switches * Allow selection of site from a list in config flow * Fix double blank lines in method * Update codeowners
2018-10-16 08:35:35 +00:00
}
CLIENT_3 = {
Black
2019-07-31 19:25:30 +00:00
"hostname": "client_3",
"ip": "10.0.0.3",
"is_wired": True,
UniFi - Improve switch tests (#27200) * Continue rewriting tests for UniFi
2019-10-05 14:16:08 +00:00
"last_seen": 1562600145,
Black
2019-07-31 19:25:30 +00:00
"mac": "00:00:00:00:00:03",
"name": "Non-POE Client 3",
"oui": "Producer",
Rewrite UniFi block client switch (#80969) * Refactor UniFi block client switch entities * Use new switch loader * Rename lambdas * Use is_on rather than _attr_is_on when applicable
2022-10-25 20:36:51 +00:00
"sw_mac": "10:00:00:00:01:01",
Black
2019-07-31 19:25:30 +00:00
"sw_port": 3,
"wired-rx_bytes": 1234000000,
"wired-tx_bytes": 5678000000,
UniFi POE control (#17011) * First commit * Feature complete? * Add dependency * Move setting poe mode logic to library * Use guard clauses * Bump requirement to 2 * Simplify saving switches with poe off * Store and use poe mode * Fix indentation * Fix flake8 * Configuration future proofing * Bump dependency to v3 * Add first test * Proper use of defaults with config flow (thanks helto) * Appease hound * Make sure there can't be duplicate entries of combination host+site * More tests * More tests * 98% coverage of controller * Fix hound comments * Config flow step init not necessary * Use async_current_entries to check if host and site for controller is used * Remove storing/restoring poe off devices to slim PR * First batch of switch tests * More switch tests. * Small improvements and clean up * Make tests pass Don't name device in device registry * Dont process clients that belong to non-UniFi POE switches * Allow selection of site from a list in config flow * Fix double blank lines in method * Update codeowners
2018-10-16 08:35:35 +00:00
}
CLIENT_4 = {
Black
2019-07-31 19:25:30 +00:00
"hostname": "client_4",
"ip": "10.0.0.4",
"is_wired": True,
UniFi - Improve switch tests (#27200) * Continue rewriting tests for UniFi
2019-10-05 14:16:08 +00:00
"last_seen": 1562600145,
Black
2019-07-31 19:25:30 +00:00
"mac": "00:00:00:00:00:04",
"name": "Non-POE Client 4",
"oui": "Producer",
Rewrite UniFi block client switch (#80969) * Refactor UniFi block client switch entities * Use new switch loader * Rename lambdas * Use is_on rather than _attr_is_on when applicable
2022-10-25 20:36:51 +00:00
"sw_mac": "10:00:00:00:01:01",
Black
2019-07-31 19:25:30 +00:00
"sw_port": 4,
"wired-rx_bytes": 1234000000,
"wired-tx_bytes": 5678000000,
UniFi POE control (#17011) * First commit * Feature complete? * Add dependency * Move setting poe mode logic to library * Use guard clauses * Bump requirement to 2 * Simplify saving switches with poe off * Store and use poe mode * Fix indentation * Fix flake8 * Configuration future proofing * Bump dependency to v3 * Add first test * Proper use of defaults with config flow (thanks helto) * Appease hound * Make sure there can't be duplicate entries of combination host+site * More tests * More tests * 98% coverage of controller * Fix hound comments * Config flow step init not necessary * Use async_current_entries to check if host and site for controller is used * Remove storing/restoring poe off devices to slim PR * First batch of switch tests * More switch tests. * Small improvements and clean up * Make tests pass Don't name device in device registry * Dont process clients that belong to non-UniFi POE switches * Allow selection of site from a list in config flow * Fix double blank lines in method * Update codeowners
2018-10-16 08:35:35 +00:00
}
POE_SWITCH_CLIENTS = [
{
Black
2019-07-31 19:25:30 +00:00
"hostname": "client_1",
"ip": "10.0.0.1",
"is_wired": True,
UniFi - Improve switch tests (#27200) * Continue rewriting tests for UniFi
2019-10-05 14:16:08 +00:00
"last_seen": 1562600145,
Black
2019-07-31 19:25:30 +00:00
"mac": "00:00:00:00:00:01",
"name": "POE Client 1",
"oui": "Producer",
Rewrite UniFi block client switch (#80969) * Refactor UniFi block client switch entities * Use new switch loader * Rename lambdas * Use is_on rather than _attr_is_on when applicable
2022-10-25 20:36:51 +00:00
"sw_mac": "10:00:00:00:01:01",
Black
2019-07-31 19:25:30 +00:00
"sw_port": 1,
"wired-rx_bytes": 1234000000,
"wired-tx_bytes": 5678000000,
UniFi POE control (#17011) * First commit * Feature complete? * Add dependency * Move setting poe mode logic to library * Use guard clauses * Bump requirement to 2 * Simplify saving switches with poe off * Store and use poe mode * Fix indentation * Fix flake8 * Configuration future proofing * Bump dependency to v3 * Add first test * Proper use of defaults with config flow (thanks helto) * Appease hound * Make sure there can't be duplicate entries of combination host+site * More tests * More tests * 98% coverage of controller * Fix hound comments * Config flow step init not necessary * Use async_current_entries to check if host and site for controller is used * Remove storing/restoring poe off devices to slim PR * First batch of switch tests * More switch tests. * Small improvements and clean up * Make tests pass Don't name device in device registry * Dont process clients that belong to non-UniFi POE switches * Allow selection of site from a list in config flow * Fix double blank lines in method * Update codeowners
2018-10-16 08:35:35 +00:00
},
{
Black
2019-07-31 19:25:30 +00:00
"hostname": "client_2",
"ip": "10.0.0.2",
"is_wired": True,
UniFi - Improve switch tests (#27200) * Continue rewriting tests for UniFi
2019-10-05 14:16:08 +00:00
"last_seen": 1562600145,
Black
2019-07-31 19:25:30 +00:00
"mac": "00:00:00:00:00:02",
"name": "POE Client 2",
"oui": "Producer",
Rewrite UniFi block client switch (#80969) * Refactor UniFi block client switch entities * Use new switch loader * Rename lambdas * Use is_on rather than _attr_is_on when applicable
2022-10-25 20:36:51 +00:00
"sw_mac": "10:00:00:00:01:01",
Black
2019-07-31 19:25:30 +00:00
"sw_port": 1,
"wired-rx_bytes": 1234000000,
"wired-tx_bytes": 5678000000,
},
UniFi POE control (#17011) * First commit * Feature complete? * Add dependency * Move setting poe mode logic to library * Use guard clauses * Bump requirement to 2 * Simplify saving switches with poe off * Store and use poe mode * Fix indentation * Fix flake8 * Configuration future proofing * Bump dependency to v3 * Add first test * Proper use of defaults with config flow (thanks helto) * Appease hound * Make sure there can't be duplicate entries of combination host+site * More tests * More tests * 98% coverage of controller * Fix hound comments * Config flow step init not necessary * Use async_current_entries to check if host and site for controller is used * Remove storing/restoring poe off devices to slim PR * First batch of switch tests * More switch tests. * Small improvements and clean up * Make tests pass Don't name device in device registry * Dont process clients that belong to non-UniFi POE switches * Allow selection of site from a list in config flow * Fix double blank lines in method * Update codeowners
2018-10-16 08:35:35 +00:00
]
DEVICE_1 = {
Refactor UniFi outlet switches (#80738) * Rewrite UniFi outlet switches * Bump aiounifi to v41 * Remove devices from items_added input
2022-10-23 18:28:45 +00:00
"board_rev": 2,
Black
2019-07-31 19:25:30 +00:00
"device_id": "mock-id",
"ip": "10.0.1.1",
Rewrite UniFi block client switch (#80969) * Refactor UniFi block client switch entities * Use new switch loader * Rename lambdas * Use is_on rather than _attr_is_on when applicable
2022-10-25 20:36:51 +00:00
"mac": "10:00:00:00:01:01",
UniFi - Improve switch tests (#27200) * Continue rewriting tests for UniFi
2019-10-05 14:16:08 +00:00
"last_seen": 1562600145,
"model": "US16P150",
Black
2019-07-31 19:25:30 +00:00
"name": "mock-name",
"port_overrides": [],
"port_table": [
UniFi POE control (#17011) * First commit * Feature complete? * Add dependency * Move setting poe mode logic to library * Use guard clauses * Bump requirement to 2 * Simplify saving switches with poe off * Store and use poe mode * Fix indentation * Fix flake8 * Configuration future proofing * Bump dependency to v3 * Add first test * Proper use of defaults with config flow (thanks helto) * Appease hound * Make sure there can't be duplicate entries of combination host+site * More tests * More tests * 98% coverage of controller * Fix hound comments * Config flow step init not necessary * Use async_current_entries to check if host and site for controller is used * Remove storing/restoring poe off devices to slim PR * First batch of switch tests * More switch tests. * Small improvements and clean up * Make tests pass Don't name device in device registry * Dont process clients that belong to non-UniFi POE switches * Allow selection of site from a list in config flow * Fix double blank lines in method * Update codeowners
2018-10-16 08:35:35 +00:00
{
Black
2019-07-31 19:25:30 +00:00
"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,
UniFi POE control (#17011) * First commit * Feature complete? * Add dependency * Move setting poe mode logic to library * Use guard clauses * Bump requirement to 2 * Simplify saving switches with poe off * Store and use poe mode * Fix indentation * Fix flake8 * Configuration future proofing * Bump dependency to v3 * Add first test * Proper use of defaults with config flow (thanks helto) * Appease hound * Make sure there can't be duplicate entries of combination host+site * More tests * More tests * 98% coverage of controller * Fix hound comments * Config flow step init not necessary * Use async_current_entries to check if host and site for controller is used * Remove storing/restoring poe off devices to slim PR * First batch of switch tests * More switch tests. * Small improvements and clean up * Make tests pass Don't name device in device registry * Dont process clients that belong to non-UniFi POE switches * Allow selection of site from a list in config flow * Fix double blank lines in method * Update codeowners
2018-10-16 08:35:35 +00:00
},
{
Black
2019-07-31 19:25:30 +00:00
"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,
UniFi POE control (#17011) * First commit * Feature complete? * Add dependency * Move setting poe mode logic to library * Use guard clauses * Bump requirement to 2 * Simplify saving switches with poe off * Store and use poe mode * Fix indentation * Fix flake8 * Configuration future proofing * Bump dependency to v3 * Add first test * Proper use of defaults with config flow (thanks helto) * Appease hound * Make sure there can't be duplicate entries of combination host+site * More tests * More tests * 98% coverage of controller * Fix hound comments * Config flow step init not necessary * Use async_current_entries to check if host and site for controller is used * Remove storing/restoring poe off devices to slim PR * First batch of switch tests * More switch tests. * Small improvements and clean up * Make tests pass Don't name device in device registry * Dont process clients that belong to non-UniFi POE switches * Allow selection of site from a list in config flow * Fix double blank lines in method * Update codeowners
2018-10-16 08:35:35 +00:00
},
{
Black
2019-07-31 19:25:30 +00:00
"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,
UniFi POE control (#17011) * First commit * Feature complete? * Add dependency * Move setting poe mode logic to library * Use guard clauses * Bump requirement to 2 * Simplify saving switches with poe off * Store and use poe mode * Fix indentation * Fix flake8 * Configuration future proofing * Bump dependency to v3 * Add first test * Proper use of defaults with config flow (thanks helto) * Appease hound * Make sure there can't be duplicate entries of combination host+site * More tests * More tests * 98% coverage of controller * Fix hound comments * Config flow step init not necessary * Use async_current_entries to check if host and site for controller is used * Remove storing/restoring poe off devices to slim PR * First batch of switch tests * More switch tests. * Small improvements and clean up * Make tests pass Don't name device in device registry * Dont process clients that belong to non-UniFi POE switches * Allow selection of site from a list in config flow * Fix double blank lines in method * Update codeowners
2018-10-16 08:35:35 +00:00
},
{
Black
2019-07-31 19:25:30 +00:00
"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,
},
],
UniFi - Improve switch tests (#27200) * Continue rewriting tests for UniFi
2019-10-05 14:16:08 +00:00
"state": 1,
"type": "usw",
"version": "4.0.42.10433",
UniFi POE control (#17011) * First commit * Feature complete? * Add dependency * Move setting poe mode logic to library * Use guard clauses * Bump requirement to 2 * Simplify saving switches with poe off * Store and use poe mode * Fix indentation * Fix flake8 * Configuration future proofing * Bump dependency to v3 * Add first test * Proper use of defaults with config flow (thanks helto) * Appease hound * Make sure there can't be duplicate entries of combination host+site * More tests * More tests * 98% coverage of controller * Fix hound comments * Config flow step init not necessary * Use async_current_entries to check if host and site for controller is used * Remove storing/restoring poe off devices to slim PR * First batch of switch tests * More switch tests. * Small improvements and clean up * Make tests pass Don't name device in device registry * Dont process clients that belong to non-UniFi POE switches * Allow selection of site from a list in config flow * Fix double blank lines in method * Update codeowners
2018-10-16 08:35:35 +00:00
}
UniFi block clients (#25478) * Allow blocking clients on UniFi networks
2019-07-25 14:56:56 +00:00
BLOCKED = {
Black
2019-07-31 19:25:30 +00:00
"blocked": True,
"hostname": "block_client_1",
"ip": "10.0.0.1",
"is_guest": False,
"is_wired": False,
UniFi - Support automatic removal of clients (#34307)
2020-04-16 22:08:53 +00:00
"last_seen": 1562600145,
Black
2019-07-31 19:25:30 +00:00
"mac": "00:00:00:00:01:01",
"name": "Block Client 1",
"noted": True,
"oui": "Producer",
UniFi block clients (#25478) * Allow blocking clients on UniFi networks
2019-07-25 14:56:56 +00:00
}
UNBLOCKED = {
Black
2019-07-31 19:25:30 +00:00
"blocked": False,
"hostname": "block_client_2",
"ip": "10.0.0.2",
"is_guest": False,
"is_wired": True,
UniFi - Support automatic removal of clients (#34307)
2020-04-16 22:08:53 +00:00
"last_seen": 1562600145,
Black
2019-07-31 19:25:30 +00:00
"mac": "00:00:00:00:01:02",
"name": "Block Client 2",
"noted": True,
"oui": "Producer",
UniFi block clients (#25478) * Allow blocking clients on UniFi networks
2019-07-25 14:56:56 +00:00
}
UniFi - Improve signalling and handling of adding new entities (#34925) * Change signalling and handling of adding devices * Bump aiounifi to v19 Use the proper variable for roamradio
2020-05-04 17:29:49 +00:00
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",
}
UniFi - Make clients proper push based (#35273) * Improve client tracker to be more comprehensible and streamlined Improve block switches Improve tests * Small clean up * Add descriptions on ssid test * Improve test * Make polling default off, only POE clients left to verify * Minor improvements * On removal cancel scheduled updates * POE works without polling now * Combine else and if to an elif
2020-05-08 20:19:27 +00:00
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",
}
UniFi - Improve signalling and handling of adding new entities (#34925) * Change signalling and handling of adding devices * Bump aiounifi to v19 Use the proper variable for roamradio
2020-05-04 17:29:49 +00:00
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",
}
UniFi POE control (#17011) * First commit * Feature complete? * Add dependency * Move setting poe mode logic to library * Use guard clauses * Bump requirement to 2 * Simplify saving switches with poe off * Store and use poe mode * Fix indentation * Fix flake8 * Configuration future proofing * Bump dependency to v3 * Add first test * Proper use of defaults with config flow (thanks helto) * Appease hound * Make sure there can't be duplicate entries of combination host+site * More tests * More tests * 98% coverage of controller * Fix hound comments * Config flow step init not necessary * Use async_current_entries to check if host and site for controller is used * Remove storing/restoring poe off devices to slim PR * First batch of switch tests * More switch tests. * Small improvements and clean up * Make tests pass Don't name device in device registry * Dont process clients that belong to non-UniFi POE switches * Allow selection of site from a list in config flow * Fix double blank lines in method * Update codeowners
2018-10-16 08:35:35 +00:00
Add DPI Restriction switch to UniFi integration (#42499) * initial implementation for controlling DPI restrictions * address PR review comments and add DataUpdateCoordinator * fix existing tests against new lib version * add tests for DPI switches * bump aiounifi * listen to events instead of polling * fix tests * remove useless test * bump aiounifi * rename device to UniFi Controller per PR feedback
2020-11-03 07:36:37 +00:00
DPI_GROUPS = [
{
"_id": "5ba29dd8e3c58f026e9d7c4a",
"attr_no_delete": True,
"attr_hidden_id": "Default",
"name": "Default",
"site_id": "name",
},
{
"_id": "5f976f4ae3c58f018ec7dff6",
"name": "Block Media Streaming",
"site_id": "name",
"dpiapp_ids": ["5f976f62e3c58f018ec7e17d"],
},
]
DPI_APPS = [
{
"_id": "5f976f62e3c58f018ec7e17d",
"apps": [],
"blocked": True,
"cats": ["4"],
"enabled": True,
"log": True,
"site_id": "name",
}
]
Fix issue where UniFi DPI groups state wasn't being updated (#58502)
2021-10-27 08:35:44 +00:00
DPI_GROUP_REMOVED_EVENT = {
"meta": {"rc": "ok", "message": "dpigroup:delete"},
"data": [
{
"_id": "5f976f4ae3c58f018ec7dff6",
Move DPI Group logic to UniFi switch platform (#58761) * Library has normalized management of DPI apps and groups, move logic to UniFi integration * Bump dependency to v29 * Use a generator instead of a list - Pylint * Minor improvements * Improve doc strings
2022-01-12 16:11:05 +00:00
"name": "Block Media Streaming",
"site_id": "name",
"dpiapp_ids": [],
}
],
}
DPI_GROUP_CREATED_EVENT = {
"meta": {"rc": "ok", "message": "dpigroup:add"},
"data": [
{
"name": "Block Media Streaming",
"site_id": "name",
"_id": "5f976f4ae3c58f018ec7dff6",
}
],
}
DPI_GROUP_ADDED_APP = {
"meta": {"rc": "ok", "message": "dpigroup:sync"},
"data": [
{
"_id": "5f976f4ae3c58f018ec7dff6",
"name": "Block Media Streaming",
"site_id": "name",
"dpiapp_ids": ["5f976f62e3c58f018ec7e17d"],
}
],
}
DPI_GROUP_REMOVE_APP = {
"meta": {"rc": "ok", "message": "dpigroup:sync"},
"data": [
{
"_id": "5f976f4ae3c58f018ec7dff6",
"name": "Block Media Streaming",
Fix issue where UniFi DPI groups state wasn't being updated (#58502)
2021-10-27 08:35:44 +00:00
"site_id": "name",
"dpiapp_ids": [],
}
],
}
DPI_APP_DISABLED_EVENT = {
"meta": {"rc": "ok", "message": "dpiapp:sync"},
"data": [
{
"_id": "5f976f62e3c58f018ec7e17d",
"apps": [],
"blocked": False,
"cats": [],
"enabled": False,
"log": False,
"site_id": "name",
}
],
}
Add outlets to UniFi switch platform (#64859) * Add outlets to UniFi switch platform * Outlet shouldn't have entity category set
2022-01-25 06:49:02 +00:00
OUTLET_UP1 = {
"_id": "600c8356942a6ade50707b56",
"ip": "192.168.0.189",
"mac": "fc:ec:da:76:4f:5f",
"model": "UP1",
"model_in_lts": False,
"model_in_eol": False,
"type": "uap",
"version": "2.2.1.511",
"adopted": True,
"site_id": "545eb1f0e4b0205d14c4e548",
"x_authkey": "345678976545678",
"cfgversion": "4c62f1e663783447",
"syslog_key": "41c4bcefcbc842d6eefb05b8fd9b78faa1841d10a09cebb170ce3e2f474b43b3",
"config_network": {"type": "dhcp"},
"setup_id": "a8730d36-8fdd-44f9-8678-1e89676f36c1",
"x_vwirekey": "2dabb7e23b048c88b60123456789",
"vwire_table": [],
"dot1x_portctrl_enabled": False,
"outlet_overrides": [],
"outlet_enabled": True,
"license_state": "registered",
"x_aes_gcm": True,
"inform_url": "http://192.168.0.5:8080/inform",
"inform_ip": "192.168.0.5",
"required_version": "2.1.3",
"anon_id": "d2744a31-1c26-92fe-423d-6b9ba204abc7",
"board_rev": 2,
"manufacturer_id": 72,
"model_incompatible": False,
"antenna_table": [],
"radio_table": [],
"scan_radio_table": [],
"ethernet_table": [],
"port_table": [],
"switch_caps": {},
"has_speaker": False,
"has_eth1": False,
"fw_caps": 0,
"hw_caps": 128,
"wifi_caps": 0,
"sys_error_caps": 0,
"has_fan": False,
"has_temperature": False,
"country_code": 10752,
"outlet_table": [
{
"index": 1,
"has_relay": True,
"has_metering": False,
Refactor UniFi outlet switches (#80738) * Rewrite UniFi outlet switches * Bump aiounifi to v41 * Remove devices from items_added input
2022-10-23 18:28:45 +00:00
"relay_state": True,
"name": "Outlet 1",
},
{
"index": 2,
"has_relay": False,
"has_metering": False,
Add outlets to UniFi switch platform (#64859) * Add outlets to UniFi switch platform * Outlet shouldn't have entity category set
2022-01-25 06:49:02 +00:00
"relay_state": False,
"name": "Outlet 1",
Refactor UniFi outlet switches (#80738) * Rewrite UniFi outlet switches * Bump aiounifi to v41 * Remove devices from items_added input
2022-10-23 18:28:45 +00:00
},
Add outlets to UniFi switch platform (#64859) * Add outlets to UniFi switch platform * Outlet shouldn't have entity category set
2022-01-25 06:49:02 +00:00
],
"element_ap_serial": "44:d9:e7:90:f4:24",
"connected_at": 1641678609,
"provisioned_at": 1642054077,
"led_override": "default",
"led_override_color": "#0000ff",
"led_override_color_brightness": 100,
"outdoor_mode_override": "default",
"lcm_brightness_override": False,
"lcm_idle_timeout_override": False,
"name": "Plug",
"unsupported": False,
"unsupported_reason": 0,
"two_phase_adopt": False,
"serial": "FCECDA764F5F",
"lcm_tracker_enabled": False,
"wlangroup_id_ng": "545eb1f0e4b0205d14c4e555",
"supports_fingerprint_ml": False,
"last_uplink": {
"uplink_mac": "78:45:58:87:93:16",
"uplink_device_name": "U6-Pro",
"type": "wireless",
},
"device_id": "600c8356942a6ade50707b56",
"uplink": {
"uplink_mac": "78:45:58:87:93:16",
"uplink_device_name": "U6-Pro",
"type": "wireless",
"up": True,
"ap_mac": "78:45:58:87:93:16",
"tx_rate": 54000,
"rx_rate": 72200,
"rssi": 60,
"is_11ax": False,
"is_11ac": False,
"is_11n": True,
"is_11b": False,
"radio": "ng",
"essid": "Network Name",
"channel": 11,
"tx_packets": 1586746,
"rx_packets": 362176,
"tx_bytes": 397773,
"rx_bytes": 24423980,
"tx_bytes-r": 0,
"rx_bytes-r": 45,
"uplink_source": "legacy",
},
"state": 1,
"start_disconnected_millis": 1641679166349,
"last_seen": 1642055273,
"next_interval": 40,
"known_cfgversion": "4c62f1e663783447",
"start_connected_millis": 1641679166355,
"upgradable": False,
"adoptable_when_upgraded": False,
"rollupgrade": False,
"uptime": 376083,
"_uptime": 376083,
"locating": False,
"connect_request_ip": "192.168.0.189",
"connect_request_port": "49155",
"sys_stats": {"mem_total": 98304, "mem_used": 87736},
"system-stats": {},
"lldp_table": [],
"displayable_version": "2.2.1",
"connection_network_name": "LAN",
"startup_timestamp": 1641679190,
"scanning": False,
"spectrum_scanning": False,
"meshv3_peer_mac": "",
"element_peer_mac": "",
"satisfaction": 100,
"uplink_bssid": "78:45:58:87:93:17",
"hide_ch_width": "none",
"isolated": False,
"radio_table_stats": [],
"port_stats": [],
"vap_table": [],
"downlink_table": [],
"vwire_vap_table": [],
"bytes-d": 0,
"tx_bytes-d": 0,
"rx_bytes-d": 0,
"bytes-r": 0,
"element_uplink_ap_mac": "78:45:58:87:93:16",
"prev_non_busy_state": 1,
"stat": {
"ap": {
"site_id": "5a32aa4ee4b0412345678910",
"o": "ap",
"oid": "fc:ec:da:76:4f:5f",
"ap": "fc:ec:da:76:4f:5f",
"time": 1641678600000,
"datetime": "2022-01-08T21:50:00Z",
"user-rx_packets": 0.0,
"guest-rx_packets": 0.0,
"rx_packets": 0.0,
"user-rx_bytes": 0.0,
"guest-rx_bytes": 0.0,
"rx_bytes": 0.0,
"user-rx_errors": 0.0,
"guest-rx_errors": 0.0,
"rx_errors": 0.0,
"user-rx_dropped": 0.0,
"guest-rx_dropped": 0.0,
"rx_dropped": 0.0,
"user-rx_crypts": 0.0,
"guest-rx_crypts": 0.0,
"rx_crypts": 0.0,
"user-rx_frags": 0.0,
"guest-rx_frags": 0.0,
"rx_frags": 0.0,
"user-tx_packets": 0.0,
"guest-tx_packets": 0.0,
"tx_packets": 0.0,
"user-tx_bytes": 0.0,
"guest-tx_bytes": 0.0,
"tx_bytes": 0.0,
"user-tx_errors": 0.0,
"guest-tx_errors": 0.0,
"tx_errors": 0.0,
"user-tx_dropped": 0.0,
"guest-tx_dropped": 0.0,
"tx_dropped": 0.0,
"user-tx_retries": 0.0,
"guest-tx_retries": 0.0,
"tx_retries": 0.0,
"user-mac_filter_rejections": 0.0,
"guest-mac_filter_rejections": 0.0,
"mac_filter_rejections": 0.0,
"user-wifi_tx_attempts": 0.0,
"guest-wifi_tx_attempts": 0.0,
"wifi_tx_attempts": 0.0,
"user-wifi_tx_dropped": 0.0,
"guest-wifi_tx_dropped": 0.0,
"wifi_tx_dropped": 0.0,
"bytes": 0.0,
"duration": 376663000.0,
}
},
"tx_bytes": 0,
"rx_bytes": 0,
"bytes": 0,
"vwireEnabled": True,
"uplink_table": [],
"num_sta": 0,
"user-num_sta": 0,
"user-wlan-num_sta": 0,
"guest-num_sta": 0,
"guest-wlan-num_sta": 0,
"x_has_ssh_hostkey": False,
}
Add DPI Restriction switch to UniFi integration (#42499) * initial implementation for controlling DPI restrictions * address PR review comments and add DataUpdateCoordinator * fix existing tests against new lib version * add tests for DPI switches * bump aiounifi * listen to events instead of polling * fix tests * remove useless test * bump aiounifi * rename device to UniFi Controller per PR feedback
2020-11-03 07:36:37 +00:00
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
async def test_no_clients(hass, aioclient_mock):
UniFi POE control (#17011) * First commit * Feature complete? * Add dependency * Move setting poe mode logic to library * Use guard clauses * Bump requirement to 2 * Simplify saving switches with poe off * Store and use poe mode * Fix indentation * Fix flake8 * Configuration future proofing * Bump dependency to v3 * Add first test * Proper use of defaults with config flow (thanks helto) * Appease hound * Make sure there can't be duplicate entries of combination host+site * More tests * More tests * 98% coverage of controller * Fix hound comments * Config flow step init not necessary * Use async_current_entries to check if host and site for controller is used * Remove storing/restoring poe off devices to slim PR * First batch of switch tests * More switch tests. * Small improvements and clean up * Make tests pass Don't name device in device registry * Dont process clients that belong to non-UniFi POE switches * Allow selection of site from a list in config flow * Fix double blank lines in method * Update codeowners
2018-10-16 08:35:35 +00:00
"""Test the update_clients function when no clients are found."""
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
await setup_unifi_integration(
Upgrade black to 20.8b1 (#39287)
2020-08-27 11:56:20 +00:00
hass,
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
aioclient_mock,
Add DPI Restriction switch to UniFi integration (#42499) * initial implementation for controlling DPI restrictions * address PR review comments and add DataUpdateCoordinator * fix existing tests against new lib version * add tests for DPI switches * bump aiounifi * listen to events instead of polling * fix tests * remove useless test * bump aiounifi * rename device to UniFi Controller per PR feedback
2020-11-03 07:36:37 +00:00
options={
CONF_TRACK_CLIENTS: False,
CONF_TRACK_DEVICES: False,
CONF_DPI_RESTRICTIONS: False,
},
UniFi - Improve switch tests (#27200) * Continue rewriting tests for UniFi
2019-10-05 14:16:08 +00:00
)
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
assert aioclient_mock.call_count == 10
UniFi - Store controller in config_entry.entry_id (#34553) * Store controller in config_entry.entry_id * Clean up imports
2020-04-23 14:48:24 +00:00
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 0
UniFi POE control (#17011) * First commit * Feature complete? * Add dependency * Move setting poe mode logic to library * Use guard clauses * Bump requirement to 2 * Simplify saving switches with poe off * Store and use poe mode * Fix indentation * Fix flake8 * Configuration future proofing * Bump dependency to v3 * Add first test * Proper use of defaults with config flow (thanks helto) * Appease hound * Make sure there can't be duplicate entries of combination host+site * More tests * More tests * 98% coverage of controller * Fix hound comments * Config flow step init not necessary * Use async_current_entries to check if host and site for controller is used * Remove storing/restoring poe off devices to slim PR * First batch of switch tests * More switch tests. * Small improvements and clean up * Make tests pass Don't name device in device registry * Dont process clients that belong to non-UniFi POE switches * Allow selection of site from a list in config flow * Fix double blank lines in method * Update codeowners
2018-10-16 08:35:35 +00:00
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
async def test_controller_not_client(hass, aioclient_mock):
Don't create a switch for POE device if said device is Cloud key (#18117)
2018-11-02 20:09:16 +00:00
"""Test that the controller doesn't become a switch."""
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
await setup_unifi_integration(
UniFi - Improve switch tests (#27200) * Continue rewriting tests for UniFi
2019-10-05 14:16:08 +00:00
hass,
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
aioclient_mock,
UniFi - Add block network access control to config option (#32004) * Add block network access control to config option * Clean up
2020-03-05 05:55:56 +00:00
options={CONF_TRACK_CLIENTS: False, CONF_TRACK_DEVICES: False},
UniFi - Improve controller tests (#27261) * Improve controller tests and harmonize setup_unifi_integration to one * Store listeners to dispatchers to be used during reset
2019-10-07 19:55:35 +00:00
clients_response=[CONTROLLER_HOST],
UniFi - Improve switch tests (#27200) * Continue rewriting tests for UniFi
2019-10-05 14:16:08 +00:00
devices_response=[DEVICE_1],
)
UniFi - Store controller in config_entry.entry_id (#34553) * Store controller in config_entry.entry_id * Clean up imports
2020-04-23 14:48:24 +00:00
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 0
Black
2019-07-31 19:25:30 +00:00
cloudkey = hass.states.get("switch.cloud_key")
Don't create a switch for POE device if said device is Cloud key (#18117)
2018-11-02 20:09:16 +00:00
assert cloudkey is None
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
async def test_not_admin(hass, aioclient_mock):
Merge UniFi device tracker to config entry (#24367) * Move device tracker to use config entry * Remove monitored conditions attributes based on ADR0003 * Add support for import of device tracker config to be backwards compatible * Remove unnecessary configuration options from device tracker * Add component configuration support
2019-07-14 19:57:09 +00:00
"""Test that switch platform only work on an admin account."""
UniFi - Sites don't declare role on UniFiOS 1.7.0 beta (#35555) * New way to identify role, compatible with standalone controller or as part of unifios * Remove error * Bump dependency to v22 * Remove unused import
2020-05-12 22:37:16 +00:00
description = deepcopy(DESCRIPTION)
description[0]["site_role"] = "not admin"
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
await setup_unifi_integration(
UniFi - Improve switch tests (#27200) * Continue rewriting tests for UniFi
2019-10-05 14:16:08 +00:00
hass,
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
aioclient_mock,
UniFi - Add block network access control to config option (#32004) * Add block network access control to config option * Clean up
2020-03-05 05:55:56 +00:00
options={CONF_TRACK_CLIENTS: False, CONF_TRACK_DEVICES: False},
UniFi - Sites don't declare role on UniFiOS 1.7.0 beta (#35555) * New way to identify role, compatible with standalone controller or as part of unifios * Remove error * Bump dependency to v22 * Remove unused import
2020-05-12 22:37:16 +00:00
site_description=description,
UniFi - Improve switch tests (#27200) * Continue rewriting tests for UniFi
2019-10-05 14:16:08 +00:00
clients_response=[CLIENT_1],
devices_response=[DEVICE_1],
)
Merge UniFi device tracker to config entry (#24367) * Move device tracker to use config entry * Remove monitored conditions attributes based on ADR0003 * Add support for import of device tracker config to be backwards compatible * Remove unnecessary configuration options from device tracker * Add component configuration support
2019-07-14 19:57:09 +00:00
UniFi - Store controller in config_entry.entry_id (#34553) * Store controller in config_entry.entry_id * Clean up imports
2020-04-23 14:48:24 +00:00
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 0
Merge UniFi device tracker to config entry (#24367) * Move device tracker to use config entry * Remove monitored conditions attributes based on ADR0003 * Add support for import of device tracker config to be backwards compatible * Remove unnecessary configuration options from device tracker * Add component configuration support
2019-07-14 19:57:09 +00:00
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
async def test_switches(hass, aioclient_mock):
Merge UniFi device tracker to config entry (#24367) * Move device tracker to use config entry * Remove monitored conditions attributes based on ADR0003 * Add support for import of device tracker config to be backwards compatible * Remove unnecessary configuration options from device tracker * Add component configuration support
2019-07-14 19:57:09 +00:00
"""Test the update_items function with some clients."""
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
config_entry = await setup_unifi_integration(
UniFi - Improve switch tests (#27200) * Continue rewriting tests for UniFi
2019-10-05 14:16:08 +00:00
hass,
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
aioclient_mock,
UniFi - Improve switch tests (#27200) * Continue rewriting tests for UniFi
2019-10-05 14:16:08 +00:00
options={
UniFi - Add block network access control to config option (#32004) * Add block network access control to config option * Clean up
2020-03-05 05:55:56 +00:00
CONF_BLOCK_CLIENT: [BLOCKED["mac"], UNBLOCKED["mac"]],
CONF_TRACK_CLIENTS: False,
CONF_TRACK_DEVICES: False,
UniFi - Improve switch tests (#27200) * Continue rewriting tests for UniFi
2019-10-05 14:16:08 +00:00
},
Remove old UniFi POE client implementation (#81749) Remove all references to POE client implementation
2022-11-08 06:48:54 +00:00
clients_response=[CLIENT_4],
UniFi - Improve switch tests (#27200) * Continue rewriting tests for UniFi
2019-10-05 14:16:08 +00:00
clients_all_response=[BLOCKED, UNBLOCKED, CLIENT_1],
Add DPI Restriction switch to UniFi integration (#42499) * initial implementation for controlling DPI restrictions * address PR review comments and add DataUpdateCoordinator * fix existing tests against new lib version * add tests for DPI switches * bump aiounifi * listen to events instead of polling * fix tests * remove useless test * bump aiounifi * rename device to UniFi Controller per PR feedback
2020-11-03 07:36:37 +00:00
dpigroup_response=DPI_GROUPS,
dpiapp_response=DPI_APPS,
UniFi - Improve switch tests (#27200) * Continue rewriting tests for UniFi
2019-10-05 14:16:08 +00:00
)
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
controller = hass.data[UNIFI_DOMAIN][config_entry.entry_id]
Merge UniFi device tracker to config entry (#24367) * Move device tracker to use config entry * Remove monitored conditions attributes based on ADR0003 * Add support for import of device tracker config to be backwards compatible * Remove unnecessary configuration options from device tracker * Add component configuration support
2019-07-14 19:57:09 +00:00
Remove old UniFi POE client implementation (#81749) Remove all references to POE client implementation
2022-11-08 06:48:54 +00:00
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 3
Black
2019-07-31 19:25:30 +00:00
switch_4 = hass.states.get("switch.poe_client_4")
Merge UniFi device tracker to config entry (#24367) * Move device tracker to use config entry * Remove monitored conditions attributes based on ADR0003 * Add support for import of device tracker config to be backwards compatible * Remove unnecessary configuration options from device tracker * Add component configuration support
2019-07-14 19:57:09 +00:00
assert switch_4 is None
UniFi POE control (#17011) * First commit * Feature complete? * Add dependency * Move setting poe mode logic to library * Use guard clauses * Bump requirement to 2 * Simplify saving switches with poe off * Store and use poe mode * Fix indentation * Fix flake8 * Configuration future proofing * Bump dependency to v3 * Add first test * Proper use of defaults with config flow (thanks helto) * Appease hound * Make sure there can't be duplicate entries of combination host+site * More tests * More tests * 98% coverage of controller * Fix hound comments * Config flow step init not necessary * Use async_current_entries to check if host and site for controller is used * Remove storing/restoring poe off devices to slim PR * First batch of switch tests * More switch tests. * Small improvements and clean up * Make tests pass Don't name device in device registry * Dont process clients that belong to non-UniFi POE switches * Allow selection of site from a list in config flow * Fix double blank lines in method * Update codeowners
2018-10-16 08:35:35 +00:00
Black
2019-07-31 19:25:30 +00:00
blocked = hass.states.get("switch.block_client_1")
UniFi block clients (#25478) * Allow blocking clients on UniFi networks
2019-07-25 14:56:56 +00:00
assert blocked is not None
UniFi - reverse connectivity logic (#25691) * Make connectivity control in line with other implementations
2019-08-04 14:57:36 +00:00
assert blocked.state == "off"
UniFi block clients (#25478) * Allow blocking clients on UniFi networks
2019-07-25 14:56:56 +00:00
Black
2019-07-31 19:25:30 +00:00
unblocked = hass.states.get("switch.block_client_2")
UniFi block clients (#25478) * Allow blocking clients on UniFi networks
2019-07-25 14:56:56 +00:00
assert unblocked is not None
UniFi - reverse connectivity logic (#25691) * Make connectivity control in line with other implementations
2019-08-04 14:57:36 +00:00
assert unblocked.state == "on"
UniFi block clients (#25478) * Allow blocking clients on UniFi networks
2019-07-25 14:56:56 +00:00
Add DPI Restriction switch to UniFi integration (#42499) * initial implementation for controlling DPI restrictions * address PR review comments and add DataUpdateCoordinator * fix existing tests against new lib version * add tests for DPI switches * bump aiounifi * listen to events instead of polling * fix tests * remove useless test * bump aiounifi * rename device to UniFi Controller per PR feedback
2020-11-03 07:36:37 +00:00
dpi_switch = hass.states.get("switch.block_media_streaming")
assert dpi_switch is not None
assert dpi_switch.state == "on"
Increase test coverage of UniFi integration (#46347) * Increase coverage of init * Increase coverage of config_flow * Improve coverage of controller * Minor improvement to switch test * Fix review comment * Mock websocket class * Replace the rest of the old websocket event tests * Improve websocket fixture for cleaner tests * Fix typing * Improve connection state signalling based on Martins feedback * Improve tests of reconnection_mechanisms based on Martins review comments * Fix unload entry * Fix isort issue after rebase * Fix martins comment on not using caplog * Fix wireless clients test * Fix martins comments on wireless clients test
2021-03-05 20:28:41 +00:00
assert dpi_switch.attributes["icon"] == "mdi:network"
Add DPI Restriction switch to UniFi integration (#42499) * initial implementation for controlling DPI restrictions * address PR review comments and add DataUpdateCoordinator * fix existing tests against new lib version * add tests for DPI switches * bump aiounifi * listen to events instead of polling * fix tests * remove useless test * bump aiounifi * rename device to UniFi Controller per PR feedback
2020-11-03 07:36:37 +00:00
Add entity category to UniFi sensors and switches (#58484)
2021-10-26 18:23:20 +00:00
ent_reg = er.async_get(hass)
Remove old UniFi POE client implementation (#81749) Remove all references to POE client implementation
2022-11-08 06:48:54 +00:00
for entry_id in ("switch.block_client_1", "switch.block_media_streaming"):
Use enums in unifi tests (#62151)
2021-12-17 10:12:43 +00:00
assert ent_reg.async_get(entry_id).entity_category is EntityCategory.CONFIG
Add entity category to UniFi sensors and switches (#58484)
2021-10-26 18:23:20 +00:00
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
# Block and unblock client
aioclient_mock.post(
f"https://{controller.host}:1234/api/s/{controller.site}/cmd/stamgr",
)
UniFi - Add block network access control to config option (#32004) * Add block network access control to config option * Clean up
2020-03-05 05:55:56 +00:00
await hass.services.async_call(
UniFi - Store controller in config_entry.entry_id (#34553) * Store controller in config_entry.entry_id * Clean up imports
2020-04-23 14:48:24 +00:00
SWITCH_DOMAIN, "turn_off", {"entity_id": "switch.block_client_1"}, blocking=True
UniFi - Add block network access control to config option (#32004) * Add block network access control to config option * Clean up
2020-03-05 05:55:56 +00:00
)
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
assert aioclient_mock.call_count == 11
assert aioclient_mock.mock_calls[10][2] == {
"mac": "00:00:00:00:01:01",
"cmd": "block-sta",
UniFi - Add block network access control to config option (#32004) * Add block network access control to config option * Clean up
2020-03-05 05:55:56 +00:00
}
await hass.services.async_call(
UniFi - Store controller in config_entry.entry_id (#34553) * Store controller in config_entry.entry_id * Clean up imports
2020-04-23 14:48:24 +00:00
SWITCH_DOMAIN, "turn_on", {"entity_id": "switch.block_client_1"}, blocking=True
UniFi - Add block network access control to config option (#32004) * Add block network access control to config option * Clean up
2020-03-05 05:55:56 +00:00
)
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
assert aioclient_mock.call_count == 12
assert aioclient_mock.mock_calls[11][2] == {
"mac": "00:00:00:00:01:01",
"cmd": "unblock-sta",
UniFi - Add block network access control to config option (#32004) * Add block network access control to config option * Clean up
2020-03-05 05:55:56 +00:00
}
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
# Enable and disable DPI
aioclient_mock.put(
f"https://{controller.host}:1234/api/s/{controller.site}/rest/dpiapp/5f976f62e3c58f018ec7e17d",
)
Add DPI Restriction switch to UniFi integration (#42499) * initial implementation for controlling DPI restrictions * address PR review comments and add DataUpdateCoordinator * fix existing tests against new lib version * add tests for DPI switches * bump aiounifi * listen to events instead of polling * fix tests * remove useless test * bump aiounifi * rename device to UniFi Controller per PR feedback
2020-11-03 07:36:37 +00:00
await hass.services.async_call(
SWITCH_DOMAIN,
"turn_off",
{"entity_id": "switch.block_media_streaming"},
blocking=True,
)
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
assert aioclient_mock.call_count == 13
assert aioclient_mock.mock_calls[12][2] == {"enabled": False}
Add DPI Restriction switch to UniFi integration (#42499) * initial implementation for controlling DPI restrictions * address PR review comments and add DataUpdateCoordinator * fix existing tests against new lib version * add tests for DPI switches * bump aiounifi * listen to events instead of polling * fix tests * remove useless test * bump aiounifi * rename device to UniFi Controller per PR feedback
2020-11-03 07:36:37 +00:00
await hass.services.async_call(
SWITCH_DOMAIN,
"turn_on",
{"entity_id": "switch.block_media_streaming"},
blocking=True,
)
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
assert aioclient_mock.call_count == 14
assert aioclient_mock.mock_calls[13][2] == {"enabled": True}
Add DPI Restriction switch to UniFi integration (#42499) * initial implementation for controlling DPI restrictions * address PR review comments and add DataUpdateCoordinator * fix existing tests against new lib version * add tests for DPI switches * bump aiounifi * listen to events instead of polling * fix tests * remove useless test * bump aiounifi * rename device to UniFi Controller per PR feedback
2020-11-03 07:36:37 +00:00
Increase test coverage of UniFi integration (#46347) * Increase coverage of init * Increase coverage of config_flow * Improve coverage of controller * Minor improvement to switch test * Fix review comment * Mock websocket class * Replace the rest of the old websocket event tests * Improve websocket fixture for cleaner tests * Fix typing * Improve connection state signalling based on Martins feedback * Improve tests of reconnection_mechanisms based on Martins review comments * Fix unload entry * Fix isort issue after rebase * Fix martins comment on not using caplog * Fix wireless clients test * Fix martins comments on wireless clients test
2021-03-05 20:28:41 +00:00
# Make sure no duplicates arise on generic signal update
async_dispatcher_send(hass, controller.signal_update)
await hass.async_block_till_done()
Remove old UniFi POE client implementation (#81749) Remove all references to POE client implementation
2022-11-08 06:48:54 +00:00
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 3
Increase test coverage of UniFi integration (#46347) * Increase coverage of init * Increase coverage of config_flow * Improve coverage of controller * Minor improvement to switch test * Fix review comment * Mock websocket class * Replace the rest of the old websocket event tests * Improve websocket fixture for cleaner tests * Fix typing * Improve connection state signalling based on Martins feedback * Improve tests of reconnection_mechanisms based on Martins review comments * Fix unload entry * Fix isort issue after rebase * Fix martins comment on not using caplog * Fix wireless clients test * Fix martins comments on wireless clients test
2021-03-05 20:28:41 +00:00
UniFi POE control (#17011) * First commit * Feature complete? * Add dependency * Move setting poe mode logic to library * Use guard clauses * Bump requirement to 2 * Simplify saving switches with poe off * Store and use poe mode * Fix indentation * Fix flake8 * Configuration future proofing * Bump dependency to v3 * Add first test * Proper use of defaults with config flow (thanks helto) * Appease hound * Make sure there can't be duplicate entries of combination host+site * More tests * More tests * 98% coverage of controller * Fix hound comments * Config flow step init not necessary * Use async_current_entries to check if host and site for controller is used * Remove storing/restoring poe off devices to slim PR * First batch of switch tests * More switch tests. * Small improvements and clean up * Make tests pass Don't name device in device registry * Dont process clients that belong to non-UniFi POE switches * Allow selection of site from a list in config flow * Fix double blank lines in method * Update codeowners
2018-10-16 08:35:35 +00:00
Increase test coverage of UniFi integration (#46347) * Increase coverage of init * Increase coverage of config_flow * Improve coverage of controller * Minor improvement to switch test * Fix review comment * Mock websocket class * Replace the rest of the old websocket event tests * Improve websocket fixture for cleaner tests * Fix typing * Improve connection state signalling based on Martins feedback * Improve tests of reconnection_mechanisms based on Martins review comments * Fix unload entry * Fix isort issue after rebase * Fix martins comment on not using caplog * Fix wireless clients test * Fix martins comments on wireless clients test
2021-03-05 20:28:41 +00:00
async def test_remove_switches(hass, aioclient_mock, mock_unifi_websocket):
UniFi - Support automatic removal of clients (#34307)
2020-04-16 22:08:53 +00:00
"""Test the update_items function with some clients."""
Increase test coverage of UniFi integration (#46347) * Increase coverage of init * Increase coverage of config_flow * Improve coverage of controller * Minor improvement to switch test * Fix review comment * Mock websocket class * Replace the rest of the old websocket event tests * Improve websocket fixture for cleaner tests * Fix typing * Improve connection state signalling based on Martins feedback * Improve tests of reconnection_mechanisms based on Martins review comments * Fix unload entry * Fix isort issue after rebase * Fix martins comment on not using caplog * Fix wireless clients test * Fix martins comments on wireless clients test
2021-03-05 20:28:41 +00:00
await setup_unifi_integration(
UniFi - Support automatic removal of clients (#34307)
2020-04-16 22:08:53 +00:00
hass,
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
aioclient_mock,
UniFi - Support automatic removal of clients (#34307)
2020-04-16 22:08:53 +00:00
options={CONF_BLOCK_CLIENT: [UNBLOCKED["mac"]]},
Remove old UniFi POE client implementation (#81749) Remove all references to POE client implementation
2022-11-08 06:48:54 +00:00
clients_response=[UNBLOCKED],
Fix issue where UniFi DPI groups state wasn't being updated (#58502)
2021-10-27 08:35:44 +00:00
dpigroup_response=DPI_GROUPS,
dpiapp_response=DPI_APPS,
UniFi - Support automatic removal of clients (#34307)
2020-04-16 22:08:53 +00:00
)
Increase test coverage of UniFi integration (#46347) * Increase coverage of init * Increase coverage of config_flow * Improve coverage of controller * Minor improvement to switch test * Fix review comment * Mock websocket class * Replace the rest of the old websocket event tests * Improve websocket fixture for cleaner tests * Fix typing * Improve connection state signalling based on Martins feedback * Improve tests of reconnection_mechanisms based on Martins review comments * Fix unload entry * Fix isort issue after rebase * Fix martins comment on not using caplog * Fix wireless clients test * Fix martins comments on wireless clients test
2021-03-05 20:28:41 +00:00
Remove old UniFi POE client implementation (#81749) Remove all references to POE client implementation
2022-11-08 06:48:54 +00:00
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 2
UniFi - Support automatic removal of clients (#34307)
2020-04-16 22:08:53 +00:00
Fix issue where UniFi DPI groups state wasn't being updated (#58502)
2021-10-27 08:35:44 +00:00
assert hass.states.get("switch.block_client_2") is not None
assert hass.states.get("switch.block_media_streaming") is not None
UniFi - Support automatic removal of clients (#34307)
2020-04-16 22:08:53 +00:00
Remove old UniFi POE client implementation (#81749) Remove all references to POE client implementation
2022-11-08 06:48:54 +00:00
mock_unifi_websocket(message=MessageKey.CLIENT_REMOVED, data=[UNBLOCKED])
UniFi - Support automatic removal of clients (#34307)
2020-04-16 22:08:53 +00:00
await hass.async_block_till_done()
Fix issue where UniFi DPI groups state wasn't being updated (#58502)
2021-10-27 08:35:44 +00:00
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 1
assert hass.states.get("switch.block_client_2") is None
assert hass.states.get("switch.block_media_streaming") is not None
UniFi - Support automatic removal of clients (#34307)
2020-04-16 22:08:53 +00:00
Fix issue where UniFi DPI groups state wasn't being updated (#58502)
2021-10-27 08:35:44 +00:00
mock_unifi_websocket(data=DPI_GROUP_REMOVED_EVENT)
await hass.async_block_till_done()
UniFi - Support automatic removal of clients (#34307)
2020-04-16 22:08:53 +00:00
Fix issue where UniFi DPI groups state wasn't being updated (#58502)
2021-10-27 08:35:44 +00:00
assert hass.states.get("switch.block_media_streaming") is None
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 0
UniFi - Support automatic removal of clients (#34307)
2020-04-16 22:08:53 +00:00
Increase test coverage of UniFi integration (#46347) * Increase coverage of init * Increase coverage of config_flow * Improve coverage of controller * Minor improvement to switch test * Fix review comment * Mock websocket class * Replace the rest of the old websocket event tests * Improve websocket fixture for cleaner tests * Fix typing * Improve connection state signalling based on Martins feedback * Improve tests of reconnection_mechanisms based on Martins review comments * Fix unload entry * Fix isort issue after rebase * Fix martins comment on not using caplog * Fix wireless clients test * Fix martins comments on wireless clients test
2021-03-05 20:28:41 +00:00
async def test_block_switches(hass, aioclient_mock, mock_unifi_websocket):
UniFi - Make clients proper push based (#35273) * Improve client tracker to be more comprehensible and streamlined Improve block switches Improve tests * Small clean up * Add descriptions on ssid test * Improve test * Make polling default off, only POE clients left to verify * Minor improvements * On removal cancel scheduled updates * POE works without polling now * Combine else and if to an elif
2020-05-08 20:19:27 +00:00
"""Test the update_items function with some clients."""
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
config_entry = await setup_unifi_integration(
UniFi - Make clients proper push based (#35273) * Improve client tracker to be more comprehensible and streamlined Improve block switches Improve tests * Small clean up * Add descriptions on ssid test * Improve test * Make polling default off, only POE clients left to verify * Minor improvements * On removal cancel scheduled updates * POE works without polling now * Combine else and if to an elif
2020-05-08 20:19:27 +00:00
hass,
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
aioclient_mock,
UniFi - Make clients proper push based (#35273) * Improve client tracker to be more comprehensible and streamlined Improve block switches Improve tests * Small clean up * Add descriptions on ssid test * Improve test * Make polling default off, only POE clients left to verify * Minor improvements * On removal cancel scheduled updates * POE works without polling now * Combine else and if to an elif
2020-05-08 20:19:27 +00:00
options={
CONF_BLOCK_CLIENT: [BLOCKED["mac"], UNBLOCKED["mac"]],
CONF_TRACK_CLIENTS: False,
CONF_TRACK_DEVICES: False,
},
clients_response=[UNBLOCKED],
clients_all_response=[BLOCKED],
)
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
controller = hass.data[UNIFI_DOMAIN][config_entry.entry_id]
UniFi - Make clients proper push based (#35273) * Improve client tracker to be more comprehensible and streamlined Improve block switches Improve tests * Small clean up * Add descriptions on ssid test * Improve test * Make polling default off, only POE clients left to verify * Minor improvements * On removal cancel scheduled updates * POE works without polling now * Combine else and if to an elif
2020-05-08 20:19:27 +00:00
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"
Simplify mocking UniFi websocket (#80652) * Simplify mocking UniFi websocket * Fix one more
2022-10-20 17:37:13 +00:00
mock_unifi_websocket(message=MessageKey.EVENT, data=EVENT_BLOCKED_CLIENT_UNBLOCKED)
UniFi - Make clients proper push based (#35273) * Improve client tracker to be more comprehensible and streamlined Improve block switches Improve tests * Small clean up * Add descriptions on ssid test * Improve test * Make polling default off, only POE clients left to verify * Minor improvements * On removal cancel scheduled updates * POE works without polling now * Combine else and if to an elif
2020-05-08 20:19:27 +00:00
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"
Simplify mocking UniFi websocket (#80652) * Simplify mocking UniFi websocket * Fix one more
2022-10-20 17:37:13 +00:00
mock_unifi_websocket(message=MessageKey.EVENT, data=EVENT_BLOCKED_CLIENT_BLOCKED)
UniFi - Make clients proper push based (#35273) * Improve client tracker to be more comprehensible and streamlined Improve block switches Improve tests * Small clean up * Add descriptions on ssid test * Improve test * Make polling default off, only POE clients left to verify * Minor improvements * On removal cancel scheduled updates * POE works without polling now * Combine else and if to an elif
2020-05-08 20:19:27 +00:00
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"
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
aioclient_mock.post(
f"https://{controller.host}:1234/api/s/{controller.site}/cmd/stamgr",
)
UniFi - Make clients proper push based (#35273) * Improve client tracker to be more comprehensible and streamlined Improve block switches Improve tests * Small clean up * Add descriptions on ssid test * Improve test * Make polling default off, only POE clients left to verify * Minor improvements * On removal cancel scheduled updates * POE works without polling now * Combine else and if to an elif
2020-05-08 20:19:27 +00:00
await hass.services.async_call(
SWITCH_DOMAIN, "turn_off", {"entity_id": "switch.block_client_1"}, blocking=True
)
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
assert aioclient_mock.call_count == 11
assert aioclient_mock.mock_calls[10][2] == {
"mac": "00:00:00:00:01:01",
"cmd": "block-sta",
UniFi - Make clients proper push based (#35273) * Improve client tracker to be more comprehensible and streamlined Improve block switches Improve tests * Small clean up * Add descriptions on ssid test * Improve test * Make polling default off, only POE clients left to verify * Minor improvements * On removal cancel scheduled updates * POE works without polling now * Combine else and if to an elif
2020-05-08 20:19:27 +00:00
}
await hass.services.async_call(
SWITCH_DOMAIN, "turn_on", {"entity_id": "switch.block_client_1"}, blocking=True
)
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
assert aioclient_mock.call_count == 12
assert aioclient_mock.mock_calls[11][2] == {
"mac": "00:00:00:00:01:01",
"cmd": "unblock-sta",
UniFi - Make clients proper push based (#35273) * Improve client tracker to be more comprehensible and streamlined Improve block switches Improve tests * Small clean up * Add descriptions on ssid test * Improve test * Make polling default off, only POE clients left to verify * Minor improvements * On removal cancel scheduled updates * POE works without polling now * Combine else and if to an elif
2020-05-08 20:19:27 +00:00
}
Fix issue where UniFi DPI groups state wasn't being updated (#58502)
2021-10-27 08:35:44 +00:00
async def test_dpi_switches(hass, aioclient_mock, mock_unifi_websocket):
"""Test the update_items function with some clients."""
await setup_unifi_integration(
hass,
aioclient_mock,
dpigroup_response=DPI_GROUPS,
dpiapp_response=DPI_APPS,
)
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 1
dpi_switch = hass.states.get("switch.block_media_streaming")
assert dpi_switch is not None
assert dpi_switch.state == STATE_ON
assert dpi_switch.attributes["icon"] == "mdi:network"
mock_unifi_websocket(data=DPI_APP_DISABLED_EVENT)
await hass.async_block_till_done()
assert hass.states.get("switch.block_media_streaming").state == STATE_OFF
Refactor UniFi DPI switch entities (#80761) * Refactor UniFi DPI switch entities * Remove dpi presence from items_added
2022-10-23 20:42:24 +00:00
# Availability signalling
# Controller disconnects
mock_unifi_websocket(state=WebsocketState.DISCONNECTED)
Move DPI Group logic to UniFi switch platform (#58761) * Library has normalized management of DPI apps and groups, move logic to UniFi integration * Bump dependency to v29 * Use a generator instead of a list - Pylint * Minor improvements * Improve doc strings
2022-01-12 16:11:05 +00:00
await hass.async_block_till_done()
Refactor UniFi DPI switch entities (#80761) * Refactor UniFi DPI switch entities * Remove dpi presence from items_added
2022-10-23 20:42:24 +00:00
assert hass.states.get("switch.block_media_streaming").state == STATE_UNAVAILABLE
# Controller reconnects
mock_unifi_websocket(state=WebsocketState.RUNNING)
Move DPI Group logic to UniFi switch platform (#58761) * Library has normalized management of DPI apps and groups, move logic to UniFi integration * Bump dependency to v29 * Use a generator instead of a list - Pylint * Minor improvements * Improve doc strings
2022-01-12 16:11:05 +00:00
await hass.async_block_till_done()
Refactor UniFi DPI switch entities (#80761) * Refactor UniFi DPI switch entities * Remove dpi presence from items_added
2022-10-23 20:42:24 +00:00
assert hass.states.get("switch.block_media_streaming").state == STATE_OFF
# Remove app
mock_unifi_websocket(data=DPI_GROUP_REMOVE_APP)
Move DPI Group logic to UniFi switch platform (#58761) * Library has normalized management of DPI apps and groups, move logic to UniFi integration * Bump dependency to v29 * Use a generator instead of a list - Pylint * Minor improvements * Improve doc strings
2022-01-12 16:11:05 +00:00
await hass.async_block_till_done()
assert hass.states.get("switch.block_media_streaming") is None
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 0
async def test_dpi_switches_add_second_app(hass, aioclient_mock, mock_unifi_websocket):
"""Test the update_items function with some clients."""
await setup_unifi_integration(
hass,
aioclient_mock,
dpigroup_response=DPI_GROUPS,
dpiapp_response=DPI_APPS,
)
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 1
assert hass.states.get("switch.block_media_streaming").state == STATE_ON
second_app_event = {
Simplify mocking UniFi websocket (#80652) * Simplify mocking UniFi websocket * Fix one more
2022-10-20 17:37:13 +00:00
"apps": [524292],
"blocked": False,
"cats": [],
"enabled": False,
"log": False,
"site_id": "name",
"_id": "61783e89c1773a18c0c61f00",
Move DPI Group logic to UniFi switch platform (#58761) * Library has normalized management of DPI apps and groups, move logic to UniFi integration * Bump dependency to v29 * Use a generator instead of a list - Pylint * Minor improvements * Improve doc strings
2022-01-12 16:11:05 +00:00
}
Simplify mocking UniFi websocket (#80652) * Simplify mocking UniFi websocket * Fix one more
2022-10-20 17:37:13 +00:00
mock_unifi_websocket(message=MessageKey.DPI_APP_ADDED, data=second_app_event)
Move DPI Group logic to UniFi switch platform (#58761) * Library has normalized management of DPI apps and groups, move logic to UniFi integration * Bump dependency to v29 * Use a generator instead of a list - Pylint * Minor improvements * Improve doc strings
2022-01-12 16:11:05 +00:00
await hass.async_block_till_done()
assert hass.states.get("switch.block_media_streaming").state == STATE_ON
add_second_app_to_group = {
Simplify mocking UniFi websocket (#80652) * Simplify mocking UniFi websocket * Fix one more
2022-10-20 17:37:13 +00:00
"_id": "5f976f4ae3c58f018ec7dff6",
"name": "Block Media Streaming",
"site_id": "name",
"dpiapp_ids": ["5f976f62e3c58f018ec7e17d", "61783e89c1773a18c0c61f00"],
Move DPI Group logic to UniFi switch platform (#58761) * Library has normalized management of DPI apps and groups, move logic to UniFi integration * Bump dependency to v29 * Use a generator instead of a list - Pylint * Minor improvements * Improve doc strings
2022-01-12 16:11:05 +00:00
}
Simplify mocking UniFi websocket (#80652) * Simplify mocking UniFi websocket * Fix one more
2022-10-20 17:37:13 +00:00
mock_unifi_websocket(
message=MessageKey.DPI_GROUP_UPDATED, data=add_second_app_to_group
)
Move DPI Group logic to UniFi switch platform (#58761) * Library has normalized management of DPI apps and groups, move logic to UniFi integration * Bump dependency to v29 * Use a generator instead of a list - Pylint * Minor improvements * Improve doc strings
2022-01-12 16:11:05 +00:00
await hass.async_block_till_done()
assert hass.states.get("switch.block_media_streaming").state == STATE_OFF
second_app_event_enabled = {
Simplify mocking UniFi websocket (#80652) * Simplify mocking UniFi websocket * Fix one more
2022-10-20 17:37:13 +00:00
"apps": [524292],
"blocked": False,
"cats": [],
"enabled": True,
"log": False,
"site_id": "name",
"_id": "61783e89c1773a18c0c61f00",
Move DPI Group logic to UniFi switch platform (#58761) * Library has normalized management of DPI apps and groups, move logic to UniFi integration * Bump dependency to v29 * Use a generator instead of a list - Pylint * Minor improvements * Improve doc strings
2022-01-12 16:11:05 +00:00
}
Simplify mocking UniFi websocket (#80652) * Simplify mocking UniFi websocket * Fix one more
2022-10-20 17:37:13 +00:00
mock_unifi_websocket(
message=MessageKey.DPI_APP_UPDATED, data=second_app_event_enabled
)
Move DPI Group logic to UniFi switch platform (#58761) * Library has normalized management of DPI apps and groups, move logic to UniFi integration * Bump dependency to v29 * Use a generator instead of a list - Pylint * Minor improvements * Improve doc strings
2022-01-12 16:11:05 +00:00
await hass.async_block_till_done()
assert hass.states.get("switch.block_media_streaming").state == STATE_ON
Fix issue where UniFi DPI groups state wasn't being updated (#58502)
2021-10-27 08:35:44 +00:00
Add outlets to UniFi switch platform (#64859) * Add outlets to UniFi switch platform * Outlet shouldn't have entity category set
2022-01-25 06:49:02 +00:00
async def test_outlet_switches(hass, aioclient_mock, mock_unifi_websocket):
Refactor UniFi outlet switches (#80738) * Rewrite UniFi outlet switches * Bump aiounifi to v41 * Remove devices from items_added input
2022-10-23 18:28:45 +00:00
"""Test the outlet entities."""
Add outlets to UniFi switch platform (#64859) * Add outlets to UniFi switch platform * Outlet shouldn't have entity category set
2022-01-25 06:49:02 +00:00
config_entry = await setup_unifi_integration(
Refactor UniFi outlet switches (#80738) * Rewrite UniFi outlet switches * Bump aiounifi to v41 * Remove devices from items_added input
2022-10-23 18:28:45 +00:00
hass, aioclient_mock, devices_response=[OUTLET_UP1]
Add outlets to UniFi switch platform (#64859) * Add outlets to UniFi switch platform * Outlet shouldn't have entity category set
2022-01-25 06:49:02 +00:00
)
controller = hass.data[UNIFI_DOMAIN][config_entry.entry_id]
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 1
Refactor UniFi outlet switches (#80738) * Rewrite UniFi outlet switches * Bump aiounifi to v41 * Remove devices from items_added input
2022-10-23 18:28:45 +00:00
# Validate state object
switch_1 = hass.states.get("switch.plug_outlet_1")
assert switch_1 is not None
assert switch_1.state == STATE_ON
assert switch_1.attributes.get(ATTR_DEVICE_CLASS) == SwitchDeviceClass.OUTLET
Add outlets to UniFi switch platform (#64859) * Add outlets to UniFi switch platform * Outlet shouldn't have entity category set
2022-01-25 06:49:02 +00:00
Refactor UniFi outlet switches (#80738) * Rewrite UniFi outlet switches * Bump aiounifi to v41 * Remove devices from items_added input
2022-10-23 18:28:45 +00:00
# Update state object
device_1 = deepcopy(OUTLET_UP1)
device_1["outlet_table"][0]["relay_state"] = False
mock_unifi_websocket(message=MessageKey.DEVICE, data=device_1)
Add outlets to UniFi switch platform (#64859) * Add outlets to UniFi switch platform * Outlet shouldn't have entity category set
2022-01-25 06:49:02 +00:00
await hass.async_block_till_done()
Refactor UniFi outlet switches (#80738) * Rewrite UniFi outlet switches * Bump aiounifi to v41 * Remove devices from items_added input
2022-10-23 18:28:45 +00:00
assert hass.states.get("switch.plug_outlet_1").state == STATE_OFF
Add outlets to UniFi switch platform (#64859) * Add outlets to UniFi switch platform * Outlet shouldn't have entity category set
2022-01-25 06:49:02 +00:00
Refactor UniFi outlet switches (#80738) * Rewrite UniFi outlet switches * Bump aiounifi to v41 * Remove devices from items_added input
2022-10-23 18:28:45 +00:00
# Turn off outlet
Add outlets to UniFi switch platform (#64859) * Add outlets to UniFi switch platform * Outlet shouldn't have entity category set
2022-01-25 06:49:02 +00:00
aioclient_mock.clear_requests()
aioclient_mock.put(
f"https://{controller.host}:1234/api/s/{controller.site}/rest/device/600c8356942a6ade50707b56",
)
await hass.services.async_call(
SWITCH_DOMAIN,
Refactor UniFi outlet switches (#80738) * Rewrite UniFi outlet switches * Bump aiounifi to v41 * Remove devices from items_added input
2022-10-23 18:28:45 +00:00
SERVICE_TURN_OFF,
Add outlets to UniFi switch platform (#64859) * Add outlets to UniFi switch platform * Outlet shouldn't have entity category set
2022-01-25 06:49:02 +00:00
{ATTR_ENTITY_ID: "switch.plug_outlet_1"},
blocking=True,
)
assert aioclient_mock.call_count == 1
assert aioclient_mock.mock_calls[0][2] == {
Refactor UniFi outlet switches (#80738) * Rewrite UniFi outlet switches * Bump aiounifi to v41 * Remove devices from items_added input
2022-10-23 18:28:45 +00:00
"outlet_overrides": [{"index": 1, "name": "Outlet 1", "relay_state": False}]
Add outlets to UniFi switch platform (#64859) * Add outlets to UniFi switch platform * Outlet shouldn't have entity category set
2022-01-25 06:49:02 +00:00
}
Refactor UniFi outlet switches (#80738) * Rewrite UniFi outlet switches * Bump aiounifi to v41 * Remove devices from items_added input
2022-10-23 18:28:45 +00:00
# Turn on outlet
Add outlets to UniFi switch platform (#64859) * Add outlets to UniFi switch platform * Outlet shouldn't have entity category set
2022-01-25 06:49:02 +00:00
await hass.services.async_call(
SWITCH_DOMAIN,
Refactor UniFi outlet switches (#80738) * Rewrite UniFi outlet switches * Bump aiounifi to v41 * Remove devices from items_added input
2022-10-23 18:28:45 +00:00
SERVICE_TURN_ON,
Add outlets to UniFi switch platform (#64859) * Add outlets to UniFi switch platform * Outlet shouldn't have entity category set
2022-01-25 06:49:02 +00:00
{ATTR_ENTITY_ID: "switch.plug_outlet_1"},
blocking=True,
)
assert aioclient_mock.call_count == 2
assert aioclient_mock.mock_calls[1][2] == {
Refactor UniFi outlet switches (#80738) * Rewrite UniFi outlet switches * Bump aiounifi to v41 * Remove devices from items_added input
2022-10-23 18:28:45 +00:00
"outlet_overrides": [{"index": 1, "name": "Outlet 1", "relay_state": True}]
Add outlets to UniFi switch platform (#64859) * Add outlets to UniFi switch platform * Outlet shouldn't have entity category set
2022-01-25 06:49:02 +00:00
}
Refactor UniFi outlet switches (#80738) * Rewrite UniFi outlet switches * Bump aiounifi to v41 * Remove devices from items_added input
2022-10-23 18:28:45 +00:00
# Availability signalling
# Controller disconnects
mock_unifi_websocket(state=WebsocketState.DISCONNECTED)
Add outlets to UniFi switch platform (#64859) * Add outlets to UniFi switch platform * Outlet shouldn't have entity category set
2022-01-25 06:49:02 +00:00
await hass.async_block_till_done()
Refactor UniFi outlet switches (#80738) * Rewrite UniFi outlet switches * Bump aiounifi to v41 * Remove devices from items_added input
2022-10-23 18:28:45 +00:00
assert hass.states.get("switch.plug_outlet_1").state == STATE_UNAVAILABLE
# Controller reconnects
mock_unifi_websocket(state=WebsocketState.RUNNING)
await hass.async_block_till_done()
assert hass.states.get("switch.plug_outlet_1").state == STATE_OFF
# Device gets disabled
device_1["disabled"] = True
mock_unifi_websocket(message=MessageKey.DEVICE, data=device_1)
await hass.async_block_till_done()
assert hass.states.get("switch.plug_outlet_1").state == STATE_UNAVAILABLE
# Device gets re-enabled
device_1["disabled"] = False
mock_unifi_websocket(message=MessageKey.DEVICE, data=device_1)
await hass.async_block_till_done()
assert hass.states.get("switch.plug_outlet_1").state == STATE_OFF
Add outlets to UniFi switch platform (#64859) * Add outlets to UniFi switch platform * Outlet shouldn't have entity category set
2022-01-25 06:49:02 +00:00
# Unload config entry
await hass.config_entries.async_unload(config_entry.entry_id)
assert hass.states.get("switch.plug_outlet_1").state == STATE_UNAVAILABLE
# Remove config entry
await hass.config_entries.async_remove(config_entry.entry_id)
await hass.async_block_till_done()
assert hass.states.get("switch.plug_outlet_1") is None
Increase test coverage of UniFi integration (#46347) * Increase coverage of init * Increase coverage of config_flow * Improve coverage of controller * Minor improvement to switch test * Fix review comment * Mock websocket class * Replace the rest of the old websocket event tests * Improve websocket fixture for cleaner tests * Fix typing * Improve connection state signalling based on Martins feedback * Improve tests of reconnection_mechanisms based on Martins review comments * Fix unload entry * Fix isort issue after rebase * Fix martins comment on not using caplog * Fix wireless clients test * Fix martins comments on wireless clients test
2021-03-05 20:28:41 +00:00
async def test_new_client_discovered_on_block_control(
hass, aioclient_mock, mock_unifi_websocket
):
UniFi - Improve controller tests (#27261) * Improve controller tests and harmonize setup_unifi_integration to one * Store listeners to dispatchers to be used during reset
2019-10-07 19:55:35 +00:00
"""Test if 2nd update has a new client."""
Increase test coverage of UniFi integration (#46347) * Increase coverage of init * Increase coverage of config_flow * Improve coverage of controller * Minor improvement to switch test * Fix review comment * Mock websocket class * Replace the rest of the old websocket event tests * Improve websocket fixture for cleaner tests * Fix typing * Improve connection state signalling based on Martins feedback * Improve tests of reconnection_mechanisms based on Martins review comments * Fix unload entry * Fix isort issue after rebase * Fix martins comment on not using caplog * Fix wireless clients test * Fix martins comments on wireless clients test
2021-03-05 20:28:41 +00:00
await setup_unifi_integration(
UniFi - Improve controller tests (#27261) * Improve controller tests and harmonize setup_unifi_integration to one * Store listeners to dispatchers to be used during reset
2019-10-07 19:55:35 +00:00
hass,
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
aioclient_mock,
UniFi - Improve controller tests (#27261) * Improve controller tests and harmonize setup_unifi_integration to one * Store listeners to dispatchers to be used during reset
2019-10-07 19:55:35 +00:00
options={
UniFi - Add block network access control to config option (#32004) * Add block network access control to config option * Clean up
2020-03-05 05:55:56 +00:00
CONF_BLOCK_CLIENT: [BLOCKED["mac"]],
CONF_TRACK_CLIENTS: False,
CONF_TRACK_DEVICES: False,
Add DPI Restriction switch to UniFi integration (#42499) * initial implementation for controlling DPI restrictions * address PR review comments and add DataUpdateCoordinator * fix existing tests against new lib version * add tests for DPI switches * bump aiounifi * listen to events instead of polling * fix tests * remove useless test * bump aiounifi * rename device to UniFi Controller per PR feedback
2020-11-03 07:36:37 +00:00
CONF_DPI_RESTRICTIONS: False,
UniFi - Improve controller tests (#27261) * Improve controller tests and harmonize setup_unifi_integration to one * Store listeners to dispatchers to be used during reset
2019-10-07 19:55:35 +00:00
},
)
UniFi - Store controller in config_entry.entry_id (#34553) * Store controller in config_entry.entry_id * Clean up imports
2020-04-23 14:48:24 +00:00
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 0
Rewrite UniFi block client switch (#80969) * Refactor UniFi block client switch entities * Use new switch loader * Rename lambdas * Use is_on rather than _attr_is_on when applicable
2022-10-25 20:36:51 +00:00
assert hass.states.get("switch.block_client_1") is None
UniFi - Improve controller tests (#27261) * Improve controller tests and harmonize setup_unifi_integration to one * Store listeners to dispatchers to be used during reset
2019-10-07 19:55:35 +00:00
Simplify mocking UniFi websocket (#80652) * Simplify mocking UniFi websocket * Fix one more
2022-10-20 17:37:13 +00:00
mock_unifi_websocket(message=MessageKey.CLIENT, data=BLOCKED)
UniFi - Improve signalling and handling of adding new entities (#34925) * Change signalling and handling of adding devices * Bump aiounifi to v19 Use the proper variable for roamradio
2020-05-04 17:29:49 +00:00
await hass.async_block_till_done()
UniFi - Store controller in config_entry.entry_id (#34553) * Store controller in config_entry.entry_id * Clean up imports
2020-04-23 14:48:24 +00:00
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 1
Rewrite UniFi block client switch (#80969) * Refactor UniFi block client switch entities * Use new switch loader * Rename lambdas * Use is_on rather than _attr_is_on when applicable
2022-10-25 20:36:51 +00:00
assert hass.states.get("switch.block_client_1") is not None
UniFi - Add block network access control to config option (#32004) * Add block network access control to config option * Clean up
2020-03-05 05:55:56 +00:00
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
async def test_option_block_clients(hass, aioclient_mock):
UniFi - Add block network access control to config option (#32004) * Add block network access control to config option * Clean up
2020-03-05 05:55:56 +00:00
"""Test the changes to option reflects accordingly."""
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
config_entry = await setup_unifi_integration(
UniFi - Add block network access control to config option (#32004) * Add block network access control to config option * Clean up
2020-03-05 05:55:56 +00:00
hass,
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
aioclient_mock,
UniFi - Add block network access control to config option (#32004) * Add block network access control to config option * Clean up
2020-03-05 05:55:56 +00:00
options={CONF_BLOCK_CLIENT: [BLOCKED["mac"]]},
clients_all_response=[BLOCKED, UNBLOCKED],
UniFi - Improve controller tests (#27261) * Improve controller tests and harmonize setup_unifi_integration to one * Store listeners to dispatchers to be used during reset
2019-10-07 19:55:35 +00:00
)
UniFi - Store controller in config_entry.entry_id (#34553) * Store controller in config_entry.entry_id * Clean up imports
2020-04-23 14:48:24 +00:00
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 1
UniFi - Improve controller tests (#27261) * Improve controller tests and harmonize setup_unifi_integration to one * Store listeners to dispatchers to be used during reset
2019-10-07 19:55:35 +00:00
UniFi - Add block network access control to config option (#32004) * Add block network access control to config option * Clean up
2020-03-05 05:55:56 +00:00
# Add a second switch
hass.config_entries.async_update_entry(
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
config_entry,
UniFi - Add block network access control to config option (#32004) * Add block network access control to config option * Clean up
2020-03-05 05:55:56 +00:00
options={CONF_BLOCK_CLIENT: [BLOCKED["mac"], UNBLOCKED["mac"]]},
UniFi - Improve controller tests (#27261) * Improve controller tests and harmonize setup_unifi_integration to one * Store listeners to dispatchers to be used during reset
2019-10-07 19:55:35 +00:00
)
UniFi - Add block network access control to config option (#32004) * Add block network access control to config option * Clean up
2020-03-05 05:55:56 +00:00
await hass.async_block_till_done()
UniFi - Move some preloading of unavailable clients earlier in setup phase (#34599) Improve readability of setting up switch platform
2020-04-23 19:29:38 +00:00
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 1
UniFi - Add block network access control to config option (#32004) * Add block network access control to config option * Clean up
2020-03-05 05:55:56 +00:00
# Remove the second switch again
hass.config_entries.async_update_entry(
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
config_entry,
Upgrade black to 20.8b1 (#39287)
2020-08-27 11:56:20 +00:00
options={CONF_BLOCK_CLIENT: [BLOCKED["mac"]]},
UniFi - Add block network access control to config option (#32004) * Add block network access control to config option * Clean up
2020-03-05 05:55:56 +00:00
)
await hass.async_block_till_done()
UniFi - Store controller in config_entry.entry_id (#34553) * Store controller in config_entry.entry_id * Clean up imports
2020-04-23 14:48:24 +00:00
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 1
UniFi - Add block network access control to config option (#32004) * Add block network access control to config option * Clean up
2020-03-05 05:55:56 +00:00
# Enable one and remove another one
hass.config_entries.async_update_entry(
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
config_entry,
Upgrade black to 20.8b1 (#39287)
2020-08-27 11:56:20 +00:00
options={CONF_BLOCK_CLIENT: [UNBLOCKED["mac"]]},
UniFi - Add block network access control to config option (#32004) * Add block network access control to config option * Clean up
2020-03-05 05:55:56 +00:00
)
await hass.async_block_till_done()
UniFi - Move some preloading of unavailable clients earlier in setup phase (#34599) Improve readability of setting up switch platform
2020-04-23 19:29:38 +00:00
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 0
UniFi - Add block network access control to config option (#32004) * Add block network access control to config option * Clean up
2020-03-05 05:55:56 +00:00
# Remove one
hass.config_entries.async_update_entry(
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
config_entry,
Upgrade black to 20.8b1 (#39287)
2020-08-27 11:56:20 +00:00
options={CONF_BLOCK_CLIENT: []},
UniFi - Add block network access control to config option (#32004) * Add block network access control to config option * Clean up
2020-03-05 05:55:56 +00:00
)
await hass.async_block_till_done()
UniFi - Store controller in config_entry.entry_id (#34553) * Store controller in config_entry.entry_id * Clean up imports
2020-04-23 14:48:24 +00:00
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 0
UniFi - Improve controller tests (#27261) * Improve controller tests and harmonize setup_unifi_integration to one * Store listeners to dispatchers to be used during reset
2019-10-07 19:55:35 +00:00
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
async def test_option_remove_switches(hass, aioclient_mock):
Add DPI Restriction switch to UniFi integration (#42499) * initial implementation for controlling DPI restrictions * address PR review comments and add DataUpdateCoordinator * fix existing tests against new lib version * add tests for DPI switches * bump aiounifi * listen to events instead of polling * fix tests * remove useless test * bump aiounifi * rename device to UniFi Controller per PR feedback
2020-11-03 07:36:37 +00:00
"""Test removal of DPI switch when options updated."""
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
config_entry = await setup_unifi_integration(
Add DPI Restriction switch to UniFi integration (#42499) * initial implementation for controlling DPI restrictions * address PR review comments and add DataUpdateCoordinator * fix existing tests against new lib version * add tests for DPI switches * bump aiounifi * listen to events instead of polling * fix tests * remove useless test * bump aiounifi * rename device to UniFi Controller per PR feedback
2020-11-03 07:36:37 +00:00
hass,
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
aioclient_mock,
Add DPI Restriction switch to UniFi integration (#42499) * initial implementation for controlling DPI restrictions * address PR review comments and add DataUpdateCoordinator * fix existing tests against new lib version * add tests for DPI switches * bump aiounifi * listen to events instead of polling * fix tests * remove useless test * bump aiounifi * rename device to UniFi Controller per PR feedback
2020-11-03 07:36:37 +00:00
options={
CONF_TRACK_CLIENTS: False,
CONF_TRACK_DEVICES: False,
},
clients_response=[CLIENT_1],
dpigroup_response=DPI_GROUPS,
dpiapp_response=DPI_APPS,
)
Remove old UniFi POE client implementation (#81749) Remove all references to POE client implementation
2022-11-08 06:48:54 +00:00
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 1
Add DPI Restriction switch to UniFi integration (#42499) * initial implementation for controlling DPI restrictions * address PR review comments and add DataUpdateCoordinator * fix existing tests against new lib version * add tests for DPI switches * bump aiounifi * listen to events instead of polling * fix tests * remove useless test * bump aiounifi * rename device to UniFi Controller per PR feedback
2020-11-03 07:36:37 +00:00
# Disable DPI Switches
hass.config_entries.async_update_entry(
Improve UniFi tests (#45871)
2021-02-05 15:31:47 +00:00
config_entry,
Remove old UniFi POE client implementation (#81749) Remove all references to POE client implementation
2022-11-08 06:48:54 +00:00
options={CONF_DPI_RESTRICTIONS: False},
Add DPI Restriction switch to UniFi integration (#42499) * initial implementation for controlling DPI restrictions * address PR review comments and add DataUpdateCoordinator * fix existing tests against new lib version * add tests for DPI switches * bump aiounifi * listen to events instead of polling * fix tests * remove useless test * bump aiounifi * rename device to UniFi Controller per PR feedback
2020-11-03 07:36:37 +00:00
)
await hass.async_block_till_done()
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 0
Expose UniFi PoE ports as individual switches (#80566) * Add simple PoE control switches * Add basic tests * Complete testing * Dont use port.up as part of available * Bump aiounifi to v40
2022-10-19 17:54:40 +00:00
async def test_poe_port_switches(hass, aioclient_mock, mock_unifi_websocket):
"""Test the update_items function with some clients."""
config_entry = await setup_unifi_integration(
hass, aioclient_mock, devices_response=[DEVICE_1]
)
controller = hass.data[UNIFI_DOMAIN][config_entry.entry_id]
assert len(hass.states.async_entity_ids(SWITCH_DOMAIN)) == 0
ent_reg = er.async_get(hass)
ent_reg_entry = ent_reg.async_get("switch.mock_name_port_1_poe")
assert ent_reg_entry.disabled_by == RegistryEntryDisabler.INTEGRATION
assert ent_reg_entry.entity_category is EntityCategory.CONFIG
# Enable entity
ent_reg.async_update_entity(
entity_id="switch.mock_name_port_1_poe", disabled_by=None
)
await hass.async_block_till_done()
async_fire_time_changed(
hass,
dt.utcnow() + timedelta(seconds=RELOAD_AFTER_UPDATE_DELAY + 1),
)
await hass.async_block_till_done()
# Validate state object
switch_1 = hass.states.get("switch.mock_name_port_1_poe")
assert switch_1 is not None
assert switch_1.state == STATE_ON
assert switch_1.attributes.get(ATTR_DEVICE_CLASS) == SwitchDeviceClass.OUTLET
# Update state object
device_1 = deepcopy(DEVICE_1)
device_1["port_table"][0]["poe_mode"] = "off"
Simplify mocking UniFi websocket (#80652) * Simplify mocking UniFi websocket * Fix one more
2022-10-20 17:37:13 +00:00
mock_unifi_websocket(message=MessageKey.DEVICE, data=device_1)
Expose UniFi PoE ports as individual switches (#80566) * Add simple PoE control switches * Add basic tests * Complete testing * Dont use port.up as part of available * Bump aiounifi to v40
2022-10-19 17:54:40 +00:00
await hass.async_block_till_done()
assert hass.states.get("switch.mock_name_port_1_poe").state == STATE_OFF
# Turn off PoE
aioclient_mock.clear_requests()
aioclient_mock.put(
f"https://{controller.host}:1234/api/s/{controller.site}/rest/device/mock-id",
)
await hass.services.async_call(
SWITCH_DOMAIN,
"turn_off",
{"entity_id": "switch.mock_name_port_1_poe"},
blocking=True,
)
assert aioclient_mock.call_count == 1
assert aioclient_mock.mock_calls[0][2] == {
"port_overrides": [{"poe_mode": "off", "port_idx": 1, "portconf_id": "1a1"}]
}
# Turn on PoE
await hass.services.async_call(
SWITCH_DOMAIN,
"turn_on",
{"entity_id": "switch.mock_name_port_1_poe"},
blocking=True,
)
assert aioclient_mock.call_count == 2
assert aioclient_mock.mock_calls[1][2] == {
"port_overrides": [{"poe_mode": "auto", "port_idx": 1, "portconf_id": "1a1"}]
}
# Availability signalling
# Controller disconnects
mock_unifi_websocket(state=WebsocketState.DISCONNECTED)
await hass.async_block_till_done()
assert hass.states.get("switch.mock_name_port_1_poe").state == STATE_UNAVAILABLE
# Controller reconnects
mock_unifi_websocket(state=WebsocketState.RUNNING)
await hass.async_block_till_done()
assert hass.states.get("switch.mock_name_port_1_poe").state == STATE_OFF
# Device gets disabled
device_1["disabled"] = True
Simplify mocking UniFi websocket (#80652) * Simplify mocking UniFi websocket * Fix one more
2022-10-20 17:37:13 +00:00
mock_unifi_websocket(message=MessageKey.DEVICE, data=device_1)
Expose UniFi PoE ports as individual switches (#80566) * Add simple PoE control switches * Add basic tests * Complete testing * Dont use port.up as part of available * Bump aiounifi to v40
2022-10-19 17:54:40 +00:00
await hass.async_block_till_done()
assert hass.states.get("switch.mock_name_port_1_poe").state == STATE_UNAVAILABLE
# Device gets re-enabled
device_1["disabled"] = False
Simplify mocking UniFi websocket (#80652) * Simplify mocking UniFi websocket * Fix one more
2022-10-20 17:37:13 +00:00
mock_unifi_websocket(message=MessageKey.DEVICE, data=device_1)
Expose UniFi PoE ports as individual switches (#80566) * Add simple PoE control switches * Add basic tests * Complete testing * Dont use port.up as part of available * Bump aiounifi to v40
2022-10-19 17:54:40 +00:00
await hass.async_block_till_done()
assert hass.states.get("switch.mock_name_port_1_poe").state == STATE_OFF
Remove old UniFi POE client implementation (#81749) Remove all references to POE client implementation
2022-11-08 06:48:54 +00:00
async def test_remove_poe_client_switches(hass, aioclient_mock):
"""Test old PoE client switches are removed."""
config_entry = config_entries.ConfigEntry(
version=1,
domain=UNIFI_DOMAIN,
title="Mock Title",
data=ENTRY_CONFIG,
source="test",
options={},
entry_id="1",
)
ent_reg = er.async_get(hass)
ent_reg.async_get_or_create(
SWITCH_DOMAIN,
UNIFI_DOMAIN,
"poe-123",
config_entry=config_entry,
)
await setup_unifi_integration(hass, aioclient_mock)
assert not [
entry
for entry in ent_reg.entities.values()
if entry.config_entry_id == config_entry.entry_id
]