jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-04-23 00:37:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix flakey influxdb test (#115442)

Browse Source
This commit is contained in:
J. Nick Koston 2024-04-11 13:26:03 -10:00 committed by GitHub
parent cfda8f64b4
commit e3680044fe
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/components/influxdb/test_init.py
View File

@ -1,5 +1,6 @@
"""The tests for the InfluxDB component."""
import asyncio
from dataclasses import dataclass
import datetime
from http import HTTPStatus
@ -1572,12 +1573,21 @@ async def test_invalid_inputs_error(
await _setup(hass, mock_client, config_ext, get_write_api)
write_api = get_write_api(mock_client)
write_api.side_effect = test_exception
write_api_done_event = asyncio.Event()
def wait_for_write(*args, **kwargs):
hass.loop.call_soon_threadsafe(write_api_done_event.set)
raise test_exception
write_api.side_effect = wait_for_write
with patch(f"{INFLUX_PATH}.time.sleep") as sleep:
write_api_done_event.clear()
hass.states.async_set("fake.something", 1)
await hass.async_block_till_done()
await async_wait_for_queue_to_process(hass)
await write_api_done_event.wait()
await hass.async_block_till_done()
write_api.assert_called_once()