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

Remove usage of event_loop fixture (#7776)

Browse Source
This commit is contained in:
Paulus Schoutsen 2017-05-25 21:40:36 -07:00 committed by GitHub
parent 81aaeaaf11
commit d0c9d6b69a
1 changed files with 18 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
tests/helpers/test_state.py
View File

@ -22,40 +22,30 @@ from homeassistant.components.sun import (STATE_ABOVE_HORIZON,
from tests.common import get_test_home_assistant, mock_service from tests.common import get_test_home_assistant, mock_service
def test_async_track_states(event_loop): @asyncio.coroutine
def test_async_track_states(hass):
"""Test AsyncTrackStates context manager.""" """Test AsyncTrackStates context manager."""
hass = get_test_home_assistant() point1 = dt_util.utcnow()
point2 = point1 + timedelta(seconds=5)
point3 = point2 + timedelta(seconds=5)
try: with patch('homeassistant.core.dt_util.utcnow') as mock_utcnow:
point1 = dt_util.utcnow() mock_utcnow.return_value = point2
point2 = point1 + timedelta(seconds=5)
point3 = point2 + timedelta(seconds=5) with state.AsyncTrackStates(hass) as states:
mock_utcnow.return_value = point1
hass.states.async_set('light.test', 'on')
@asyncio.coroutine
@patch('homeassistant.core.dt_util.utcnow')
def run_test(mock_utcnow):
"""Run the test."""
mock_utcnow.return_value = point2 mock_utcnow.return_value = point2
hass.states.async_set('light.test2', 'on')
state2 = hass.states.get('light.test2')
with state.AsyncTrackStates(hass) as states: mock_utcnow.return_value = point3
mock_utcnow.return_value = point1 hass.states.async_set('light.test3', 'on')
hass.states.set('light.test', 'on') state3 = hass.states.get('light.test3')
mock_utcnow.return_value = point2 assert [state2, state3] == \
hass.states.set('light.test2', 'on') sorted(states, key=lambda state: state.entity_id)
state2 = hass.states.get('light.test2')
mock_utcnow.return_value = point3
hass.states.set('light.test3', 'on')
state3 = hass.states.get('light.test3')
assert [state2, state3] == \
sorted(states, key=lambda state: state.entity_id)
event_loop.run_until_complete(run_test())
finally:
hass.stop()
class TestStateHelpers(unittest.TestCase): class TestStateHelpers(unittest.TestCase):