Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 21:06:19 +00:00
Code Activity

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

303
tests/components/axis/test_config_flow.py
View File

@@ -13,8 +13,9 @@ async def test_configured_devices(hass):
assert not result
entry = MockConfigEntry(domain=axis.DOMAIN,
data={axis.config_flow.CONF_MAC: '1234'})
entry = MockConfigEntry(
domain=axis.DOMAIN, data={axis.config_flow.CONF_MAC: "1234"}
)
entry.add_to_hass(hass)
result = config_flow.configured_devices(hass)
@@ -24,9 +25,9 @@ async def test_configured_devices(hass):
async def test_flow_works(hass):
"""Test that config flow works."""
with patch('axis.AxisDevice') as mock_device:
def mock_constructor(
loop, host, username, password, port, web_proto):
with patch("axis.AxisDevice") as mock_device:
def mock_constructor(loop, host, username, password, port, web_proto):
"""Fake the controller constructor."""
mock_device.loop = loop
mock_device.host = host
@@ -36,39 +37,38 @@ async def test_flow_works(hass):
return mock_device
mock_device.side_effect = mock_constructor
mock_device.vapix.params.system_serialnumber = 'serialnumber'
mock_device.vapix.params.prodnbr = 'prodnbr'
mock_device.vapix.params.system_serialnumber = "serialnumber"
mock_device.vapix.params.prodnbr = "prodnbr"
result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN,
context={'source': 'user'}
config_flow.DOMAIN, context={"source": "user"}
)
assert result['type'] == 'form'
assert result['step_id'] == 'user'
assert result["type"] == "form"
assert result["step_id"] == "user"
result = await hass.config_entries.flow.async_configure(
result['flow_id'],
result["flow_id"],
user_input={
config_flow.CONF_HOST: '1.2.3.4',
config_flow.CONF_USERNAME: 'user',
config_flow.CONF_PASSWORD: 'pass',
config_flow.CONF_PORT: 80
}
config_flow.CONF_HOST: "1.2.3.4",
config_flow.CONF_USERNAME: "user",
config_flow.CONF_PASSWORD: "pass",
config_flow.CONF_PORT: 80,
},
)
assert result['type'] == 'create_entry'
assert result['title'] == '{} - {}'.format('prodnbr', 'serialnumber')
assert result['data'] == {
assert result["type"] == "create_entry"
assert result["title"] == "{} - {}".format("prodnbr", "serialnumber")
assert result["data"] == {
axis.CONF_DEVICE: {
config_flow.CONF_HOST: '1.2.3.4',
config_flow.CONF_USERNAME: 'user',
config_flow.CONF_PASSWORD: 'pass',
config_flow.CONF_PORT: 80
config_flow.CONF_HOST: "1.2.3.4",
config_flow.CONF_USERNAME: "user",
config_flow.CONF_PASSWORD: "pass",
config_flow.CONF_PORT: 80,
},
config_flow.CONF_MAC: 'serialnumber',
config_flow.CONF_MODEL: 'prodnbr',
config_flow.CONF_NAME: 'prodnbr 0'
config_flow.CONF_MAC: "serialnumber",
config_flow.CONF_MODEL: "prodnbr",
config_flow.CONF_NAME: "prodnbr 0",
}
@@ -77,23 +77,28 @@ async def test_flow_fails_already_configured(hass):
flow = config_flow.AxisFlowHandler()
flow.hass = hass
entry = MockConfigEntry(domain=axis.DOMAIN,
data={axis.config_flow.CONF_MAC: '1234'})
entry = MockConfigEntry(
domain=axis.DOMAIN, data={axis.config_flow.CONF_MAC: "1234"}
)
entry.add_to_hass(hass)
mock_device = Mock()
mock_device.vapix.params.system_serialnumber = '1234'
mock_device.vapix.params.system_serialnumber = "1234"
with patch('homeassistant.components.axis.config_flow.get_device',
return_value=mock_coro(mock_device)):
result = await flow.async_step_user(user_input={
config_flow.CONF_HOST: '1.2.3.4',
config_flow.CONF_USERNAME: 'user',
config_flow.CONF_PASSWORD: 'pass',
config_flow.CONF_PORT: 80
})
with patch(
"homeassistant.components.axis.config_flow.get_device",
return_value=mock_coro(mock_device),
):
result = await flow.async_step_user(
user_input={
config_flow.CONF_HOST: "1.2.3.4",
config_flow.CONF_USERNAME: "user",
config_flow.CONF_PASSWORD: "pass",
config_flow.CONF_PORT: 80,
}
)
assert result['errors'] == {'base': 'already_configured'}
assert result["errors"] == {"base": "already_configured"}
async def test_flow_fails_faulty_credentials(hass):
@@ -101,16 +106,20 @@ async def test_flow_fails_faulty_credentials(hass):
flow = config_flow.AxisFlowHandler()
flow.hass = hass
with patch('homeassistant.components.axis.config_flow.get_device',
side_effect=config_flow.AuthenticationRequired):
result = await flow.async_step_user(user_input={
config_flow.CONF_HOST: '1.2.3.4',
config_flow.CONF_USERNAME: 'user',
config_flow.CONF_PASSWORD: 'pass',
config_flow.CONF_PORT: 80
})
with patch(
"homeassistant.components.axis.config_flow.get_device",
side_effect=config_flow.AuthenticationRequired,
):
result = await flow.async_step_user(
user_input={
config_flow.CONF_HOST: "1.2.3.4",
config_flow.CONF_USERNAME: "user",
config_flow.CONF_PASSWORD: "pass",
config_flow.CONF_PORT: 80,
}
)
assert result['errors'] == {'base': 'faulty_credentials'}
assert result["errors"] == {"base": "faulty_credentials"}
async def test_flow_fails_device_unavailable(hass):
@@ -118,64 +127,70 @@ async def test_flow_fails_device_unavailable(hass):
flow = config_flow.AxisFlowHandler()
flow.hass = hass
with patch('homeassistant.components.axis.config_flow.get_device',
side_effect=config_flow.CannotConnect):
result = await flow.async_step_user(user_input={
config_flow.CONF_HOST: '1.2.3.4',
config_flow.CONF_USERNAME: 'user',
config_flow.CONF_PASSWORD: 'pass',
config_flow.CONF_PORT: 80
})
with patch(
"homeassistant.components.axis.config_flow.get_device",
side_effect=config_flow.CannotConnect,
):
result = await flow.async_step_user(
user_input={
config_flow.CONF_HOST: "1.2.3.4",
config_flow.CONF_USERNAME: "user",
config_flow.CONF_PASSWORD: "pass",
config_flow.CONF_PORT: 80,
}
)
assert result['errors'] == {'base': 'device_unavailable'}
assert result["errors"] == {"base": "device_unavailable"}
async def test_flow_create_entry(hass):
"""Test that create entry can generate a name without other entries."""
flow = config_flow.AxisFlowHandler()
flow.hass = hass
flow.model = 'model'
flow.model = "model"
result = await flow._create_entry()
assert result['data'][config_flow.CONF_NAME] == 'model 0'
assert result["data"][config_flow.CONF_NAME] == "model 0"
async def test_flow_create_entry_more_entries(hass):
"""Test that create entry can generate a name with other entries."""
entry = MockConfigEntry(
domain=axis.DOMAIN, data={config_flow.CONF_NAME: 'model 0',
config_flow.CONF_MODEL: 'model'})
domain=axis.DOMAIN,
data={config_flow.CONF_NAME: "model 0", config_flow.CONF_MODEL: "model"},
)
entry.add_to_hass(hass)
entry2 = MockConfigEntry(
domain=axis.DOMAIN, data={config_flow.CONF_NAME: 'model 1',
config_flow.CONF_MODEL: 'model'})
domain=axis.DOMAIN,
data={config_flow.CONF_NAME: "model 1", config_flow.CONF_MODEL: "model"},
)
entry2.add_to_hass(hass)
flow = config_flow.AxisFlowHandler()
flow.hass = hass
flow.model = 'model'
flow.model = "model"
result = await flow._create_entry()
assert result['data'][config_flow.CONF_NAME] == 'model 2'
assert result["data"][config_flow.CONF_NAME] == "model 2"
async def test_zeroconf_flow(hass):
"""Test that zeroconf discovery for new devices work."""
with patch.object(axis, 'get_device', return_value=mock_coro(Mock())):
with patch.object(axis, "get_device", return_value=mock_coro(Mock())):
result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN,
data={
config_flow.CONF_HOST: '1.2.3.4',
config_flow.CONF_HOST: "1.2.3.4",
config_flow.CONF_PORT: 80,
'properties': {'macaddress': '00408C12345'}
"properties": {"macaddress": "00408C12345"},
},
context={'source': 'zeroconf'}
context={"source": "zeroconf"},
)
assert result['type'] == 'form'
assert result['step_id'] == 'user'
assert result["type"] == "form"
assert result["step_id"] == "user"
async def test_zeroconf_flow_known_device(hass):
@@ -183,15 +198,19 @@ async def test_zeroconf_flow_known_device(hass):
This is legacy support from devices registered with configurator.
"""
with patch('homeassistant.components.axis.config_flow.load_json',
return_value={'00408C12345': {
config_flow.CONF_HOST: '2.3.4.5',
config_flow.CONF_USERNAME: 'user',
config_flow.CONF_PASSWORD: 'pass',
config_flow.CONF_PORT: 80}}), \
patch('axis.AxisDevice') as mock_device:
def mock_constructor(
loop, host, username, password, port, web_proto):
with patch(
"homeassistant.components.axis.config_flow.load_json",
return_value={
"00408C12345": {
config_flow.CONF_HOST: "2.3.4.5",
config_flow.CONF_USERNAME: "user",
config_flow.CONF_PASSWORD: "pass",
config_flow.CONF_PORT: 80,
}
},
), patch("axis.AxisDevice") as mock_device:
def mock_constructor(loop, host, username, password, port, web_proto):
"""Fake the controller constructor."""
mock_device.loop = loop
mock_device.host = host
@@ -205,41 +224,43 @@ async def test_zeroconf_flow_known_device(hass):
result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN,
data={
config_flow.CONF_HOST: '1.2.3.4',
config_flow.CONF_HOST: "1.2.3.4",
config_flow.CONF_PORT: 80,
'hostname': 'name',
'properties': {'macaddress': '00408C12345'}
"hostname": "name",
"properties": {"macaddress": "00408C12345"},
},
context={'source': 'zeroconf'}
context={"source": "zeroconf"},
)
assert result['type'] == 'create_entry'
assert result["type"] == "create_entry"
async def test_zeroconf_flow_already_configured(hass):
"""Test that zeroconf doesn't setup already configured devices."""
entry = MockConfigEntry(
domain=axis.DOMAIN,
data={axis.CONF_DEVICE: {axis.config_flow.CONF_HOST: '1.2.3.4'},
axis.config_flow.CONF_MAC: '00408C12345'}
data={
axis.CONF_DEVICE: {axis.config_flow.CONF_HOST: "1.2.3.4"},
axis.config_flow.CONF_MAC: "00408C12345",
},
)
entry.add_to_hass(hass)
result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN,
data={
config_flow.CONF_HOST: '1.2.3.4',
config_flow.CONF_USERNAME: 'user',
config_flow.CONF_PASSWORD: 'pass',
config_flow.CONF_HOST: "1.2.3.4",
config_flow.CONF_USERNAME: "user",
config_flow.CONF_PASSWORD: "pass",
config_flow.CONF_PORT: 80,
'hostname': 'name',
'properties': {'macaddress': '00408C12345'}
"hostname": "name",
"properties": {"macaddress": "00408C12345"},
},
context={'source': 'zeroconf'}
context={"source": "zeroconf"},
)
assert result['type'] == 'abort'
assert result['reason'] == 'already_configured'
assert result["type"] == "abort"
assert result["reason"] == "already_configured"
async def test_zeroconf_flow_ignore_non_axis_device(hass):
@@ -247,14 +268,14 @@ async def test_zeroconf_flow_ignore_non_axis_device(hass):
result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN,
data={
config_flow.CONF_HOST: '169.254.3.4',
'properties': {'macaddress': '01234567890'}
config_flow.CONF_HOST: "169.254.3.4",
"properties": {"macaddress": "01234567890"},
},
context={'source': 'zeroconf'}
context={"source": "zeroconf"},
)
assert result['type'] == 'abort'
assert result['reason'] == 'not_axis_device'
assert result["type"] == "abort"
assert result["reason"] == "not_axis_device"
async def test_zeroconf_flow_ignore_link_local_address(hass):
@@ -262,44 +283,50 @@ async def test_zeroconf_flow_ignore_link_local_address(hass):
result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN,
data={
config_flow.CONF_HOST: '169.254.3.4',
'properties': {'macaddress': '00408C12345'}
config_flow.CONF_HOST: "169.254.3.4",
"properties": {"macaddress": "00408C12345"},
},
context={'source': 'zeroconf'}
context={"source": "zeroconf"},
)
assert result['type'] == 'abort'
assert result['reason'] == 'link_local_address'
assert result["type"] == "abort"
assert result["reason"] == "link_local_address"
async def test_zeroconf_flow_bad_config_file(hass):
"""Test that zeroconf discovery with bad config files abort."""
with patch('homeassistant.components.axis.config_flow.load_json',
return_value={'00408C12345': {
config_flow.CONF_HOST: '2.3.4.5',
config_flow.CONF_USERNAME: 'user',
config_flow.CONF_PASSWORD: 'pass',
config_flow.CONF_PORT: 80}}), \
patch('homeassistant.components.axis.config_flow.DEVICE_SCHEMA',
side_effect=config_flow.vol.Invalid('')):
with patch(
"homeassistant.components.axis.config_flow.load_json",
return_value={
"00408C12345": {
config_flow.CONF_HOST: "2.3.4.5",
config_flow.CONF_USERNAME: "user",
config_flow.CONF_PASSWORD: "pass",
config_flow.CONF_PORT: 80,
}
},
), patch(
"homeassistant.components.axis.config_flow.DEVICE_SCHEMA",
side_effect=config_flow.vol.Invalid(""),
):
result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN,
data={
config_flow.CONF_HOST: '1.2.3.4',
'properties': {'macaddress': '00408C12345'}
config_flow.CONF_HOST: "1.2.3.4",
"properties": {"macaddress": "00408C12345"},
},
context={'source': 'zeroconf'}
context={"source": "zeroconf"},
)
assert result['type'] == 'abort'
assert result['reason'] == 'bad_config_file'
assert result["type"] == "abort"
assert result["reason"] == "bad_config_file"
async def test_import_flow_works(hass):
"""Test that import flow works."""
with patch('axis.AxisDevice') as mock_device:
def mock_constructor(
loop, host, username, password, port, web_proto):
with patch("axis.AxisDevice") as mock_device:
def mock_constructor(loop, host, username, password, port, web_proto):
"""Fake the controller constructor."""
mock_device.loop = loop
mock_device.host = host
@@ -309,31 +336,31 @@ async def test_import_flow_works(hass):
return mock_device
mock_device.side_effect = mock_constructor
mock_device.vapix.params.system_serialnumber = 'serialnumber'
mock_device.vapix.params.prodnbr = 'prodnbr'
mock_device.vapix.params.system_serialnumber = "serialnumber"
mock_device.vapix.params.prodnbr = "prodnbr"
result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN,
data={
config_flow.CONF_HOST: '1.2.3.4',
config_flow.CONF_USERNAME: 'user',
config_flow.CONF_PASSWORD: 'pass',
config_flow.CONF_HOST: "1.2.3.4",
config_flow.CONF_USERNAME: "user",
config_flow.CONF_PASSWORD: "pass",
config_flow.CONF_PORT: 80,
config_flow.CONF_NAME: 'name'
config_flow.CONF_NAME: "name",
},
context={'source': 'import'}
context={"source": "import"},
)
assert result['type'] == 'create_entry'
assert result['title'] == '{} - {}'.format('prodnbr', 'serialnumber')
assert result['data'] == {
assert result["type"] == "create_entry"
assert result["title"] == "{} - {}".format("prodnbr", "serialnumber")
assert result["data"] == {
axis.CONF_DEVICE: {
config_flow.CONF_HOST: '1.2.3.4',
config_flow.CONF_USERNAME: 'user',
config_flow.CONF_PASSWORD: 'pass',
config_flow.CONF_PORT: 80
config_flow.CONF_HOST: "1.2.3.4",
config_flow.CONF_USERNAME: "user",
config_flow.CONF_PASSWORD: "pass",
config_flow.CONF_PORT: 80,
},
config_flow.CONF_MAC: 'serialnumber',
config_flow.CONF_MODEL: 'prodnbr',
config_flow.CONF_NAME: 'name'
config_flow.CONF_MAC: "serialnumber",
config_flow.CONF_MODEL: "prodnbr",
config_flow.CONF_NAME: "name",
}