38 lines
1.4 KiB
Python
38 lines
1.4 KiB
Python
"""The tests for Radarr sensor platform."""
|
|
from unittest.mock import AsyncMock
|
|
|
|
from homeassistant.components.sensor import SensorDeviceClass
|
|
from homeassistant.const import ATTR_DEVICE_CLASS, ATTR_UNIT_OF_MEASUREMENT
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from . import setup_integration
|
|
|
|
from tests.test_util.aiohttp import AiohttpClientMocker
|
|
|
|
|
|
async def test_sensors(
|
|
hass: HomeAssistant,
|
|
aioclient_mock: AiohttpClientMocker,
|
|
entity_registry_enabled_by_default: AsyncMock,
|
|
):
|
|
"""Test for successfully setting up the Radarr platform."""
|
|
await setup_integration(hass, aioclient_mock)
|
|
|
|
state = hass.states.get("sensor.mock_title_disk_space_downloads")
|
|
assert state.state == "263.10"
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == "GB"
|
|
state = hass.states.get("sensor.mock_title_movies")
|
|
assert state.state == "1"
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == "Movies"
|
|
state = hass.states.get("sensor.mock_title_start_time")
|
|
assert state.state == "2020-09-01T23:50:20+00:00"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.TIMESTAMP
|
|
|
|
|
|
async def test_windows(hass: HomeAssistant, aioclient_mock: AiohttpClientMocker):
|
|
"""Test for successfully setting up the Radarr platform on Windows."""
|
|
await setup_integration(hass, aioclient_mock, windows=True)
|
|
|
|
state = hass.states.get("sensor.mock_title_disk_space_tv")
|
|
assert state.state == "263.10"
|