mirror of
https://github.com/home-assistant/core
synced 2024-09-18 19:55:20 +02:00
6d57dbde68
* Support passing callbacks to start.async_at_start * Update homeassistant/helpers/start.py Co-authored-by: Franck Nijhof <git@frenck.dev> * Fix imports Co-authored-by: Franck Nijhof <git@frenck.dev>
91 lines
2.3 KiB
Python
91 lines
2.3 KiB
Python
"""Test starting HA helpers."""
|
|
from homeassistant import core
|
|
from homeassistant.const import EVENT_HOMEASSISTANT_START
|
|
from homeassistant.helpers import start
|
|
|
|
|
|
async def test_at_start_when_running_awaitable(hass):
|
|
"""Test at start when already running."""
|
|
assert hass.state == core.CoreState.running
|
|
assert hass.is_running
|
|
|
|
calls = []
|
|
|
|
async def cb_at_start(hass):
|
|
"""Home Assistant is started."""
|
|
calls.append(1)
|
|
|
|
start.async_at_start(hass, cb_at_start)
|
|
await hass.async_block_till_done()
|
|
assert len(calls) == 1
|
|
|
|
hass.state = core.CoreState.starting
|
|
assert hass.is_running
|
|
|
|
start.async_at_start(hass, cb_at_start)
|
|
await hass.async_block_till_done()
|
|
assert len(calls) == 2
|
|
|
|
|
|
async def test_at_start_when_running_callback(hass):
|
|
"""Test at start when already running."""
|
|
assert hass.state == core.CoreState.running
|
|
assert hass.is_running
|
|
|
|
calls = []
|
|
|
|
@core.callback
|
|
def cb_at_start(hass):
|
|
"""Home Assistant is started."""
|
|
calls.append(1)
|
|
|
|
start.async_at_start(hass, cb_at_start)
|
|
assert len(calls) == 1
|
|
|
|
hass.state = core.CoreState.starting
|
|
assert hass.is_running
|
|
|
|
start.async_at_start(hass, cb_at_start)
|
|
assert len(calls) == 2
|
|
|
|
|
|
async def test_at_start_when_starting_awaitable(hass):
|
|
"""Test at start when yet to start."""
|
|
hass.state = core.CoreState.not_running
|
|
assert not hass.is_running
|
|
|
|
calls = []
|
|
|
|
async def cb_at_start(hass):
|
|
"""Home Assistant is started."""
|
|
calls.append(1)
|
|
|
|
start.async_at_start(hass, cb_at_start)
|
|
await hass.async_block_till_done()
|
|
assert len(calls) == 0
|
|
|
|
hass.bus.async_fire(EVENT_HOMEASSISTANT_START)
|
|
await hass.async_block_till_done()
|
|
assert len(calls) == 1
|
|
|
|
|
|
async def test_at_start_when_starting_callback(hass):
|
|
"""Test at start when yet to start."""
|
|
hass.state = core.CoreState.not_running
|
|
assert not hass.is_running
|
|
|
|
calls = []
|
|
|
|
@core.callback
|
|
def cb_at_start(hass):
|
|
"""Home Assistant is started."""
|
|
calls.append(1)
|
|
|
|
start.async_at_start(hass, cb_at_start)
|
|
await hass.async_block_till_done()
|
|
assert len(calls) == 0
|
|
|
|
hass.bus.async_fire(EVENT_HOMEASSISTANT_START)
|
|
await hass.async_block_till_done()
|
|
assert len(calls) == 1
|