Migrate demo test to use freezegun (#105644)

pull/105653/head
Jan-Philipp Benecke 2023-12-13 14:33:59 +01:00 committed by GitHub
parent ac53b78a0c
commit 61a99c911c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 8 deletions

View File

@ -2,6 +2,7 @@
from unittest.mock import patch
from freezegun.api import FrozenDateTimeFactory
import pytest
from homeassistant.components.button import DOMAIN, SERVICE_PRESS
@ -37,20 +38,20 @@ def test_setup_params(hass: HomeAssistant) -> None:
assert state.state == STATE_UNKNOWN
async def test_press(hass: HomeAssistant) -> None:
async def test_press(hass: HomeAssistant, freezer: FrozenDateTimeFactory) -> None:
"""Test pressing the button."""
state = hass.states.get(ENTITY_PUSH)
assert state
assert state.state == STATE_UNKNOWN
now = dt_util.parse_datetime("2021-01-09 12:00:00+00:00")
with patch("homeassistant.util.dt.utcnow", return_value=now):
await hass.services.async_call(
DOMAIN,
SERVICE_PRESS,
{ATTR_ENTITY_ID: ENTITY_PUSH},
blocking=True,
)
freezer.move_to(now)
await hass.services.async_call(
DOMAIN,
SERVICE_PRESS,
{ATTR_ENTITY_ID: ENTITY_PUSH},
blocking=True,
)
state = hass.states.get(ENTITY_PUSH)
assert state