e08661dad3
* Bump pyAussieBB * rolling back to previous version * patching the pydantic 2.x issue in aussie_broadband integration * adding test for validate_service_type * adding test for validate_service_type * fixing tests, again * adding additional test * doing fixes for live tests * Implement Feedback * Add test to detect pydantic2 * Update test_init.py * Update docstring --------- Co-authored-by: James Hodgkinson <james@terminaloutcomes.com>
63 lines
2.3 KiB
Python
63 lines
2.3 KiB
Python
"""Test the Aussie Broadband init."""
|
|
from unittest.mock import patch
|
|
|
|
from aiohttp import ClientConnectionError
|
|
from aussiebb.exceptions import AuthenticationException, UnrecognisedServiceType
|
|
import pydantic
|
|
import pytest
|
|
|
|
from homeassistant import data_entry_flow
|
|
from homeassistant.components.aussie_broadband import validate_service_type
|
|
from homeassistant.config_entries import ConfigEntryState
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from .common import setup_platform
|
|
|
|
|
|
async def test_unload(hass: HomeAssistant) -> None:
|
|
"""Test unload."""
|
|
entry = await setup_platform(hass)
|
|
assert await hass.config_entries.async_unload(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
assert entry.state is ConfigEntryState.NOT_LOADED
|
|
|
|
|
|
async def test_validate_service_type() -> None:
|
|
"""Testing the validation function."""
|
|
test_service = {"type": "Hardware", "name": "test service"}
|
|
validate_service_type(test_service)
|
|
|
|
with pytest.raises(ValueError):
|
|
test_service = {"name": "test service"}
|
|
validate_service_type(test_service)
|
|
with pytest.raises(UnrecognisedServiceType):
|
|
test_service = {"type": "FunkyBob", "name": "test service"}
|
|
validate_service_type(test_service)
|
|
|
|
|
|
async def test_auth_failure(hass: HomeAssistant) -> None:
|
|
"""Test init with an authentication failure."""
|
|
with patch(
|
|
"homeassistant.components.aussie_broadband.config_flow.ConfigFlow.async_step_reauth",
|
|
return_value={"type": data_entry_flow.FlowResultType.FORM},
|
|
) as mock_async_step_reauth:
|
|
await setup_platform(hass, side_effect=AuthenticationException())
|
|
mock_async_step_reauth.assert_called_once()
|
|
|
|
|
|
async def test_net_failure(hass: HomeAssistant) -> None:
|
|
"""Test init with a network failure."""
|
|
entry = await setup_platform(hass, side_effect=ClientConnectionError())
|
|
assert entry.state is ConfigEntryState.SETUP_RETRY
|
|
|
|
|
|
async def test_service_failure(hass: HomeAssistant) -> None:
|
|
"""Test init with a invalid service."""
|
|
entry = await setup_platform(hass, usage_effect=UnrecognisedServiceType())
|
|
assert entry.state is ConfigEntryState.SETUP_RETRY
|
|
|
|
|
|
async def test_not_pydantic2() -> None:
|
|
"""Test that Home Assistant still does not support Pydantic 2."""
|
|
"""For PR#99077 and validate_service_type backport"""
|
|
assert pydantic.__version__ < "2"
|