evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Automatically onboard DiscoveryFlows (#73841)

Browse source
This commit is contained in:
Franck Nijhof 2022-06-22 22:37:49 +02:00 • committed by GitHub
parent ec119ae718
commit a8a033681f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 26 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

24
tests/helpers/test_config_entry_flow.py
View file

@ -139,6 +139,30 @@ async def test_discovery_confirmation(hass, discovery_flow_conf, source):
assert result["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
@pytest.mark.parametrize(
"source",
[
config_entries.SOURCE_DISCOVERY,
config_entries.SOURCE_MQTT,
config_entries.SOURCE_SSDP,
config_entries.SOURCE_ZEROCONF,
config_entries.SOURCE_DHCP,
],
)
async def test_discovery_during_onboarding(hass, discovery_flow_conf, source):
"""Test we create config entry via discovery during onboarding."""
flow = config_entries.HANDLERS["test"]()
flow.hass = hass
flow.context = {"source": source}
with patch(
"homeassistant.components.onboarding.async_is_onboarded", return_value=False
):
result = await getattr(flow, f"async_step_{source}")({})
assert result["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
async def test_multiple_discoveries(hass, discovery_flow_conf):
"""Test we only create one instance for multiple discoveries."""
mock_entity_platform(hass, "config_flow.test", None)