21 lines
557 B
Python
21 lines
557 B
Python
"""Ecovacs test util."""
|
|
import asyncio
|
|
|
|
from deebot_client.event_bus import EventBus
|
|
from deebot_client.events import Event
|
|
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
|
|
async def block_till_done(hass: HomeAssistant, event_bus: EventBus) -> None:
|
|
"""Block till done."""
|
|
await asyncio.gather(*event_bus._tasks)
|
|
await hass.async_block_till_done()
|
|
|
|
|
|
async def notify_and_wait(
|
|
hass: HomeAssistant, event_bus: EventBus, event: Event
|
|
) -> None:
|
|
"""Block till done."""
|
|
event_bus.notify(event)
|
|
await block_till_done(hass, event_bus)
|