Add support for system health to Airly integrarion (#43220)

pull/43866/head
Maciej Bieniek 2020-12-02 19:32:42 +01:00 committed by GitHub
parent 5fa00244da
commit 86043b2957
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 77 additions and 0 deletions

View File

@ -19,5 +19,10 @@
"abort": {
"already_configured": "[%key:common::config_flow::abort::already_configured_location%]"
}
},
"system_health": {
"info": {
"can_reach_server": "Reach Airly server"
}
}
}

View File

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

View File

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