Avoid touching internals in Radarr tests (#119821)

* Avoid touching internals in Radarr tests

* Fix

* Fix
This commit is contained in:
Joost Lekkerkerker 2024-06-17 13:33:36 +02:00 committed by GitHub
parent cfbc854c84
commit 9f46b582d3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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