e13d5bdc10
* refactoring for multi platform * adopted test_bridge to refactoring * refactoring tests for multi-platform additional coverage in config and init * comment for clarity * more specific imports from lib * library version bump * removed async_update * changed parameter order to start with hass * removed pylint disable * unsubscribe from signal dispatcher inherit from Entity * use device.unique_id * changed hass_obj to hass * added test for remove entity bug fix from the test * removed the polling try_connect. hate polling... it is now part of the async_setup() significantly makes the code clearer and simplifies the tests * removed leftover debug logs in the library * changed tests to get the entry_id from hass * changed place to assign hass.data only after success * fixes for test_init * removed assert * removed device_info * removed bridge internal from common * modified test_bridge to work without the bridge directly * removed bridge from test_existing_update * changed update to not use bridge internals * dyn_bridge fixture no longer used - removed
94 lines
3.3 KiB
Python
Executable file
94 lines
3.3 KiB
Python
Executable file
"""Test Dynalite config flow."""
|
|
|
|
from asynctest import CoroutineMock, patch
|
|
|
|
from homeassistant import config_entries
|
|
from homeassistant.components import dynalite
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
async def run_flow(hass, connection):
|
|
"""Run a flow with or without errors and return result."""
|
|
host = "1.2.3.4"
|
|
with patch(
|
|
"homeassistant.components.dynalite.bridge.DynaliteDevices.async_setup",
|
|
side_effect=connection,
|
|
):
|
|
result = await hass.config_entries.flow.async_init(
|
|
dynalite.DOMAIN,
|
|
context={"source": config_entries.SOURCE_IMPORT},
|
|
data={dynalite.CONF_HOST: host},
|
|
)
|
|
await hass.async_block_till_done()
|
|
return result
|
|
|
|
|
|
async def test_flow_works(hass):
|
|
"""Test a successful config flow."""
|
|
result = await run_flow(hass, [True, True])
|
|
assert result["type"] == "create_entry"
|
|
assert result["result"].state == "loaded"
|
|
|
|
|
|
async def test_flow_setup_fails(hass):
|
|
"""Test a flow where async_setup fails."""
|
|
result = await run_flow(hass, [False])
|
|
assert result["type"] == "abort"
|
|
assert result["reason"] == "no_connection"
|
|
|
|
|
|
async def test_flow_setup_fails_in_setup_entry(hass):
|
|
"""Test a flow where the initial check works but inside setup_entry, the bridge setup fails."""
|
|
result = await run_flow(hass, [True, False])
|
|
assert result["type"] == "create_entry"
|
|
assert result["result"].state == "setup_retry"
|
|
|
|
|
|
async def test_existing(hass):
|
|
"""Test when the entry exists with the same config."""
|
|
host = "1.2.3.4"
|
|
MockConfigEntry(
|
|
domain=dynalite.DOMAIN, unique_id=host, data={dynalite.CONF_HOST: host}
|
|
).add_to_hass(hass)
|
|
with patch(
|
|
"homeassistant.components.dynalite.bridge.DynaliteDevices.async_setup",
|
|
return_value=True,
|
|
):
|
|
result = await hass.config_entries.flow.async_init(
|
|
dynalite.DOMAIN,
|
|
context={"source": config_entries.SOURCE_IMPORT},
|
|
data={dynalite.CONF_HOST: host},
|
|
)
|
|
assert result["type"] == "abort"
|
|
assert result["reason"] == "already_configured"
|
|
|
|
|
|
async def test_existing_update(hass):
|
|
"""Test when the entry exists with a different config."""
|
|
host = "1.2.3.4"
|
|
port1 = 7777
|
|
port2 = 8888
|
|
entry = MockConfigEntry(
|
|
domain=dynalite.DOMAIN,
|
|
data={dynalite.CONF_HOST: host, dynalite.CONF_PORT: port1},
|
|
)
|
|
entry.add_to_hass(hass)
|
|
with patch(
|
|
"homeassistant.components.dynalite.bridge.DynaliteDevices"
|
|
) as mock_dyn_dev:
|
|
mock_dyn_dev().async_setup = CoroutineMock(return_value=True)
|
|
assert await hass.config_entries.async_setup(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
mock_dyn_dev().configure.assert_called_once()
|
|
assert mock_dyn_dev().configure.mock_calls[0][1][0][dynalite.CONF_PORT] == port1
|
|
result = await hass.config_entries.flow.async_init(
|
|
dynalite.DOMAIN,
|
|
context={"source": config_entries.SOURCE_IMPORT},
|
|
data={dynalite.CONF_HOST: host, dynalite.CONF_PORT: port2},
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert mock_dyn_dev().configure.call_count == 2
|
|
assert mock_dyn_dev().configure.mock_calls[1][1][0][dynalite.CONF_PORT] == port2
|
|
assert result["type"] == "abort"
|
|
assert result["reason"] == "already_configured"
|