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

Migrate snooze config flow to use eager_start (#115658)

Browse source
This commit is contained in:
J. Nick Koston 2024-04-17 14:59:16 -05:00 • committed by GitHub
parent 7188d62340
commit b829f1030b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 10 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
homeassistant/components/snooz/config_flow.py
View file

@ -132,7 +132,7 @@ class SnoozConfigFlow(ConfigFlow, domain=DOMAIN):
"""Wait for device to enter pairing mode.""" """Wait for device to enter pairing mode."""
if not self._pairing_task: if not self._pairing_task:
self._pairing_task = self.hass.async_create_task( self._pairing_task = self.hass.async_create_task(
self._async_wait_for_pairing_mode(), eager_start=False self._async_wait_for_pairing_mode()
) )
if not self._pairing_task.done(): if not self._pairing_task.done():

11
tests/components/snooz/test_config_flow.py
View file

@ -2,7 +2,7 @@
from __future__ import annotations from __future__ import annotations
from asyncio import Event from asyncio import Event, sleep
from unittest.mock import patch from unittest.mock import patch
from homeassistant import config_entries from homeassistant import config_entries
@ -298,9 +298,16 @@ async def _test_pairs(
async def _test_pairs_timeout( async def _test_pairs_timeout(
hass: HomeAssistant, flow_id: str, user_input: dict | None = None hass: HomeAssistant, flow_id: str, user_input: dict | None = None
) -> str: ) -> str:
async def _async_process_advertisements(
_hass, _callback, _matcher, _mode, _timeout
):
"""Simulate a timeout waiting for pairing mode."""
await sleep(0)
raise TimeoutError
with patch( with patch(
"homeassistant.components.snooz.config_flow.async_process_advertisements", "homeassistant.components.snooz.config_flow.async_process_advertisements",
side_effect=TimeoutError(), _async_process_advertisements,
): ):
result = await hass.config_entries.flow.async_configure( result = await hass.config_entries.flow.async_configure(
flow_id, user_input=user_input or {} flow_id, user_input=user_input or {}