Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-27 14:31:13 +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

338
tests/components/deconz/test_config_flow.py
View File

@@ -12,93 +12,103 @@ import pydeconz
async def test_flow_works(hass, aioclient_mock):
"""Test that config flow works."""
aioclient_mock.get(pydeconz.utils.URL_DISCOVER, json=[
{'id': 'id', 'internalipaddress': '1.2.3.4', 'internalport': 80}
], headers={'content-type': 'application/json'})
aioclient_mock.post('http://1.2.3.4:80/api', json=[
{"success": {"username": "1234567890ABCDEF"}}
], headers={'content-type': 'application/json'})
aioclient_mock.get(
pydeconz.utils.URL_DISCOVER,
json=[{"id": "id", "internalipaddress": "1.2.3.4", "internalport": 80}],
headers={"content-type": "application/json"},
)
aioclient_mock.post(
"http://1.2.3.4:80/api",
json=[{"success": {"username": "1234567890ABCDEF"}}],
headers={"content-type": "application/json"},
)
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'] == 'link'
assert result["type"] == "form"
assert result["step_id"] == "link"
result = await hass.config_entries.flow.async_configure(
result['flow_id'],
user_input={}
result["flow_id"], user_input={}
)
assert result['type'] == 'create_entry'
assert result['title'] == 'deCONZ-id'
assert result['data'] == {
config_flow.CONF_BRIDGEID: 'id',
config_flow.CONF_HOST: '1.2.3.4',
assert result["type"] == "create_entry"
assert result["title"] == "deCONZ-id"
assert result["data"] == {
config_flow.CONF_BRIDGEID: "id",
config_flow.CONF_HOST: "1.2.3.4",
config_flow.CONF_PORT: 80,
config_flow.CONF_API_KEY: '1234567890ABCDEF'
config_flow.CONF_API_KEY: "1234567890ABCDEF",
}
async def test_user_step_bridge_discovery_fails(hass, aioclient_mock):
"""Test config flow works when discovery fails."""
with patch('homeassistant.components.deconz.config_flow.async_discovery',
side_effect=asyncio.TimeoutError):
with patch(
"homeassistant.components.deconz.config_flow.async_discovery",
side_effect=asyncio.TimeoutError,
):
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'] == 'init'
assert result["type"] == "form"
assert result["step_id"] == "init"
async def test_user_step_no_discovered_bridges(hass, aioclient_mock):
"""Test config flow discovers no bridges."""
aioclient_mock.get(pydeconz.utils.URL_DISCOVER, json=[],
headers={'content-type': 'application/json'})
result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN,
context={'source': 'user'}
aioclient_mock.get(
pydeconz.utils.URL_DISCOVER,
json=[],
headers={"content-type": "application/json"},
)
assert result['type'] == 'form'
assert result['step_id'] == 'init'
result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN, context={"source": "user"}
)
assert result["type"] == "form"
assert result["step_id"] == "init"
async def test_user_step_one_bridge_discovered(hass, aioclient_mock):
"""Test config flow discovers one bridge."""
aioclient_mock.get(pydeconz.utils.URL_DISCOVER, json=[
{'id': 'id', 'internalipaddress': '1.2.3.4', 'internalport': 80}
], headers={'content-type': 'application/json'})
aioclient_mock.get(
pydeconz.utils.URL_DISCOVER,
json=[{"id": "id", "internalipaddress": "1.2.3.4", "internalport": 80}],
headers={"content-type": "application/json"},
)
flow = config_flow.DeconzFlowHandler()
flow.hass = hass
result = await flow.async_step_user()
assert result['type'] == 'form'
assert result['step_id'] == 'link'
assert flow.deconz_config[config_flow.CONF_HOST] == '1.2.3.4'
assert result["type"] == "form"
assert result["step_id"] == "link"
assert flow.deconz_config[config_flow.CONF_HOST] == "1.2.3.4"
async def test_user_step_two_bridges_discovered(hass, aioclient_mock):
"""Test config flow discovers two bridges."""
aioclient_mock.get(pydeconz.utils.URL_DISCOVER, json=[
{'id': 'id1', 'internalipaddress': '1.2.3.4', 'internalport': 80},
{'id': 'id2', 'internalipaddress': '5.6.7.8', 'internalport': 80}
], headers={'content-type': 'application/json'})
result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN,
context={'source': 'user'}
aioclient_mock.get(
pydeconz.utils.URL_DISCOVER,
json=[
{"id": "id1", "internalipaddress": "1.2.3.4", "internalport": 80},
{"id": "id2", "internalipaddress": "5.6.7.8", "internalport": 80},
],
headers={"content-type": "application/json"},
)
assert result['data_schema']({config_flow.CONF_HOST: '1.2.3.4'})
assert result['data_schema']({config_flow.CONF_HOST: '5.6.7.8'})
result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN, context={"source": "user"}
)
assert result["data_schema"]({config_flow.CONF_HOST: "1.2.3.4"})
assert result["data_schema"]({config_flow.CONF_HOST: "5.6.7.8"})
async def test_user_step_two_bridges_selection(hass, aioclient_mock):
@@ -107,66 +117,62 @@ async def test_user_step_two_bridges_selection(hass, aioclient_mock):
flow.hass = hass
flow.bridges = [
{
config_flow.CONF_BRIDGEID: 'id1',
config_flow.CONF_HOST: '1.2.3.4',
config_flow.CONF_PORT: 80
config_flow.CONF_BRIDGEID: "id1",
config_flow.CONF_HOST: "1.2.3.4",
config_flow.CONF_PORT: 80,
},
{
config_flow.CONF_BRIDGEID: 'id2',
config_flow.CONF_HOST: '5.6.7.8',
config_flow.CONF_PORT: 80
}
config_flow.CONF_BRIDGEID: "id2",
config_flow.CONF_HOST: "5.6.7.8",
config_flow.CONF_PORT: 80,
},
]
result = await flow.async_step_user(
user_input={config_flow.CONF_HOST: '1.2.3.4'})
assert result['type'] == 'form'
assert result['step_id'] == 'link'
assert flow.deconz_config[config_flow.CONF_HOST] == '1.2.3.4'
result = await flow.async_step_user(user_input={config_flow.CONF_HOST: "1.2.3.4"})
assert result["type"] == "form"
assert result["step_id"] == "link"
assert flow.deconz_config[config_flow.CONF_HOST] == "1.2.3.4"
async def test_user_step_manual_configuration(hass, aioclient_mock):
"""Test config flow with manual input."""
aioclient_mock.get(pydeconz.utils.URL_DISCOVER, json=[],
headers={'content-type': 'application/json'})
aioclient_mock.get(
pydeconz.utils.URL_DISCOVER,
json=[],
headers={"content-type": "application/json"},
)
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'] == 'init'
assert result["type"] == "form"
assert result["step_id"] == "init"
result = await hass.config_entries.flow.async_configure(
result['flow_id'],
user_input={
config_flow.CONF_HOST: '1.2.3.4',
config_flow.CONF_PORT: 80
}
result["flow_id"],
user_input={config_flow.CONF_HOST: "1.2.3.4", config_flow.CONF_PORT: 80},
)
assert result['type'] == 'form'
assert result['step_id'] == 'link'
assert result["type"] == "form"
assert result["step_id"] == "link"
async def test_link_no_api_key(hass):
"""Test config flow should abort if no API key was possible to retrieve."""
flow = config_flow.DeconzFlowHandler()
flow.hass = hass
flow.deconz_config = {
config_flow.CONF_HOST: '1.2.3.4',
config_flow.CONF_PORT: 80
}
flow.deconz_config = {config_flow.CONF_HOST: "1.2.3.4", config_flow.CONF_PORT: 80}
with patch(
'homeassistant.components.deconz.config_flow.async_get_api_key',
side_effect=pydeconz.errors.ResponseError):
"homeassistant.components.deconz.config_flow.async_get_api_key",
side_effect=pydeconz.errors.ResponseError,
):
result = await flow.async_step_link(user_input={})
assert result['type'] == 'form'
assert result['step_id'] == 'link'
assert result['errors'] == {'base': 'no_key'}
assert result["type"] == "form"
assert result["step_id"] == "link"
assert result["errors"] == {"base": "no_key"}
async def test_bridge_ssdp_discovery(hass):
@@ -174,73 +180,70 @@ async def test_bridge_ssdp_discovery(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,
ATTR_SERIAL: 'id',
ATTR_MANUFACTURERURL:
config_flow.DECONZ_MANUFACTURERURL,
config_flow.ATTR_UUID: 'uuid:1234'
ATTR_SERIAL: "id",
ATTR_MANUFACTURERURL: config_flow.DECONZ_MANUFACTURERURL,
config_flow.ATTR_UUID: "uuid:1234",
},
context={'source': 'ssdp'}
context={"source": "ssdp"},
)
assert result['type'] == 'form'
assert result['step_id'] == 'link'
assert result["type"] == "form"
assert result["step_id"] == "link"
async def test_bridge_ssdp_discovery_not_deconz_bridge(hass):
"""Test a non deconz bridge being discovered over ssdp."""
result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN,
data={
ATTR_MANUFACTURERURL: 'not deconz bridge'
},
context={'source': 'ssdp'}
data={ATTR_MANUFACTURERURL: "not deconz bridge"},
context={"source": "ssdp"},
)
assert result['type'] == 'abort'
assert result['reason'] == 'not_deconz_bridge'
assert result["type"] == "abort"
assert result["reason"] == "not_deconz_bridge"
async def test_bridge_discovery_update_existing_entry(hass):
"""Test if a discovered bridge has already been configured."""
entry = MockConfigEntry(domain=config_flow.DOMAIN, data={
config_flow.CONF_HOST: '1.2.3.4', config_flow.CONF_BRIDGEID: 'id'
})
entry = MockConfigEntry(
domain=config_flow.DOMAIN,
data={config_flow.CONF_HOST: "1.2.3.4", config_flow.CONF_BRIDGEID: "id"},
)
entry.add_to_hass(hass)
gateway = Mock()
gateway.config_entry = entry
gateway.api.config.uuid = '1234'
hass.data[config_flow.DOMAIN] = {'id': gateway}
gateway.api.config.uuid = "1234"
hass.data[config_flow.DOMAIN] = {"id": gateway}
result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN,
data={
config_flow.CONF_HOST: 'mock-deconz',
ATTR_SERIAL: 'id',
ATTR_MANUFACTURERURL:
config_flow.DECONZ_MANUFACTURERURL,
config_flow.ATTR_UUID: 'uuid:1234'
config_flow.CONF_HOST: "mock-deconz",
ATTR_SERIAL: "id",
ATTR_MANUFACTURERURL: config_flow.DECONZ_MANUFACTURERURL,
config_flow.ATTR_UUID: "uuid:1234",
},
context={'source': 'ssdp'}
context={"source": "ssdp"},
)
assert result['type'] == 'abort'
assert result['reason'] == 'updated_instance'
assert entry.data[config_flow.CONF_HOST] == 'mock-deconz'
assert result["type"] == "abort"
assert result["reason"] == "updated_instance"
assert entry.data[config_flow.CONF_HOST] == "mock-deconz"
async def test_import_without_api_key(hass):
"""Test importing a host without an API key."""
result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN,
data={config_flow.CONF_HOST: '1.2.3.4'},
context={'source': 'import'}
data={config_flow.CONF_HOST: "1.2.3.4"},
context={"source": "import"},
)
assert result['type'] == 'form'
assert result['step_id'] == 'link'
assert result["type"] == "form"
assert result["step_id"] == "link"
async def test_import_with_api_key(hass):
@@ -248,47 +251,49 @@ async def test_import_with_api_key(hass):
result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN,
data={
config_flow.CONF_BRIDGEID: 'id',
config_flow.CONF_HOST: 'mock-deconz',
config_flow.CONF_BRIDGEID: "id",
config_flow.CONF_HOST: "mock-deconz",
config_flow.CONF_PORT: 80,
config_flow.CONF_API_KEY: '1234567890ABCDEF'
config_flow.CONF_API_KEY: "1234567890ABCDEF",
},
context={'source': 'import'}
context={"source": "import"},
)
assert result['type'] == 'create_entry'
assert result['title'] == 'deCONZ-id'
assert result['data'] == {
config_flow.CONF_BRIDGEID: 'id',
config_flow.CONF_HOST: 'mock-deconz',
assert result["type"] == "create_entry"
assert result["title"] == "deCONZ-id"
assert result["data"] == {
config_flow.CONF_BRIDGEID: "id",
config_flow.CONF_HOST: "mock-deconz",
config_flow.CONF_PORT: 80,
config_flow.CONF_API_KEY: '1234567890ABCDEF'
config_flow.CONF_API_KEY: "1234567890ABCDEF",
}
async def test_create_entry(hass, aioclient_mock):
"""Test that _create_entry work and that bridgeid can be requested."""
aioclient_mock.get('http://1.2.3.4:80/api/1234567890ABCDEF/config',
json={"bridgeid": "id"},
headers={'content-type': 'application/json'})
aioclient_mock.get(
"http://1.2.3.4:80/api/1234567890ABCDEF/config",
json={"bridgeid": "id"},
headers={"content-type": "application/json"},
)
flow = config_flow.DeconzFlowHandler()
flow.hass = hass
flow.deconz_config = {
config_flow.CONF_HOST: '1.2.3.4',
config_flow.CONF_HOST: "1.2.3.4",
config_flow.CONF_PORT: 80,
config_flow.CONF_API_KEY: '1234567890ABCDEF'
config_flow.CONF_API_KEY: "1234567890ABCDEF",
}
result = await flow._create_entry()
assert result['type'] == 'create_entry'
assert result['title'] == 'deCONZ-id'
assert result['data'] == {
config_flow.CONF_BRIDGEID: 'id',
config_flow.CONF_HOST: '1.2.3.4',
assert result["type"] == "create_entry"
assert result["title"] == "deCONZ-id"
assert result["data"] == {
config_flow.CONF_BRIDGEID: "id",
config_flow.CONF_HOST: "1.2.3.4",
config_flow.CONF_PORT: 80,
config_flow.CONF_API_KEY: '1234567890ABCDEF'
config_flow.CONF_API_KEY: "1234567890ABCDEF",
}
@@ -297,40 +302,38 @@ async def test_create_entry_timeout(hass, aioclient_mock):
flow = config_flow.DeconzFlowHandler()
flow.hass = hass
flow.deconz_config = {
config_flow.CONF_HOST: '1.2.3.4',
config_flow.CONF_HOST: "1.2.3.4",
config_flow.CONF_PORT: 80,
config_flow.CONF_API_KEY: '1234567890ABCDEF'
config_flow.CONF_API_KEY: "1234567890ABCDEF",
}
with patch(
'homeassistant.components.deconz.config_flow.async_get_bridgeid',
side_effect=asyncio.TimeoutError):
"homeassistant.components.deconz.config_flow.async_get_bridgeid",
side_effect=asyncio.TimeoutError,
):
result = await flow._create_entry()
assert result['type'] == 'abort'
assert result['reason'] == 'no_bridges'
assert result["type"] == "abort"
assert result["reason"] == "no_bridges"
async def test_hassio_update_instance(hass):
"""Test we can update an existing config entry."""
entry = MockConfigEntry(domain=config_flow.DOMAIN, data={
config_flow.CONF_BRIDGEID: 'id',
config_flow.CONF_HOST: '1.2.3.4'
})
entry = MockConfigEntry(
domain=config_flow.DOMAIN,
data={config_flow.CONF_BRIDGEID: "id", config_flow.CONF_HOST: "1.2.3.4"},
)
entry.add_to_hass(hass)
result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN,
data={
config_flow.CONF_HOST: 'mock-deconz',
config_flow.CONF_SERIAL: 'id'
},
context={'source': 'hassio'}
data={config_flow.CONF_HOST: "mock-deconz", config_flow.CONF_SERIAL: "id"},
context={"source": "hassio"},
)
assert result['type'] == 'abort'
assert result['reason'] == 'updated_instance'
assert entry.data[config_flow.CONF_HOST] == 'mock-deconz'
assert result["type"] == "abort"
assert result["reason"] == "updated_instance"
assert entry.data[config_flow.CONF_HOST] == "mock-deconz"
async def test_hassio_confirm(hass):
@@ -338,27 +341,26 @@ async def test_hassio_confirm(hass):
result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN,
data={
'addon': 'Mock Addon',
config_flow.CONF_HOST: 'mock-deconz',
"addon": "Mock Addon",
config_flow.CONF_HOST: "mock-deconz",
config_flow.CONF_PORT: 80,
config_flow.CONF_SERIAL: 'id',
config_flow.CONF_API_KEY: '1234567890ABCDEF',
config_flow.CONF_SERIAL: "id",
config_flow.CONF_API_KEY: "1234567890ABCDEF",
},
context={'source': 'hassio'}
context={"source": "hassio"},
)
assert result['type'] == 'form'
assert result['step_id'] == 'hassio_confirm'
assert result['description_placeholders'] == {'addon': 'Mock Addon'}
assert result["type"] == "form"
assert result["step_id"] == "hassio_confirm"
assert result["description_placeholders"] == {"addon": "Mock Addon"}
result = await hass.config_entries.flow.async_configure(
result['flow_id'],
user_input={}
result["flow_id"], user_input={}
)
assert result['type'] == 'create_entry'
assert result['result'].data == {
config_flow.CONF_HOST: 'mock-deconz',
assert result["type"] == "create_entry"
assert result["result"].data == {
config_flow.CONF_HOST: "mock-deconz",
config_flow.CONF_PORT: 80,
config_flow.CONF_BRIDGEID: 'id',
config_flow.CONF_API_KEY: '1234567890ABCDEF'
config_flow.CONF_BRIDGEID: "id",
config_flow.CONF_API_KEY: "1234567890ABCDEF",
}