self-hosted/home-assistant-core
self-hosted/home-assistant-core
1
0
Fork 0
mirror of https://github.com/home-assistant/core synced 2024-10-05 17:48:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix event test (#42078)

Browse source
This commit is contained in:
Paulus Schoutsen 2020-10-19 16:50:46 +02:00 committed by GitHub
parent 16c402f583
commit 8d4675713a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 47 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
tests/components/time_date/test_sensor.py
View file

@ -1,9 +1,20 @@
"""The tests for time_date sensor platform."""
import pytest
import homeassistant.components.time_date.sensor as time_date
import homeassistant.util.dt as dt_util
from tests.async_mock import patch
ORIG_TZ = dt_util.DEFAULT_TIME_ZONE
@pytest.fixture(autouse=True)
def restore_ts():
"""Restore default TZ."""
yield
dt_util.DEFAULT_TIME_ZONE = ORIG_TZ
# pylint: disable=protected-access
async def test_intervals(hass):

42
tests/helpers/test_event.py
View file

@ -1545,7 +1545,12 @@ async def test_track_template_rate_limit_suppress_listener(hass):
await hass.async_block_till_done()
info.async_refresh()
assert info.listeners == {"all": True, "domains": set(), "entities": set()}
assert info.listeners == {
"all": True,
"domains": set(),
"entities": set(),
"time": False,
}
await hass.async_block_till_done()
assert refresh_runs == [0]
@ -1557,7 +1562,12 @@ async def test_track_template_rate_limit_suppress_listener(hass):
hass.states.async_set("sensor.two", "any")
await hass.async_block_till_done()
# Should be suppressed during the rate limit
assert info.listeners == {"all": False, "domains": set(), "entities": set()}
assert info.listeners == {
"all": False,
"domains": set(),
"entities": set(),
"time": False,
}
assert refresh_runs == [0, 1]
next_time = dt_util.utcnow() + timedelta(seconds=0.125)
with patch(
@ -1566,7 +1576,12 @@ async def test_track_template_rate_limit_suppress_listener(hass):
async_fire_time_changed(hass, next_time)
await hass.async_block_till_done()
# Rate limit released and the all listener returns
assert info.listeners == {"all": True, "domains": set(), "entities": set()}
assert info.listeners == {
"all": True,
"domains": set(),
"entities": set(),
"time": False,
}
assert refresh_runs == [0, 1, 2]
hass.states.async_set("sensor.three", "any")
await hass.async_block_till_done()
@ -1575,7 +1590,12 @@ async def test_track_template_rate_limit_suppress_listener(hass):
await hass.async_block_till_done()
assert refresh_runs == [0, 1, 2]
# Rate limit hit and the all listener is shut off
assert info.listeners == {"all": False, "domains": set(), "entities": set()}
assert info.listeners == {
"all": False,
"domains": set(),
"entities": set(),
"time": False,
}
next_time = dt_util.utcnow() + timedelta(seconds=0.125 * 2)
with patch(
"homeassistant.helpers.ratelimit.dt_util.utcnow", return_value=next_time
@ -1583,12 +1603,22 @@ async def test_track_template_rate_limit_suppress_listener(hass):
async_fire_time_changed(hass, next_time)
await hass.async_block_till_done()
# Rate limit released and the all listener returns
assert info.listeners == {"all": True, "domains": set(), "entities": set()}
assert info.listeners == {
"all": True,
"domains": set(),
"entities": set(),
"time": False,
}
assert refresh_runs == [0, 1, 2, 4]
hass.states.async_set("sensor.five", "any")
await hass.async_block_till_done()
# Rate limit hit and the all listener is shut off
assert info.listeners == {"all": False, "domains": set(), "entities": set()}
assert info.listeners == {
"all": False,
"domains": set(),
"entities": set(),
"time": False,
}
assert refresh_runs == [0, 1, 2, 4]