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

Migrate timer tests from coroutine to async/await (#30352)

Browse Source
This commit is contained in:
Franck Nijhof 2020-01-01 16:20:18 +01:00 committed by Martin Hjelmare
parent 9ba0daa358
commit 33828ae514
1 changed files with 11 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
tests/components/timer/test_init.py
View File

@ -1,6 +1,5 @@
"""The tests for the timer component.""" """The tests for the timer component."""
# pylint: disable=protected-access # pylint: disable=protected-access
import asyncio
from datetime import timedelta from datetime import timedelta
import logging import logging
from unittest.mock import patch from unittest.mock import patch
@ -89,14 +88,11 @@ async def test_config_options(hass):
assert "0:00:10" == state_2.attributes.get(ATTR_DURATION) assert "0:00:10" == state_2.attributes.get(ATTR_DURATION)
@asyncio.coroutine async def test_methods_and_events(hass):
def test_methods_and_events(hass):
"""Test methods and events.""" """Test methods and events."""
hass.state = CoreState.starting hass.state = CoreState.starting
yield from async_setup_component( await async_setup_component(hass, DOMAIN, {DOMAIN: {"test1": {CONF_DURATION: 10}}})
hass, DOMAIN, {DOMAIN: {"test1": {CONF_DURATION: 10}}}
)
state = hass.states.get("timer.test1") state = hass.states.get("timer.test1")
assert state assert state
@ -129,10 +125,10 @@ def test_methods_and_events(hass):
expectedEvents = 0 expectedEvents = 0
for step in steps: for step in steps:
if step["call"] is not None: if step["call"] is not None:
yield from hass.services.async_call( await hass.services.async_call(
DOMAIN, step["call"], {CONF_ENTITY_ID: "timer.test1"} DOMAIN, step["call"], {CONF_ENTITY_ID: "timer.test1"}
) )
yield from hass.async_block_till_done() await hass.async_block_till_done()
state = hass.states.get("timer.test1") state = hass.states.get("timer.test1")
assert state assert state
@ -145,14 +141,11 @@ def test_methods_and_events(hass):
assert len(results) == expectedEvents assert len(results) == expectedEvents
@asyncio.coroutine async def test_wait_till_timer_expires(hass):
def test_wait_till_timer_expires(hass):
"""Test for a timer to end.""" """Test for a timer to end."""
hass.state = CoreState.starting hass.state = CoreState.starting
yield from async_setup_component( await async_setup_component(hass, DOMAIN, {DOMAIN: {"test1": {CONF_DURATION: 10}}})
hass, DOMAIN, {DOMAIN: {"test1": {CONF_DURATION: 10}}}
)
state = hass.states.get("timer.test1") state = hass.states.get("timer.test1")
assert state assert state
@ -169,10 +162,10 @@ def test_wait_till_timer_expires(hass):
hass.bus.async_listen(EVENT_TIMER_FINISHED, fake_event_listener) hass.bus.async_listen(EVENT_TIMER_FINISHED, fake_event_listener)
hass.bus.async_listen(EVENT_TIMER_CANCELLED, fake_event_listener) hass.bus.async_listen(EVENT_TIMER_CANCELLED, fake_event_listener)
yield from hass.services.async_call( await hass.services.async_call(
DOMAIN, SERVICE_START, {CONF_ENTITY_ID: "timer.test1"} DOMAIN, SERVICE_START, {CONF_ENTITY_ID: "timer.test1"}
) )
yield from hass.async_block_till_done() await hass.async_block_till_done()
state = hass.states.get("timer.test1") state = hass.states.get("timer.test1")
assert state assert state
@ -182,7 +175,7 @@ def test_wait_till_timer_expires(hass):
assert len(results) == 1 assert len(results) == 1
async_fire_time_changed(hass, utcnow() + timedelta(seconds=10)) async_fire_time_changed(hass, utcnow() + timedelta(seconds=10))
yield from hass.async_block_till_done() await hass.async_block_till_done()
state = hass.states.get("timer.test1") state = hass.states.get("timer.test1")
assert state assert state
@ -192,14 +185,11 @@ def test_wait_till_timer_expires(hass):
assert len(results) == 2 assert len(results) == 2
@asyncio.coroutine async def test_no_initial_state_and_no_restore_state(hass):
def test_no_initial_state_and_no_restore_state(hass):
"""Ensure that entity is create without initial and restore feature.""" """Ensure that entity is create without initial and restore feature."""
hass.state = CoreState.starting hass.state = CoreState.starting
yield from async_setup_component( await async_setup_component(hass, DOMAIN, {DOMAIN: {"test1": {CONF_DURATION: 10}}})
hass, DOMAIN, {DOMAIN: {"test1": {CONF_DURATION: 10}}}
)
state = hass.states.get("timer.test1") state = hass.states.get("timer.test1")
assert state assert state