Skip to content

Commit

Permalink
test: add test for cancelation during job execution
Browse files Browse the repository at this point in the history
  • Loading branch information
swoga committed May 22, 2024
1 parent d0730b3 commit 0291ed7
Showing 1 changed file with 34 additions and 0 deletions.
34 changes: 34 additions & 0 deletions tests/helpers/test_event.py
Original file line number Diff line number Diff line change
Expand Up @@ -4589,6 +4589,40 @@ def run_callback(local_time):
assert "US/Hawaii" in str(times[0].tzinfo)


async def test_async_track_point_in_time_cancel_in_job(
hass: HomeAssistant, freezer: FrozenDateTimeFactory
) -> None:
"""Test cancel of async track point in time during job execution."""

now = dt_util.utcnow()
times = []

time_that_will_not_match_right_away = datetime(
now.year + 1, 5, 24, 11, 59, 55, tzinfo=dt_util.UTC
)
freezer.move_to(time_that_will_not_match_right_away)

@callback
def action(x: datetime):
nonlocal times
times.append(x)
unsub()

unsub = async_track_utc_time_change(hass, action, minute=0, second="*")

async_fire_time_changed(
hass, datetime(now.year + 1, 5, 24, 12, 0, 0, 999999, tzinfo=dt_util.UTC)
)
await hass.async_block_till_done()
assert len(times) == 1

async_fire_time_changed(
hass, datetime(now.year + 1, 5, 24, 13, 0, 0, 999999, tzinfo=dt_util.UTC)
)
await hass.async_block_till_done()
assert len(times) == 1


async def test_async_track_entity_registry_updated_event(hass: HomeAssistant) -> None:
"""Test tracking entity registry updates for an entity_id."""

Expand Down

0 comments on commit 0291ed7

Please sign in to comment.