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

Add test for radarr update failure (#116882)

Browse source
This commit is contained in:
Robert Hillis 2024-05-12 14:23:53 -04:00 • committed by GitHub
parent 3434fb70fb
commit 0a8feae49a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 37 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

38
tests/components/radarr/test_sensor.py
View file

@ -1,5 +1,9 @@
"""The tests for Radarr sensor platform."""
from datetime import timedelta
from unittest.mock import patch
from aiopyarr.exceptions import ArrConnectionException
import pytest
from homeassistant.components.sensor import (
@ -7,11 +11,18 @@ from homeassistant.components.sensor import (
SensorDeviceClass,
SensorStateClass,
)
from homeassistant.const import ATTR_DEVICE_CLASS, ATTR_UNIT_OF_MEASUREMENT
from homeassistant.config_entries import ConfigEntryState
from homeassistant.const import (
ATTR_DEVICE_CLASS,
ATTR_UNIT_OF_MEASUREMENT,
STATE_UNAVAILABLE,
)
from homeassistant.core import HomeAssistant
import homeassistant.util.dt as dt_util
from . import setup_integration
from tests.common import async_fire_time_changed
from tests.test_util.aiohttp import AiohttpClientMocker
@ -76,3 +87,28 @@ async def test_windows(
state = hass.states.get("sensor.mock_title_disk_space_tv")
assert state.state == "263.10"
async def test_update_failed(
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker
) -> None:
"""Test coordinator updates handle failures."""
entry = await setup_integration(hass, aioclient_mock)
assert entry.state is ConfigEntryState.LOADED
entity = "sensor.mock_title_disk_space_downloads"
assert hass.states.get(entity).state == "263.10"
with patch(
"homeassistant.components.radarr.RadarrClient._async_request",
side_effect=ArrConnectionException,
) as updater:
next_update = dt_util.utcnow() + timedelta(minutes=1)
async_fire_time_changed(hass, next_update)
await hass.async_block_till_done()
assert updater.call_count == 2
assert hass.states.get(entity).state == STATE_UNAVAILABLE
next_update = dt_util.utcnow() + timedelta(minutes=1)
async_fire_time_changed(hass, next_update)
await hass.async_block_till_done()
assert hass.states.get(entity).state == "263.10"