home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-08-25 23:41:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

second background task

Browse Source
This commit is contained in:
Thomas Protzner
2024-08-02 20:32:02 +00:00
parent 9645412148
commit 95807c6650
2 changed files with 12 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
homeassistant/components/husqvarna_automower/coordinator.py
View File

@@ -74,28 +74,13 @@ class AutomowerDataUpdateCoordinator(DataUpdateCoordinator[dict[str, MowerAttrib
"Failed to connect to websocket. Trying to reconnect: %s", err "Failed to connect to websocket. Trying to reconnect: %s", err
) )
if not hass.is_stopping: if not hass.is_stopping:
# self.hass.async_create_background_task(
# hass,
# self.client_listen(hass, entry, automower_client),
# "reconnect_task",
# )
reconnect_time = 2 reconnect_time = 2
await asyncio.sleep(reconnect_time) await asyncio.sleep(reconnect_time)
reconnect_time = min(reconnect_time * 2, MAX_WS_RECONNECT_TIME) reconnect_time = min(reconnect_time * 2, MAX_WS_RECONNECT_TIME)
await self.client_listen( entry.async_create_background_task(
hass=hass, hass,
entry=entry, self.client_listen(hass, entry, automower_client),
automower_client=automower_client, "reconnect_task",
reconnect_time=reconnect_time,
) )
else: else:
await automower_client.start_listening() await automower_client.start_listening()
async def try_reconnect(
self,
hass: HomeAssistant,
entry: ConfigEntry,
automower_client: AutomowerSession,
reconnect_time: int = 2,
) -> None:
"""Listen with the client."""

9
tests/components/husqvarna_automower/test_init.py
View File

@@ -141,12 +141,19 @@ async def test_websocket_not_available(
reconnect_time = 2 reconnect_time = 2
for count in range(1, 945): for count in range(1, 945):
reconnect_time = min(reconnect_time * 2, MAX_WS_RECONNECT_TIME) reconnect_time = min(reconnect_time * 2, MAX_WS_RECONNECT_TIME)
print("reconnect_t", reconnect_time)
freezer.tick(timedelta(seconds=reconnect_time)) freezer.tick(timedelta(seconds=reconnect_time))
async_fire_time_changed(hass) async_fire_time_changed(hass)
await hass.async_block_till_done() await hass.async_block_till_done()
assert mock_automower_client.auth.websocket_connect.call_count == count + 1 assert mock_automower_client.auth.websocket_connect.call_count == count + 1
assert mock_config_entry.state is ConfigEntryState.LOADED assert mock_config_entry.state is ConfigEntryState.LOADED
mock_automower_client.auth.websocket_connect.side_effect = None
freezer.tick(timedelta(seconds=MAX_WS_RECONNECT_TIME))
async_fire_time_changed(hass)
await hass.async_block_till_done()
assert mock_automower_client.start_listening.call_count == 1
mock_automower_client.auth.websocket_connect.side_effect = (
HusqvarnaWSServerHandshakeError("Boom")
)
async def test_device_info( async def test_device_info(