mirror of
https://github.com/home-assistant/core.git
synced 2026-07-03 20:56:06 +01:00
kohai-ut
117 lines
3.5 KiB
Python
117 lines
3.5 KiB
Python
"""Tests for the Envisalink setup."""
|
|
|
|
from collections.abc import Awaitable, Callable
|
|
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components.envisalink.alarm_control_panel import (
|
|
async_setup_platform as alarm_setup_platform,
|
|
)
|
|
from homeassistant.components.envisalink.binary_sensor import (
|
|
async_setup_platform as binary_sensor_setup_platform,
|
|
)
|
|
from homeassistant.components.envisalink.sensor import (
|
|
async_setup_platform as sensor_setup_platform,
|
|
)
|
|
from homeassistant.const import EVENT_HOMEASSISTANT_STOP
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
|
|
|
|
from .conftest import (
|
|
ALARM_ENTITY,
|
|
DOMAIN,
|
|
KEYPAD_ENTITY,
|
|
MOCK_CODE,
|
|
ZONE_ENTITY,
|
|
setup_envisalink,
|
|
)
|
|
|
|
SetupPlatform = Callable[
|
|
[HomeAssistant, ConfigType, AddEntitiesCallback, DiscoveryInfoType | None],
|
|
Awaitable[None],
|
|
]
|
|
|
|
|
|
async def test_setup_creates_entities(
|
|
hass: HomeAssistant, mock_controller: MagicMock
|
|
) -> None:
|
|
"""Test setup creates the alarm, keypad and zone entities."""
|
|
assert await setup_envisalink(hass)
|
|
|
|
assert hass.states.get(ALARM_ENTITY) is not None
|
|
assert hass.states.get(KEYPAD_ENTITY) is not None
|
|
assert hass.states.get(ZONE_ENTITY) is not None
|
|
|
|
|
|
async def test_setup_fails_on_login_failure(
|
|
hass: HomeAssistant, mock_controller: MagicMock
|
|
) -> None:
|
|
"""Test setup returns False when the Envisalink rejects the login."""
|
|
mock_controller.start.side_effect = lambda: mock_controller.callback_login_failure(
|
|
None
|
|
)
|
|
|
|
assert await setup_envisalink(hass) is False
|
|
assert hass.states.get(ALARM_ENTITY) is None
|
|
|
|
|
|
async def test_setup_succeeds_on_connection_timeout(
|
|
hass: HomeAssistant, mock_controller: MagicMock
|
|
) -> None:
|
|
"""Test setup proceeds (retry mode) when the first connection times out."""
|
|
mock_controller.start.side_effect = lambda: mock_controller.callback_login_timeout(
|
|
None
|
|
)
|
|
|
|
assert await setup_envisalink(hass)
|
|
assert hass.states.get(ALARM_ENTITY) is not None
|
|
|
|
|
|
async def test_controller_stopped_on_shutdown(
|
|
hass: HomeAssistant, mock_controller: MagicMock
|
|
) -> None:
|
|
"""Test the controller connection is stopped on Home Assistant shutdown."""
|
|
assert await setup_envisalink(hass)
|
|
|
|
hass.bus.async_fire(EVENT_HOMEASSISTANT_STOP)
|
|
await hass.async_block_till_done()
|
|
|
|
mock_controller.stop.assert_called_once()
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"setup_platform",
|
|
[
|
|
alarm_setup_platform,
|
|
binary_sensor_setup_platform,
|
|
sensor_setup_platform,
|
|
],
|
|
)
|
|
async def test_platform_setup_without_discovery_is_noop(
|
|
hass: HomeAssistant, setup_platform: SetupPlatform
|
|
) -> None:
|
|
"""Test a platform set up directly (no discovery info) adds no entities."""
|
|
add_entities = MagicMock()
|
|
await setup_platform(hass, {}, add_entities, None)
|
|
add_entities.assert_not_called()
|
|
|
|
|
|
async def test_invoke_custom_function_service(
|
|
hass: HomeAssistant, mock_controller: MagicMock
|
|
) -> None:
|
|
"""Test the PGM service forwards the code, partition and function."""
|
|
assert await setup_envisalink(hass)
|
|
|
|
# Distinct pgm/partition values so the assertion pins each positional arg
|
|
# of command_output(code, partition, custom_function).
|
|
await hass.services.async_call(
|
|
DOMAIN,
|
|
"invoke_custom_function",
|
|
{"pgm": "7", "partition": "2"},
|
|
blocking=True,
|
|
)
|
|
|
|
mock_controller.command_output.assert_called_once_with(MOCK_CODE, "2", "7")
|