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

Support updating deCONZ host address (#22784)

Browse source 
* Update config flow to support updating host address
Improve tests

* Update gateway to handle new address signal

* Improve description why whe need to keep step_init
This commit is contained in:
Robert Svensson 2019-04-15 06:50:01 +02:00 • committed by Paulus Schoutsen
parent 96f689a70f
commit 23cb579f9f
4 changed files with 217 additions and 116 deletions
Download patch file Download diff file Expand all files Collapse all files

273
tests/components/deconz/test_config_flow.py
View file

@ -18,49 +18,57 @@ async def test_flow_works(hass, aioclient_mock):
{"success": {"username": "1234567890ABCDEF"}}
], headers={'content-type': 'application/json'})
flow = config_flow.DeconzFlowHandler()
flow.hass = hass
result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN,
context={'source': 'user'}
)
await flow.async_step_user()
result = await flow.async_step_link(user_input={})
assert result['type'] == 'form'
assert result['step_id'] == 'link'
result = await hass.config_entries.flow.async_configure(
result['flow_id'],
user_input={}
)
assert result['type'] == 'create_entry'
assert result['title'] == 'deCONZ-id'
assert result['data'] == {
'bridgeid': 'id',
'host': '1.2.3.4',
'port': 80,
'api_key': '1234567890ABCDEF'
config_flow.CONF_BRIDGEID: 'id',
config_flow.CONF_HOST: '1.2.3.4',
config_flow.CONF_PORT: 80,
config_flow.CONF_API_KEY: '1234567890ABCDEF'
}
async def test_flow_bridge_discovery_fails(hass, aioclient_mock):
async def test_user_step_bridge_discovery_fails(hass, aioclient_mock):
"""Test config flow works when discovery fails."""
flow = config_flow.DeconzFlowHandler()
flow.hass = hass
with patch('pydeconz.utils.async_discovery',
side_effect=asyncio.TimeoutError):
result = await flow.async_step_user()
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_flow_no_discovered_bridges(hass, aioclient_mock):
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'})
flow = config_flow.DeconzFlowHandler()
flow.hass = hass
result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN,
context={'source': 'user'}
)
result = await flow.async_step_user()
assert result['type'] == 'form'
assert result['step_id'] == 'init'
async def test_flow_one_bridge_discovered(hass, aioclient_mock):
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}
@ -70,61 +78,85 @@ async def test_flow_one_bridge_discovered(hass, aioclient_mock):
flow.hass = hass
result = await flow.async_step_user()
assert result['type'] == 'form'
assert result['step_id'] == 'link'
assert flow.deconz_config['host'] == '1.2.3.4'
assert flow.deconz_config[config_flow.CONF_HOST] == '1.2.3.4'
async def test_flow_two_bridges_discovered(hass, aioclient_mock):
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'})
flow = config_flow.DeconzFlowHandler()
flow.hass = hass
result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN,
context={'source': 'user'}
)
result = await flow.async_step_user()
assert result['data_schema']({'host': '1.2.3.4'})
assert result['data_schema']({'host': '5.6.7.8'})
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_flow_two_bridges_selection(hass, aioclient_mock):
async def test_user_step_two_bridges_selection(hass, aioclient_mock):
"""Test config flow selection of one of two bridges."""
flow = config_flow.DeconzFlowHandler()
flow.hass = hass
flow.bridges = [
{'bridgeid': 'id1', 'host': '1.2.3.4', 'port': 80},
{'bridgeid': 'id2', 'host': '5.6.7.8', '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
}
]
result = await flow.async_step_user(user_input={'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['host'] == '1.2.3.4'
assert flow.deconz_config[config_flow.CONF_HOST] == '1.2.3.4'
async def test_flow_manual_configuration(hass, aioclient_mock):
async def test_user_step_manual_configuration(hass, aioclient_mock):
"""Test config flow with manual input."""
aioclient_mock.get(pydeconz.utils.URL_DISCOVER, json=[])
aioclient_mock.get(pydeconz.utils.URL_DISCOVER, json=[],
headers={'content-type': 'application/json'})
flow = config_flow.DeconzFlowHandler()
flow.hass = hass
result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN,
context={'source': 'user'}
)
user_input = {'host': '1.2.3.4', 'port': 80}
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 = await flow.async_step_user(user_input)
assert result['type'] == 'form'
assert result['step_id'] == 'link'
assert flow.deconz_config == user_input
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 = {'host': '1.2.3.4', 'port': 80}
flow.deconz_config = {
config_flow.CONF_HOST: '1.2.3.4',
config_flow.CONF_PORT: 80
}
with patch('pydeconz.utils.async_get_api_key',
side_effect=pydeconz.errors.ResponseError):
@ -137,45 +169,48 @@ async def test_link_no_api_key(hass):
async def test_bridge_discovery(hass):
"""Test a bridge being discovered."""
flow = config_flow.DeconzFlowHandler()
flow.hass = hass
result = await flow.async_step_discovery({
'host': '1.2.3.4',
'port': 80,
'serial': 'id'
})
result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN,
data={
config_flow.CONF_HOST: '1.2.3.4',
config_flow.CONF_PORT: 80,
config_flow.CONF_SERIAL: 'id',
},
context={'source': 'discovery'}
)
assert result['type'] == 'form'
assert result['step_id'] == 'link'
async def test_bridge_discovery_already_configured(hass):
async def test_bridge_discovery_update_existing_entry(hass):
"""Test if a discovered bridge has already been configured."""
MockConfigEntry(domain='deconz', data={
'host': '1.2.3.4'
}).add_to_hass(hass)
flow = config_flow.DeconzFlowHandler()
flow.hass = hass
result = await flow.async_step_discovery({
'host': '1.2.3.4',
'port': 80,
'serial': '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)
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': 'discovery'}
)
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."""
flow = config_flow.DeconzFlowHandler()
flow.hass = hass
result = await flow.async_step_import({
'host': '1.2.3.4',
})
result = await hass.config_entries.flow.async_init(
config_flow.DOMAIN,
data={config_flow.CONF_HOST: '1.2.3.4'},
context={'source': 'import'}
)
assert result['type'] == 'form'
assert result['step_id'] == 'link'
@ -183,23 +218,24 @@ async def test_import_without_api_key(hass):
async def test_import_with_api_key(hass):
"""Test importing a host with an API key."""
flow = config_flow.DeconzFlowHandler()
flow.hass = hass
result = await flow.async_step_import({
'bridgeid': 'id',
'host': '1.2.3.4',
'port': 80,
'api_key': '1234567890ABCDEF'
})
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_PORT: 80,
config_flow.CONF_API_KEY: '1234567890ABCDEF'
},
context={'source': 'import'}
)
assert result['type'] == 'create_entry'
assert result['title'] == 'deCONZ-id'
assert result['data'] == {
'bridgeid': 'id',
'host': '1.2.3.4',
'port': 80,
'api_key': '1234567890ABCDEF'
config_flow.CONF_BRIDGEID: 'id',
config_flow.CONF_HOST: 'mock-deconz',
config_flow.CONF_PORT: 80,
config_flow.CONF_API_KEY: '1234567890ABCDEF'
}
@ -211,61 +247,90 @@ async def test_create_entry(hass, aioclient_mock):
flow = config_flow.DeconzFlowHandler()
flow.hass = hass
flow.deconz_config = {'host': '1.2.3.4',
'port': 80,
'api_key': '1234567890ABCDEF'}
flow.deconz_config = {
config_flow.CONF_HOST: '1.2.3.4',
config_flow.CONF_PORT: 80,
config_flow.CONF_API_KEY: '1234567890ABCDEF'
}
result = await flow._create_entry()
assert result['type'] == 'create_entry'
assert result['title'] == 'deCONZ-id'
assert result['data'] == {
'bridgeid': 'id',
'host': '1.2.3.4',
'port': 80,
'api_key': '1234567890ABCDEF'
config_flow.CONF_BRIDGEID: 'id',
config_flow.CONF_HOST: '1.2.3.4',
config_flow.CONF_PORT: 80,
config_flow.CONF_API_KEY: '1234567890ABCDEF'
}
async def test_hassio_single_instance(hass):
"""Test we only allow a single config flow."""
MockConfigEntry(domain='deconz', data={
'host': '1.2.3.4'
}).add_to_hass(hass)
async def test_create_entry_timeout(hass, aioclient_mock):
"""Test that _create_entry handles a timeout."""
flow = config_flow.DeconzFlowHandler()
flow.hass = hass
flow.deconz_config = {
config_flow.CONF_HOST: '1.2.3.4',
config_flow.CONF_PORT: 80,
config_flow.CONF_API_KEY: '1234567890ABCDEF'
}
with patch('pydeconz.utils.async_get_bridgeid',
side_effect=asyncio.TimeoutError):
result = await flow._create_entry()
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.add_to_hass(hass)
result = await hass.config_entries.flow.async_init(
'deconz', context={'source': 'hassio'})
config_flow.DOMAIN,
data={
config_flow.CONF_HOST: 'mock-deconz',
config_flow.CONF_SERIAL: 'id'
},
context={'source': 'hassio'}
)
assert result['type'] == 'abort'
assert result['reason'] == 'one_instance_only'
assert result['reason'] == 'updated_instance'
assert entry.data[config_flow.CONF_HOST] == 'mock-deconz'
async def test_hassio_confirm(hass):
"""Test we can finish a config flow."""
result = await hass.config_entries.flow.async_init(
'deconz',
config_flow.DOMAIN,
data={
'addon': 'Mock Addon',
'host': 'mock-deconz',
'port': 80,
'serial': 'id',
'api_key': '1234567890ABCDEF',
config_flow.CONF_HOST: 'mock-deconz',
config_flow.CONF_PORT: 80,
config_flow.CONF_SERIAL: 'id',
config_flow.CONF_API_KEY: '1234567890ABCDEF',
},
context={'source': 'hassio'}
)
assert result['type'] == 'form'
assert result['step_id'] == 'hassio_confirm'
assert result['description_placeholders'] == {
'addon': 'Mock Addon',
}
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 == {
'host': 'mock-deconz',
'port': 80,
'bridgeid': 'id',
'api_key': '1234567890ABCDEF'
config_flow.CONF_HOST: 'mock-deconz',
config_flow.CONF_PORT: 80,
config_flow.CONF_BRIDGEID: 'id',
config_flow.CONF_API_KEY: '1234567890ABCDEF'
}