118 lines
3.6 KiB
Python
118 lines
3.6 KiB
Python
"""Test OTBR Utility functions."""
|
|
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
import python_otbr_api
|
|
|
|
from homeassistant.components import otbr
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.exceptions import HomeAssistantError
|
|
|
|
OTBR_MULTIPAN_URL = "http://core-silabs-multiprotocol:8081"
|
|
OTBR_NON_MULTIPAN_URL = "/dev/ttyAMA1"
|
|
|
|
|
|
async def test_get_allowed_channel(
|
|
hass: HomeAssistant, multiprotocol_addon_manager_mock
|
|
) -> None:
|
|
"""Test get_allowed_channel."""
|
|
|
|
# OTBR multipan + No configured channel -> no restriction
|
|
multiprotocol_addon_manager_mock.async_get_channel.return_value = None
|
|
assert await otbr.util.get_allowed_channel(hass, OTBR_MULTIPAN_URL) is None
|
|
|
|
# OTBR multipan + multipan using channel 15 -> 15
|
|
multiprotocol_addon_manager_mock.async_get_channel.return_value = 15
|
|
assert await otbr.util.get_allowed_channel(hass, OTBR_MULTIPAN_URL) == 15
|
|
|
|
# OTBR no multipan + multipan using channel 15 -> no restriction
|
|
multiprotocol_addon_manager_mock.async_get_channel.return_value = 15
|
|
assert await otbr.util.get_allowed_channel(hass, OTBR_NON_MULTIPAN_URL) is None
|
|
|
|
|
|
async def test_factory_reset(hass: HomeAssistant, otbr_config_entry_multipan) -> None:
|
|
"""Test factory_reset."""
|
|
data: otbr.OTBRData = hass.data[otbr.DOMAIN]
|
|
|
|
with (
|
|
patch("python_otbr_api.OTBR.factory_reset") as factory_reset_mock,
|
|
patch(
|
|
"python_otbr_api.OTBR.delete_active_dataset"
|
|
) as delete_active_dataset_mock,
|
|
):
|
|
await data.factory_reset()
|
|
|
|
delete_active_dataset_mock.assert_not_called()
|
|
factory_reset_mock.assert_called_once_with()
|
|
|
|
|
|
async def test_factory_reset_not_supported(
|
|
hass: HomeAssistant, otbr_config_entry_multipan
|
|
) -> None:
|
|
"""Test factory_reset."""
|
|
data: otbr.OTBRData = hass.data[otbr.DOMAIN]
|
|
|
|
with (
|
|
patch(
|
|
"python_otbr_api.OTBR.factory_reset",
|
|
side_effect=python_otbr_api.FactoryResetNotSupportedError,
|
|
) as factory_reset_mock,
|
|
patch(
|
|
"python_otbr_api.OTBR.delete_active_dataset"
|
|
) as delete_active_dataset_mock,
|
|
):
|
|
await data.factory_reset()
|
|
|
|
delete_active_dataset_mock.assert_called_once_with()
|
|
factory_reset_mock.assert_called_once_with()
|
|
|
|
|
|
async def test_factory_reset_error_1(
|
|
hass: HomeAssistant, otbr_config_entry_multipan
|
|
) -> None:
|
|
"""Test factory_reset."""
|
|
data: otbr.OTBRData = hass.data[otbr.DOMAIN]
|
|
|
|
with (
|
|
patch(
|
|
"python_otbr_api.OTBR.factory_reset",
|
|
side_effect=python_otbr_api.OTBRError,
|
|
) as factory_reset_mock,
|
|
patch(
|
|
"python_otbr_api.OTBR.delete_active_dataset"
|
|
) as delete_active_dataset_mock,
|
|
pytest.raises(
|
|
HomeAssistantError,
|
|
),
|
|
):
|
|
await data.factory_reset()
|
|
|
|
delete_active_dataset_mock.assert_not_called()
|
|
factory_reset_mock.assert_called_once_with()
|
|
|
|
|
|
async def test_factory_reset_error_2(
|
|
hass: HomeAssistant, otbr_config_entry_multipan
|
|
) -> None:
|
|
"""Test factory_reset."""
|
|
data: otbr.OTBRData = hass.data[otbr.DOMAIN]
|
|
|
|
with (
|
|
patch(
|
|
"python_otbr_api.OTBR.factory_reset",
|
|
side_effect=python_otbr_api.FactoryResetNotSupportedError,
|
|
) as factory_reset_mock,
|
|
patch(
|
|
"python_otbr_api.OTBR.delete_active_dataset",
|
|
side_effect=python_otbr_api.OTBRError,
|
|
) as delete_active_dataset_mock,
|
|
pytest.raises(
|
|
HomeAssistantError,
|
|
),
|
|
):
|
|
await data.factory_reset()
|
|
|
|
delete_active_dataset_mock.assert_called_once_with()
|
|
factory_reset_mock.assert_called_once_with()
|