jeena/home-assistant
1
0
Fork 0
Code Issues Pull requests Projects Packages Wiki Activity Actions

Black

Browse source
This commit is contained in:
Paulus Schoutsen 2019-07-31 12:25:30 -07:00
parent da05dfe708
commit 4de97abc3a
2676 changed files with 163166 additions and 140084 deletions
Download patch file Download diff file Expand all files Collapse all files

324
tests/components/unifi/test_init.py
View file

@ -7,7 +7,12 @@ from homeassistant.components.unifi import config_flow
from homeassistant.setup import async_setup_component
from homeassistant.components.unifi.const import CONF_CONTROLLER, CONF_SITE_ID
from homeassistant.const import (
CONF_HOST, CONF_PASSWORD, CONF_PORT, CONF_USERNAME, CONF_VERIFY_SSL)
CONF_HOST,
CONF_PASSWORD,
CONF_PORT,
CONF_USERNAME,
CONF_VERIFY_SSL,
)
from tests.common import mock_coro, MockConfigEntry
@ -24,45 +29,51 @@ async def test_setup_with_config(hass):
config = {
unifi.DOMAIN: {
unifi.CONF_CONTROLLERS: {
unifi.CONF_HOST: '1.2.3.4',
unifi.CONF_SITE_ID: 'My site',
unifi.CONF_BLOCK_CLIENT: ['12:34:56:78:90:AB'],
unifi.CONF_HOST: "1.2.3.4",
unifi.CONF_SITE_ID: "My site",
unifi.CONF_BLOCK_CLIENT: ["12:34:56:78:90:AB"],
unifi.CONF_DETECTION_TIME: 3,
unifi.CONF_SSID_FILTER: ['ssid']
unifi.CONF_SSID_FILTER: ["ssid"],
}
}
}
assert await async_setup_component(hass, unifi.DOMAIN, config) is True
assert unifi.DOMAIN not in hass.data
assert hass.data[unifi.UNIFI_CONFIG] == [{
unifi.CONF_HOST: '1.2.3.4',
unifi.CONF_SITE_ID: 'My site',
unifi.CONF_BLOCK_CLIENT: ['12:34:56:78:90:AB'],
unifi.CONF_DETECTION_TIME: timedelta(seconds=3),
unifi.CONF_SSID_FILTER: ['ssid']
}]
assert hass.data[unifi.UNIFI_CONFIG] == [
{
unifi.CONF_HOST: "1.2.3.4",
unifi.CONF_SITE_ID: "My site",
unifi.CONF_BLOCK_CLIENT: ["12:34:56:78:90:AB"],
unifi.CONF_DETECTION_TIME: timedelta(seconds=3),
unifi.CONF_SSID_FILTER: ["ssid"],
}
]
async def test_successful_config_entry(hass):
"""Test that configured options for a host are loaded via config entry."""
entry = MockConfigEntry(domain=unifi.DOMAIN, data={
'controller': {
'host': '0.0.0.0',
'username': 'user',
'password': 'pass',
'port': 80,
'site': 'default',
'verify_ssl': True
entry = MockConfigEntry(
domain=unifi.DOMAIN,
data={
"controller": {
"host": "0.0.0.0",
"username": "user",
"password": "pass",
"port": 80,
"site": "default",
"verify_ssl": True,
},
"poe_control": True,
},
'poe_control': True
})
)
entry.add_to_hass(hass)
mock_registry = Mock()
with patch.object(unifi, 'UniFiController') as mock_controller, \
patch('homeassistant.helpers.device_registry.async_get_registry',
return_value=mock_coro(mock_registry)):
with patch.object(unifi, "UniFiController") as mock_controller, patch(
"homeassistant.helpers.device_registry.async_get_registry",
return_value=mock_coro(mock_registry),
):
mock_controller.return_value.async_setup.return_value = mock_coro(True)
mock_controller.return_value.mac = '00:11:22:33:44:55'
mock_controller.return_value.mac = "00:11:22:33:44:55"
assert await unifi.async_setup_entry(hass, entry) is True
assert len(mock_controller.mock_calls) == 2
@ -73,59 +84,62 @@ async def test_successful_config_entry(hass):
assert len(mock_registry.mock_calls) == 1
assert mock_registry.mock_calls[0][2] == {
'config_entry_id': entry.entry_id,
'connections': {
('mac', '00:11:22:33:44:55')
},
'manufacturer': unifi.ATTR_MANUFACTURER,
'model': "UniFi Controller",
'name': "UniFi Controller",
"config_entry_id": entry.entry_id,
"connections": {("mac", "00:11:22:33:44:55")},
"manufacturer": unifi.ATTR_MANUFACTURER,
"model": "UniFi Controller",
"name": "UniFi Controller",
}
async def test_controller_fail_setup(hass):
"""Test that a failed setup still stores controller."""
entry = MockConfigEntry(domain=unifi.DOMAIN, data={
'controller': {
'host': '0.0.0.0',
'username': 'user',
'password': 'pass',
'port': 80,
'site': 'default',
'verify_ssl': True
entry = MockConfigEntry(
domain=unifi.DOMAIN,
data={
"controller": {
"host": "0.0.0.0",
"username": "user",
"password": "pass",
"port": 80,
"site": "default",
"verify_ssl": True,
},
"poe_control": True,
},
'poe_control': True
})
)
entry.add_to_hass(hass)
with patch.object(unifi, 'UniFiController') as mock_cntrlr:
with patch.object(unifi, "UniFiController") as mock_cntrlr:
mock_cntrlr.return_value.async_setup.return_value = mock_coro(False)
assert await unifi.async_setup_entry(hass, entry) is False
controller_id = unifi.CONTROLLER_ID.format(
host='0.0.0.0', site='default'
)
controller_id = unifi.CONTROLLER_ID.format(host="0.0.0.0", site="default")
assert controller_id in hass.data[unifi.DOMAIN]
async def test_controller_no_mac(hass):
"""Test that configured options for a host are loaded via config entry."""
entry = MockConfigEntry(domain=unifi.DOMAIN, data={
'controller': {
'host': '0.0.0.0',
'username': 'user',
'password': 'pass',
'port': 80,
'site': 'default',
'verify_ssl': True
entry = MockConfigEntry(
domain=unifi.DOMAIN,
data={
"controller": {
"host": "0.0.0.0",
"username": "user",
"password": "pass",
"port": 80,
"site": "default",
"verify_ssl": True,
},
"poe_control": True,
},
'poe_control': True
})
)
entry.add_to_hass(hass)
mock_registry = Mock()
with patch.object(unifi, 'UniFiController') as mock_controller, \
patch('homeassistant.helpers.device_registry.async_get_registry',
return_value=mock_coro(mock_registry)):
with patch.object(unifi, "UniFiController") as mock_controller, patch(
"homeassistant.helpers.device_registry.async_get_registry",
return_value=mock_coro(mock_registry),
):
mock_controller.return_value.async_setup.return_value = mock_coro(True)
mock_controller.return_value.mac = None
assert await unifi.async_setup_entry(hass, entry) is True
@ -137,24 +151,28 @@ async def test_controller_no_mac(hass):
async def test_unload_entry(hass):
"""Test being able to unload an entry."""
entry = MockConfigEntry(domain=unifi.DOMAIN, data={
'controller': {
'host': '0.0.0.0',
'username': 'user',
'password': 'pass',
'port': 80,
'site': 'default',
'verify_ssl': True
entry = MockConfigEntry(
domain=unifi.DOMAIN,
data={
"controller": {
"host": "0.0.0.0",
"username": "user",
"password": "pass",
"port": 80,
"site": "default",
"verify_ssl": True,
},
"poe_control": True,
},
'poe_control': True
})
)
entry.add_to_hass(hass)
with patch.object(unifi, 'UniFiController') as mock_controller, \
patch('homeassistant.helpers.device_registry.async_get_registry',
return_value=mock_coro(Mock())):
with patch.object(unifi, "UniFiController") as mock_controller, patch(
"homeassistant.helpers.device_registry.async_get_registry",
return_value=mock_coro(Mock()),
):
mock_controller.return_value.async_setup.return_value = mock_coro(True)
mock_controller.return_value.mac = '00:11:22:33:44:55'
mock_controller.return_value.mac = "00:11:22:33:44:55"
assert await unifi.async_setup_entry(hass, entry) is True
assert len(mock_controller.return_value.mock_calls) == 1
@ -170,9 +188,11 @@ async def test_flow_works(hass, aioclient_mock):
flow = config_flow.UnifiFlowHandler()
flow.hass = hass
with patch('aiounifi.Controller') as mock_controller:
with patch("aiounifi.Controller") as mock_controller:
def mock_constructor(
host, username, password, port, site, websession, sslcontext):
host, username, password, port, site, websession, sslcontext
):
"""Fake the controller constructor."""
mock_controller.host = host
mock_controller.username = username
@ -183,34 +203,36 @@ async def test_flow_works(hass, aioclient_mock):
mock_controller.side_effect = mock_constructor
mock_controller.login.return_value = mock_coro()
mock_controller.sites.return_value = mock_coro({
'site1': {'name': 'default', 'role': 'admin', 'desc': 'site name'}
})
mock_controller.sites.return_value = mock_coro(
{"site1": {"name": "default", "role": "admin", "desc": "site name"}}
)
await flow.async_step_user(user_input={
CONF_HOST: '1.2.3.4',
CONF_USERNAME: 'username',
CONF_PASSWORD: 'password',
CONF_PORT: 1234,
CONF_VERIFY_SSL: True
})
await flow.async_step_user(
user_input={
CONF_HOST: "1.2.3.4",
CONF_USERNAME: "username",
CONF_PASSWORD: "password",
CONF_PORT: 1234,
CONF_VERIFY_SSL: True,
}
)
result = await flow.async_step_site(user_input={})
assert mock_controller.host == '1.2.3.4'
assert mock_controller.host == "1.2.3.4"
assert len(mock_controller.login.mock_calls) == 1
assert len(mock_controller.sites.mock_calls) == 1
assert result['type'] == 'create_entry'
assert result['title'] == 'site name'
assert result['data'] == {
assert result["type"] == "create_entry"
assert result["title"] == "site name"
assert result["data"] == {
CONF_CONTROLLER: {
CONF_HOST: '1.2.3.4',
CONF_USERNAME: 'username',
CONF_PASSWORD: 'password',
CONF_HOST: "1.2.3.4",
CONF_USERNAME: "username",
CONF_PASSWORD: "password",
CONF_PORT: 1234,
CONF_SITE_ID: 'default',
CONF_VERIFY_SSL: True
CONF_SITE_ID: "default",
CONF_VERIFY_SSL: True,
}
}
@ -221,26 +243,22 @@ async def test_controller_multiple_sites(hass):
flow.hass = hass
flow.config = {
CONF_HOST: '1.2.3.4',
CONF_USERNAME: 'username',
CONF_PASSWORD: 'password',
CONF_HOST: "1.2.3.4",
CONF_USERNAME: "username",
CONF_PASSWORD: "password",
}
flow.sites = {
'site1': {
'name': 'default', 'role': 'admin', 'desc': 'site name'
},
'site2': {
'name': 'site2', 'role': 'admin', 'desc': 'site2 name'
}
"site1": {"name": "default", "role": "admin", "desc": "site name"},
"site2": {"name": "site2", "role": "admin", "desc": "site2 name"},
}
result = await flow.async_step_site()
assert result['type'] == 'form'
assert result['step_id'] == 'site'
assert result["type"] == "form"
assert result["step_id"] == "site"
assert result['data_schema']({'site': 'site name'})
assert result['data_schema']({'site': 'site2 name'})
assert result["data_schema"]({"site": "site name"})
assert result["data_schema"]({"site": "site2 name"})
async def test_controller_site_already_configured(hass):
@ -248,29 +266,22 @@ async def test_controller_site_already_configured(hass):
flow = config_flow.UnifiFlowHandler()
flow.hass = hass
entry = MockConfigEntry(domain=unifi.DOMAIN, data={
'controller': {
'host': '1.2.3.4',
'site': 'default',
}
})
entry = MockConfigEntry(
domain=unifi.DOMAIN, data={"controller": {"host": "1.2.3.4", "site": "default"}}
)
entry.add_to_hass(hass)
flow.config = {
CONF_HOST: '1.2.3.4',
CONF_USERNAME: 'username',
CONF_PASSWORD: 'password',
}
flow.desc = 'site name'
flow.sites = {
'site1': {
'name': 'default', 'role': 'admin', 'desc': 'site name'
}
CONF_HOST: "1.2.3.4",
CONF_USERNAME: "username",
CONF_PASSWORD: "password",
}
flow.desc = "site name"
flow.sites = {"site1": {"name": "default", "role": "admin", "desc": "site name"}}
result = await flow.async_step_site()
assert result['type'] == 'abort'
assert result["type"] == "abort"
async def test_user_credentials_faulty(hass, aioclient_mock):
@ -278,17 +289,20 @@ async def test_user_credentials_faulty(hass, aioclient_mock):
flow = config_flow.UnifiFlowHandler()
flow.hass = hass
with patch.object(config_flow, 'get_controller',
side_effect=unifi.errors.AuthenticationRequired):
result = await flow.async_step_user({
CONF_HOST: '1.2.3.4',
CONF_USERNAME: 'username',
CONF_PASSWORD: 'password',
CONF_SITE_ID: 'default',
})
with patch.object(
config_flow, "get_controller", side_effect=unifi.errors.AuthenticationRequired
):
result = await flow.async_step_user(
{
CONF_HOST: "1.2.3.4",
CONF_USERNAME: "username",
CONF_PASSWORD: "password",
CONF_SITE_ID: "default",
}
)
assert result['type'] == 'form'
assert result['errors'] == {'base': 'faulty_credentials'}
assert result["type"] == "form"
assert result["errors"] == {"base": "faulty_credentials"}
async def test_controller_is_unavailable(hass, aioclient_mock):
@ -296,17 +310,20 @@ async def test_controller_is_unavailable(hass, aioclient_mock):
flow = config_flow.UnifiFlowHandler()
flow.hass = hass
with patch.object(config_flow, 'get_controller',
side_effect=unifi.errors.CannotConnect):
result = await flow.async_step_user({
CONF_HOST: '1.2.3.4',
CONF_USERNAME: 'username',
CONF_PASSWORD: 'password',
CONF_SITE_ID: 'default',
})
with patch.object(
config_flow, "get_controller", side_effect=unifi.errors.CannotConnect
):
result = await flow.async_step_user(
{
CONF_HOST: "1.2.3.4",
CONF_USERNAME: "username",
CONF_PASSWORD: "password",
CONF_SITE_ID: "default",
}
)
assert result['type'] == 'form'
assert result['errors'] == {'base': 'service_unavailable'}
assert result["type"] == "form"
assert result["errors"] == {"base": "service_unavailable"}
async def test_controller_unkown_problem(hass, aioclient_mock):
@ -314,13 +331,14 @@ async def test_controller_unkown_problem(hass, aioclient_mock):
flow = config_flow.UnifiFlowHandler()
flow.hass = hass
with patch.object(config_flow, 'get_controller',
side_effect=Exception):
result = await flow.async_step_user({
CONF_HOST: '1.2.3.4',
CONF_USERNAME: 'username',
CONF_PASSWORD: 'password',
CONF_SITE_ID: 'default',
})
with patch.object(config_flow, "get_controller", side_effect=Exception):
result = await flow.async_step_user(
{
CONF_HOST: "1.2.3.4",
CONF_USERNAME: "username",
CONF_PASSWORD: "password",
CONF_SITE_ID: "default",
}
)
assert result['type'] == 'abort'
assert result["type"] == "abort"