"""Test ISY system health."""
import asyncio
from unittest.mock import Mock

from aiohttp import ClientError

from homeassistant.components.isy994.const import DOMAIN, ISY_ROOT, ISY_URL_POSTFIX
from homeassistant.const import CONF_HOST
from homeassistant.setup import async_setup_component

from .test_config_flow import MOCK_HOSTNAME, MOCK_UUID

from tests.common import MockConfigEntry, get_system_health_info

MOCK_ENTRY_ID = "cad4af20b811990e757588519917d6af"
MOCK_CONNECTED = "connected"
MOCK_HEARTBEAT = "2021-05-01T00:00:00.000000"


async def test_system_health(hass, aioclient_mock):
    """Test system health."""
    aioclient_mock.get(f"http://{MOCK_HOSTNAME}{ISY_URL_POSTFIX}", text="")

    hass.config.components.add(DOMAIN)
    assert await async_setup_component(hass, "system_health", {})

    MockConfigEntry(
        domain=DOMAIN,
        entry_id=MOCK_ENTRY_ID,
        data={CONF_HOST: f"http://{MOCK_HOSTNAME}"},
        unique_id=MOCK_UUID,
    ).add_to_hass(hass)

    hass.data[DOMAIN] = {}
    hass.data[DOMAIN][MOCK_ENTRY_ID] = {}
    hass.data[DOMAIN][MOCK_ENTRY_ID][ISY_ROOT] = Mock(
        connected=True,
        websocket=Mock(
            last_heartbeat=MOCK_HEARTBEAT,
            status=MOCK_CONNECTED,
        ),
    )

    info = await get_system_health_info(hass, DOMAIN)

    for key, val in info.items():
        if asyncio.iscoroutine(val):
            info[key] = await val

    assert info["host_reachable"] == "ok"
    assert info["device_connected"]
    assert info["last_heartbeat"] == MOCK_HEARTBEAT
    assert info["websocket_status"] == MOCK_CONNECTED


async def test_system_health_failed_connect(hass, aioclient_mock):
    """Test system health."""
    aioclient_mock.get(f"http://{MOCK_HOSTNAME}{ISY_URL_POSTFIX}", exc=ClientError)

    hass.config.components.add(DOMAIN)
    assert await async_setup_component(hass, "system_health", {})

    MockConfigEntry(
        domain=DOMAIN,
        entry_id=MOCK_ENTRY_ID,
        data={CONF_HOST: f"http://{MOCK_HOSTNAME}"},
        unique_id=MOCK_UUID,
    ).add_to_hass(hass)

    hass.data[DOMAIN] = {}
    hass.data[DOMAIN][MOCK_ENTRY_ID] = {}
    hass.data[DOMAIN][MOCK_ENTRY_ID][ISY_ROOT] = Mock(
        connected=True,
        websocket=Mock(
            last_heartbeat=MOCK_HEARTBEAT,
            status=MOCK_CONNECTED,
        ),
    )

    info = await get_system_health_info(hass, DOMAIN)

    for key, val in info.items():
        if asyncio.iscoroutine(val):
            info[key] = await val

    assert info["host_reachable"] == {"error": "unreachable", "type": "failed"}