60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
"""Test UptimeRobot sensor."""
|
|
|
|
from unittest.mock import patch
|
|
|
|
from pyuptimerobot import UptimeRobotAuthenticationException
|
|
|
|
from homeassistant.components.sensor import SensorDeviceClass
|
|
from homeassistant.components.uptimerobot.const import COORDINATOR_UPDATE_INTERVAL
|
|
from homeassistant.const import STATE_UNAVAILABLE
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.util import dt as dt_util
|
|
|
|
from .common import (
|
|
MOCK_UPTIMEROBOT_MONITOR,
|
|
STATE_UP,
|
|
UPTIMEROBOT_SENSOR_TEST_ENTITY,
|
|
setup_uptimerobot_integration,
|
|
)
|
|
|
|
from tests.common import async_fire_time_changed
|
|
|
|
SENSOR_ICON = "mdi:television-shimmer"
|
|
|
|
|
|
async def test_presentation(hass: HomeAssistant) -> None:
|
|
"""Test the presenstation of UptimeRobot sensors."""
|
|
await setup_uptimerobot_integration(hass)
|
|
|
|
entity = hass.states.get(UPTIMEROBOT_SENSOR_TEST_ENTITY)
|
|
|
|
assert entity.state == STATE_UP
|
|
assert entity.attributes["icon"] == SENSOR_ICON
|
|
assert entity.attributes["target"] == MOCK_UPTIMEROBOT_MONITOR["url"]
|
|
assert entity.attributes["device_class"] == SensorDeviceClass.ENUM
|
|
assert entity.attributes["options"] == [
|
|
"down",
|
|
"not_checked_yet",
|
|
"pause",
|
|
"seems_down",
|
|
"up",
|
|
]
|
|
|
|
|
|
async def test_unaviable_on_update_failure(hass: HomeAssistant) -> None:
|
|
"""Test entity unaviable on update failure."""
|
|
await setup_uptimerobot_integration(hass)
|
|
|
|
entity = hass.states.get(UPTIMEROBOT_SENSOR_TEST_ENTITY)
|
|
assert entity.state == STATE_UP
|
|
|
|
with patch(
|
|
"pyuptimerobot.UptimeRobot.async_get_monitors",
|
|
side_effect=UptimeRobotAuthenticationException,
|
|
):
|
|
async_fire_time_changed(hass, dt_util.utcnow() + COORDINATOR_UPDATE_INTERVAL)
|
|
await hass.async_block_till_done()
|
|
|
|
entity = hass.states.get(UPTIMEROBOT_SENSOR_TEST_ENTITY)
|
|
assert entity.state == STATE_UNAVAILABLE
|