Add support for system health to Airly integrarion (#43220)
parent
5fa00244da
commit
86043b2957
|
@ -19,5 +19,10 @@
|
|||
"abort": {
|
||||
"already_configured": "[%key:common::config_flow::abort::already_configured_location%]"
|
||||
}
|
||||
},
|
||||
"system_health": {
|
||||
"info": {
|
||||
"can_reach_server": "Reach Airly server"
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
@ -0,0 +1,22 @@
|
|||
"""Provide info to system health."""
|
||||
from airly import Airly
|
||||
|
||||
from homeassistant.components import system_health
|
||||
from homeassistant.core import HomeAssistant, callback
|
||||
|
||||
|
||||
@callback
|
||||
def async_register(
|
||||
hass: HomeAssistant, register: system_health.SystemHealthRegistration
|
||||
) -> None:
|
||||
"""Register system health callbacks."""
|
||||
register.async_register_info(system_health_info)
|
||||
|
||||
|
||||
async def system_health_info(hass):
|
||||
"""Get info for the info page."""
|
||||
return {
|
||||
"can_reach_server": system_health.async_check_can_reach_url(
|
||||
hass, Airly.AIRLY_API_URL
|
||||
)
|
||||
}
|
|
@ -0,0 +1,50 @@
|
|||
"""Test Airly system health."""
|
||||
import asyncio
|
||||
|
||||
from aiohttp import ClientError
|
||||
|
||||
from homeassistant.components.airly.const import DOMAIN
|
||||
from homeassistant.setup import async_setup_component
|
||||
|
||||
from tests.async_mock import Mock
|
||||
from tests.common import get_system_health_info
|
||||
|
||||
|
||||
async def test_airly_system_health(hass, aioclient_mock):
|
||||
"""Test Airly system health."""
|
||||
aioclient_mock.get("https://airapi.airly.eu/v2/", text="")
|
||||
hass.config.components.add(DOMAIN)
|
||||
assert await async_setup_component(hass, "system_health", {})
|
||||
|
||||
hass.data[DOMAIN] = {}
|
||||
hass.data[DOMAIN]["0123xyz"] = Mock(
|
||||
airly=Mock(AIRLY_API_URL="https://airapi.airly.eu/v2/")
|
||||
)
|
||||
|
||||
info = await get_system_health_info(hass, DOMAIN)
|
||||
|
||||
for key, val in info.items():
|
||||
if asyncio.iscoroutine(val):
|
||||
info[key] = await val
|
||||
|
||||
assert info == {"can_reach_server": "ok"}
|
||||
|
||||
|
||||
async def test_airly_system_health_fail(hass, aioclient_mock):
|
||||
"""Test Airly system health."""
|
||||
aioclient_mock.get("https://airapi.airly.eu/v2/", exc=ClientError)
|
||||
hass.config.components.add(DOMAIN)
|
||||
assert await async_setup_component(hass, "system_health", {})
|
||||
|
||||
hass.data[DOMAIN] = {}
|
||||
hass.data[DOMAIN]["0123xyz"] = Mock(
|
||||
airly=Mock(AIRLY_API_URL="https://airapi.airly.eu/v2/")
|
||||
)
|
||||
|
||||
info = await get_system_health_info(hass, DOMAIN)
|
||||
|
||||
for key, val in info.items():
|
||||
if asyncio.iscoroutine(val):
|
||||
info[key] = await val
|
||||
|
||||
assert info == {"can_reach_server": {"type": "failed", "error": "unreachable"}}
|
Loading…
Reference in New Issue