mirror of
https://github.com/home-assistant/core
synced 2024-10-05 09:12:11 +00:00
108 lines
3.6 KiB
Python
108 lines
3.6 KiB
Python
"""Test ZHA Silicon Labs Multiprotocol support."""
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
from unittest.mock import call, patch
|
|
|
|
import pytest
|
|
import zigpy.backups
|
|
import zigpy.state
|
|
|
|
from homeassistant.components import zha
|
|
from homeassistant.components.zha import silabs_multiprotocol
|
|
from homeassistant.components.zha.core.helpers import get_zha_gateway
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
if TYPE_CHECKING:
|
|
from zigpy.application import ControllerApplication
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def required_platform_only():
|
|
"""Only set up the required and required base platforms to speed up tests."""
|
|
with patch("homeassistant.components.zha.PLATFORMS", ()):
|
|
yield
|
|
|
|
|
|
async def test_async_get_channel_active(hass: HomeAssistant, setup_zha) -> None:
|
|
"""Test reading channel with an active ZHA installation."""
|
|
await setup_zha()
|
|
|
|
assert await silabs_multiprotocol.async_get_channel(hass) == 15
|
|
|
|
|
|
async def test_async_get_channel_missing(
|
|
hass: HomeAssistant, setup_zha, zigpy_app_controller: ControllerApplication
|
|
) -> None:
|
|
"""Test reading channel with an inactive ZHA installation, no valid channel."""
|
|
await setup_zha()
|
|
|
|
gateway = get_zha_gateway(hass)
|
|
await zha.async_unload_entry(hass, gateway.config_entry)
|
|
|
|
# Network settings were never loaded for whatever reason
|
|
zigpy_app_controller.state.network_info = zigpy.state.NetworkInfo()
|
|
zigpy_app_controller.state.node_info = zigpy.state.NodeInfo()
|
|
|
|
assert await silabs_multiprotocol.async_get_channel(hass) is None
|
|
|
|
|
|
async def test_async_get_channel_no_zha(hass: HomeAssistant) -> None:
|
|
"""Test reading channel with no ZHA config entries and no database."""
|
|
assert await silabs_multiprotocol.async_get_channel(hass) is None
|
|
|
|
|
|
async def test_async_using_multipan_active(hass: HomeAssistant, setup_zha) -> None:
|
|
"""Test async_using_multipan with an active ZHA installation."""
|
|
await setup_zha()
|
|
|
|
assert await silabs_multiprotocol.async_using_multipan(hass) is False
|
|
|
|
|
|
async def test_async_using_multipan_no_zha(hass: HomeAssistant) -> None:
|
|
"""Test async_using_multipan with no ZHA config entries and no database."""
|
|
assert await silabs_multiprotocol.async_using_multipan(hass) is False
|
|
|
|
|
|
async def test_change_channel(
|
|
hass: HomeAssistant, setup_zha, zigpy_app_controller: ControllerApplication
|
|
) -> None:
|
|
"""Test changing the channel."""
|
|
await setup_zha()
|
|
|
|
task = await silabs_multiprotocol.async_change_channel(hass, 20)
|
|
await task
|
|
|
|
assert zigpy_app_controller.move_network_to_channel.mock_calls == [call(20)]
|
|
|
|
|
|
async def test_change_channel_no_zha(
|
|
hass: HomeAssistant, zigpy_app_controller: ControllerApplication
|
|
) -> None:
|
|
"""Test changing the channel with no ZHA config entries and no database."""
|
|
task = await silabs_multiprotocol.async_change_channel(hass, 20)
|
|
assert task is None
|
|
|
|
assert zigpy_app_controller.mock_calls == []
|
|
|
|
|
|
@pytest.mark.parametrize(("delay", "sleep"), [(0, 0), (5, 0), (15, 15 - 10.27)])
|
|
async def test_change_channel_delay(
|
|
hass: HomeAssistant,
|
|
setup_zha,
|
|
zigpy_app_controller: ControllerApplication,
|
|
delay: float,
|
|
sleep: float,
|
|
) -> None:
|
|
"""Test changing the channel with a delay."""
|
|
await setup_zha()
|
|
|
|
with patch(
|
|
"homeassistant.components.zha.silabs_multiprotocol.asyncio.sleep", autospec=True
|
|
) as mock_sleep:
|
|
task = await silabs_multiprotocol.async_change_channel(hass, 20, delay=delay)
|
|
await task
|
|
|
|
assert zigpy_app_controller.move_network_to_channel.mock_calls == [call(20)]
|
|
assert mock_sleep.mock_calls == [call(sleep)]
|