diff --git a/tests/components/radarr/test_calendar.py b/tests/components/radarr/test_calendar.py index e82760cadba..ecf8433a445 100644 --- a/tests/components/radarr/test_calendar.py +++ b/tests/components/radarr/test_calendar.py @@ -4,13 +4,12 @@ from datetime import timedelta from freezegun.api import FrozenDateTimeFactory -from homeassistant.components.radarr.const import DOMAIN from homeassistant.const import STATE_OFF, STATE_ON from homeassistant.core import HomeAssistant -from homeassistant.helpers.update_coordinator import DataUpdateCoordinator from . import setup_integration +from tests.common import async_fire_time_changed from tests.test_util.aiohttp import AiohttpClientMocker @@ -21,8 +20,7 @@ async def test_calendar( ) -> None: """Test for successfully setting up the Radarr platform.""" freezer.move_to("2021-12-02 00:00:00-08:00") - entry = await setup_integration(hass, aioclient_mock) - coordinator: DataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]["calendar"] + await setup_integration(hass, aioclient_mock) state = hass.states.get("calendar.mock_title") assert state.state == STATE_ON @@ -33,8 +31,9 @@ async def test_calendar( assert state.attributes.get("release_type") == "physicalRelease" assert state.attributes.get("start_time") == "2021-12-02 00:00:00" - freezer.tick(timedelta(hours=16)) - await coordinator.async_refresh() + freezer.tick(timedelta(days=1)) + async_fire_time_changed(hass) + await hass.async_block_till_done() state = hass.states.get("calendar.mock_title") assert state.state == STATE_OFF