evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Increase test line coverage of homeassistant/helpers/event.py to 100% (#37927)

Browse source 
* Increase test line coverage of homeassistant/helpers/event.py to 100%

* fix test
This commit is contained in:
J. Nick Koston 2020-07-16 18:47:53 -10:00 • committed by GitHub
parent aaf084d713
commit fa4e9c0485
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 72 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

7
homeassistant/helpers/event.py
View file

@ -317,14 +317,13 @@ def async_track_point_in_time(
hass: HomeAssistant, action: Callable[..., None], point_in_time: datetime
) -> CALLBACK_TYPE:
"""Add a listener that fires once after a specific point in time."""
utc_point_in_time = dt_util.as_utc(point_in_time)
@callback
def utc_converter(utc_now: datetime) -> None:
"""Convert passed in UTC now to local now."""
hass.async_run_job(action, dt_util.as_local(utc_now))
return async_track_point_in_utc_time(hass, utc_converter, utc_point_in_time)
return async_track_point_in_utc_time(hass, utc_converter, point_in_time)
track_point_in_time = threaded_listener_factory(async_track_point_in_time)
@ -337,13 +336,13 @@ def async_track_point_in_utc_time(
) -> CALLBACK_TYPE:
"""Add a listener that fires once after a specific point in UTC time."""
# Ensure point_in_time is UTC
point_in_time = dt_util.as_utc(point_in_time)
utc_point_in_time = dt_util.as_utc(point_in_time)
cancel_callback = hass.loop.call_at(
hass.loop.time() + point_in_time.timestamp() - time.time(),
hass.async_run_job,
action,
point_in_time,
utc_point_in_time,
)
@callback