mirror of
https://github.com/home-assistant/core.git
synced 2025-07-21 04:07:08 +00:00
Fix Twinkly raise on progress (#133601)
This commit is contained in:
parent
8c1a18b383
commit
fdde9d3a52
@ -45,7 +45,9 @@ class TwinklyConfigFlow(ConfigFlow, domain=DOMAIN):
|
|||||||
except (TimeoutError, ClientError):
|
except (TimeoutError, ClientError):
|
||||||
errors[CONF_HOST] = "cannot_connect"
|
errors[CONF_HOST] = "cannot_connect"
|
||||||
else:
|
else:
|
||||||
await self.async_set_unique_id(device_info[DEV_ID])
|
await self.async_set_unique_id(
|
||||||
|
device_info[DEV_ID], raise_on_progress=False
|
||||||
|
)
|
||||||
self._abort_if_unique_id_configured()
|
self._abort_if_unique_id_configured()
|
||||||
|
|
||||||
return self._create_entry_from_device(device_info, host)
|
return self._create_entry_from_device(device_info, host)
|
||||||
|
@ -5,6 +5,7 @@ from unittest.mock import patch
|
|||||||
from homeassistant import config_entries
|
from homeassistant import config_entries
|
||||||
from homeassistant.components import dhcp
|
from homeassistant.components import dhcp
|
||||||
from homeassistant.components.twinkly.const import DOMAIN as TWINKLY_DOMAIN
|
from homeassistant.components.twinkly.const import DOMAIN as TWINKLY_DOMAIN
|
||||||
|
from homeassistant.config_entries import SOURCE_USER
|
||||||
from homeassistant.const import CONF_HOST, CONF_ID, CONF_MODEL, CONF_NAME
|
from homeassistant.const import CONF_HOST, CONF_ID, CONF_MODEL, CONF_NAME
|
||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
from homeassistant.data_entry_flow import FlowResultType
|
from homeassistant.data_entry_flow import FlowResultType
|
||||||
@ -157,3 +158,39 @@ async def test_dhcp_already_exists(hass: HomeAssistant) -> None:
|
|||||||
|
|
||||||
assert result["type"] is FlowResultType.ABORT
|
assert result["type"] is FlowResultType.ABORT
|
||||||
assert result["reason"] == "already_configured"
|
assert result["reason"] == "already_configured"
|
||||||
|
|
||||||
|
|
||||||
|
async def test_user_flow_works_discovery(hass: HomeAssistant) -> None:
|
||||||
|
"""Test user flow can continue after discovery happened."""
|
||||||
|
client = ClientMock()
|
||||||
|
with (
|
||||||
|
patch(
|
||||||
|
"homeassistant.components.twinkly.config_flow.Twinkly", return_value=client
|
||||||
|
),
|
||||||
|
patch("homeassistant.components.twinkly.async_setup_entry", return_value=True),
|
||||||
|
):
|
||||||
|
await hass.config_entries.flow.async_init(
|
||||||
|
TWINKLY_DOMAIN,
|
||||||
|
context={"source": config_entries.SOURCE_DHCP},
|
||||||
|
data=dhcp.DhcpServiceInfo(
|
||||||
|
hostname="Twinkly_XYZ",
|
||||||
|
ip="1.2.3.4",
|
||||||
|
macaddress="aabbccddeeff",
|
||||||
|
),
|
||||||
|
)
|
||||||
|
result = await hass.config_entries.flow.async_init(
|
||||||
|
TWINKLY_DOMAIN,
|
||||||
|
context={"source": SOURCE_USER},
|
||||||
|
)
|
||||||
|
assert len(hass.config_entries.flow.async_progress(TWINKLY_DOMAIN)) == 2
|
||||||
|
assert result["type"] is FlowResultType.FORM
|
||||||
|
assert result["step_id"] == "user"
|
||||||
|
|
||||||
|
result = await hass.config_entries.flow.async_configure(
|
||||||
|
result["flow_id"],
|
||||||
|
{CONF_HOST: "10.0.0.131"},
|
||||||
|
)
|
||||||
|
assert result["type"] is FlowResultType.CREATE_ENTRY
|
||||||
|
|
||||||
|
# Verify the discovery flow was aborted
|
||||||
|
assert not hass.config_entries.flow.async_progress(TWINKLY_DOMAIN)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user