94 lines
3.2 KiB
Python
94 lines
3.2 KiB
Python
"""Test the Open Thread Border Router integration."""
|
|
|
|
from http import HTTPStatus
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components import otbr
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.exceptions import HomeAssistantError
|
|
|
|
from . import BASE_URL
|
|
|
|
from tests.test_util.aiohttp import AiohttpClientMocker
|
|
|
|
|
|
async def test_remove_entry(
|
|
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker, thread_config_entry
|
|
):
|
|
"""Test async_get_thread_state."""
|
|
|
|
aioclient_mock.get(f"{BASE_URL}/node/dataset/active", text="0E")
|
|
|
|
assert await otbr.async_get_active_dataset_tlvs(hass) == bytes.fromhex("0E")
|
|
|
|
config_entry = hass.config_entries.async_entries(otbr.DOMAIN)[0]
|
|
await hass.config_entries.async_remove(config_entry.entry_id)
|
|
|
|
with pytest.raises(HomeAssistantError):
|
|
assert await otbr.async_get_active_dataset_tlvs(hass)
|
|
|
|
|
|
async def test_get_active_dataset_tlvs(
|
|
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker, thread_config_entry
|
|
):
|
|
"""Test async_get_active_dataset_tlvs."""
|
|
|
|
mock_response = (
|
|
"0E080000000000010000000300001035060004001FFFE00208F642646DA209B1C00708FDF57B5A"
|
|
"0FE2AAF60510DE98B5BA1A528FEE049D4B4B01835375030D4F70656E5468726561642048410102"
|
|
"25A40410F5DD18371BFD29E1A601EF6FFAD94C030C0402A0F7F8"
|
|
)
|
|
|
|
aioclient_mock.get(f"{BASE_URL}/node/dataset/active", text=mock_response)
|
|
|
|
assert await otbr.async_get_active_dataset_tlvs(hass) == bytes.fromhex(
|
|
mock_response
|
|
)
|
|
|
|
|
|
async def test_get_active_dataset_tlvs_empty(
|
|
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker, thread_config_entry
|
|
):
|
|
"""Test async_get_active_dataset_tlvs."""
|
|
|
|
aioclient_mock.get(f"{BASE_URL}/node/dataset/active", status=HTTPStatus.NO_CONTENT)
|
|
assert await otbr.async_get_active_dataset_tlvs(hass) is None
|
|
|
|
|
|
async def test_get_active_dataset_tlvs_addon_not_installed(hass: HomeAssistant):
|
|
"""Test async_get_active_dataset_tlvs when the multi-PAN addon is not installed."""
|
|
|
|
with pytest.raises(HomeAssistantError):
|
|
await otbr.async_get_active_dataset_tlvs(hass)
|
|
|
|
|
|
async def test_get_active_dataset_tlvs_404(
|
|
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker, thread_config_entry
|
|
):
|
|
"""Test async_get_active_dataset_tlvs with error."""
|
|
|
|
aioclient_mock.get(f"{BASE_URL}/node/dataset/active", status=HTTPStatus.NOT_FOUND)
|
|
with pytest.raises(HomeAssistantError):
|
|
await otbr.async_get_active_dataset_tlvs(hass)
|
|
|
|
|
|
async def test_get_active_dataset_tlvs_201(
|
|
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker, thread_config_entry
|
|
):
|
|
"""Test async_get_active_dataset_tlvs with error."""
|
|
|
|
aioclient_mock.get(f"{BASE_URL}/node/dataset/active", status=HTTPStatus.CREATED)
|
|
with pytest.raises(HomeAssistantError):
|
|
assert await otbr.async_get_active_dataset_tlvs(hass) is None
|
|
|
|
|
|
async def test_get_active_dataset_tlvs_invalid(
|
|
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker, thread_config_entry
|
|
):
|
|
"""Test async_get_active_dataset_tlvs with error."""
|
|
|
|
aioclient_mock.get(f"{BASE_URL}/node/dataset/active", text="unexpected")
|
|
with pytest.raises(HomeAssistantError):
|
|
assert await otbr.async_get_active_dataset_tlvs(hass) is None
|