mirror of
https://github.com/home-assistant/core.git
synced 2026-08-06 13:26:29 +01:00
Josef Zweck
127 lines
4.0 KiB
Python
127 lines
4.0 KiB
Python
"""Tests for the Gatus integration setup and unload lifecycle."""
|
|
|
|
from unittest.mock import AsyncMock
|
|
|
|
from freezegun.api import FrozenDateTimeFactory
|
|
from gatus_api import GatusClientError
|
|
import pytest
|
|
|
|
from homeassistant.components.gatus.coordinator import GatusDataUpdateCoordinator
|
|
from homeassistant.config_entries import ConfigEntryState
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import device_registry as dr
|
|
|
|
from . import setup_integration
|
|
|
|
from tests.common import MockConfigEntry, async_fire_time_changed
|
|
|
|
|
|
@pytest.mark.usefixtures("mock_gatus_client")
|
|
async def test_setup_and_unload_entry(
|
|
hass: HomeAssistant, mock_config_entry: MockConfigEntry
|
|
) -> None:
|
|
"""Test standard successful setup and unload cycle of the integration."""
|
|
await setup_integration(hass, mock_config_entry)
|
|
|
|
assert mock_config_entry.state is ConfigEntryState.LOADED
|
|
assert mock_config_entry.runtime_data is not None
|
|
assert isinstance(mock_config_entry.runtime_data, GatusDataUpdateCoordinator)
|
|
|
|
assert await hass.config_entries.async_unload(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert mock_config_entry.state is ConfigEntryState.NOT_LOADED
|
|
|
|
|
|
async def test_setup_failure_retry(
|
|
hass: HomeAssistant,
|
|
mock_gatus_client: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Test that an API connection failure during initial setup places the entry in retry state."""
|
|
mock_gatus_client.get_endpoints_statuses.side_effect = GatusClientError(
|
|
"Cannot connect to Gatus API during initial setup"
|
|
)
|
|
|
|
await setup_integration(hass, mock_config_entry)
|
|
|
|
assert mock_config_entry.state is ConfigEntryState.SETUP_RETRY
|
|
|
|
|
|
@pytest.mark.usefixtures("mock_gatus_client")
|
|
async def test_remove_stale_device_runtime(
|
|
hass: HomeAssistant,
|
|
device_registry: dr.DeviceRegistry,
|
|
mock_config_entry: MockConfigEntry,
|
|
mock_gatus_client: AsyncMock,
|
|
freezer: FrozenDateTimeFactory,
|
|
) -> None:
|
|
"""Test that a device is removed at runtime when it is no longer returned by the Gatus API."""
|
|
await setup_integration(hass, mock_config_entry)
|
|
|
|
device = next(
|
|
(
|
|
dev
|
|
for dev in dr.async_entries_for_config_entry(
|
|
device_registry, mock_config_entry.entry_id
|
|
)
|
|
if ("gatus", f"{mock_config_entry.entry_id}_backend_service")
|
|
in dev.identifiers
|
|
),
|
|
None,
|
|
)
|
|
assert device is not None
|
|
|
|
mock_gatus_client.get_endpoints_statuses.return_value = []
|
|
|
|
freezer.tick(30)
|
|
async_fire_time_changed(hass)
|
|
await hass.async_block_till_done()
|
|
|
|
device = next(
|
|
(
|
|
dev
|
|
for dev in dr.async_entries_for_config_entry(
|
|
device_registry, mock_config_entry.entry_id
|
|
)
|
|
if ("gatus", f"{mock_config_entry.entry_id}_backend_service")
|
|
in dev.identifiers
|
|
),
|
|
None,
|
|
)
|
|
assert device is None
|
|
|
|
|
|
@pytest.mark.usefixtures("mock_gatus_client")
|
|
async def test_remove_stale_device_on_startup(
|
|
hass: HomeAssistant,
|
|
device_registry: dr.DeviceRegistry,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Test that stale devices in the registry are removed on startup."""
|
|
mock_config_entry.add_to_hass(hass)
|
|
|
|
stale_device = device_registry.async_get_or_create(
|
|
config_entry_id=mock_config_entry.entry_id,
|
|
identifiers={("gatus", f"{mock_config_entry.entry_id}_stale_service")},
|
|
name="Stale Service",
|
|
)
|
|
|
|
await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert device_registry.async_get(stale_device.id) is None
|
|
|
|
active_device = next(
|
|
(
|
|
dev
|
|
for dev in dr.async_entries_for_config_entry(
|
|
device_registry, mock_config_entry.entry_id
|
|
)
|
|
if ("gatus", f"{mock_config_entry.entry_id}_backend_service")
|
|
in dev.identifiers
|
|
),
|
|
None,
|
|
)
|
|
assert active_device is not None
|