91 lines
3.2 KiB
Python
91 lines
3.2 KiB
Python
"""Test the bang_olufsen __init__."""
|
|
|
|
from aiohttp.client_exceptions import ServerTimeoutError
|
|
|
|
from homeassistant.components.bang_olufsen import DOMAIN
|
|
from homeassistant.config_entries import ConfigEntryState
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.device_registry import DeviceRegistry
|
|
|
|
from .const import TEST_MODEL_BALANCE, TEST_NAME, TEST_SERIAL_NUMBER
|
|
|
|
|
|
async def test_setup_entry(
|
|
hass: HomeAssistant,
|
|
mock_config_entry,
|
|
mock_mozart_client,
|
|
device_registry: DeviceRegistry,
|
|
) -> None:
|
|
"""Test async_setup_entry."""
|
|
|
|
assert mock_config_entry.state == ConfigEntryState.NOT_LOADED
|
|
|
|
# Load entry
|
|
mock_config_entry.add_to_hass(hass)
|
|
await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
|
|
assert mock_config_entry.state == ConfigEntryState.LOADED
|
|
|
|
# Check that the device has been registered properly
|
|
device = device_registry.async_get_device(
|
|
identifiers={(DOMAIN, TEST_SERIAL_NUMBER)}
|
|
)
|
|
assert device is not None
|
|
assert device.name == TEST_NAME
|
|
assert device.model == TEST_MODEL_BALANCE
|
|
|
|
# Ensure that the connection has been checked WebSocket connection has been initialized
|
|
assert mock_mozart_client.check_device_connection.call_count == 1
|
|
assert mock_mozart_client.close_api_client.call_count == 0
|
|
assert mock_mozart_client.connect_notifications.call_count == 1
|
|
|
|
|
|
async def test_setup_entry_failed(
|
|
hass: HomeAssistant, mock_config_entry, mock_mozart_client
|
|
) -> None:
|
|
"""Test failed async_setup_entry."""
|
|
|
|
# Set the device connection check to fail
|
|
mock_mozart_client.check_device_connection.side_effect = ExceptionGroup(
|
|
"", (ServerTimeoutError(), TimeoutError())
|
|
)
|
|
|
|
assert mock_config_entry.state == ConfigEntryState.NOT_LOADED
|
|
|
|
# Load entry
|
|
mock_config_entry.add_to_hass(hass)
|
|
await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
|
|
assert mock_config_entry.state == ConfigEntryState.SETUP_RETRY
|
|
|
|
# Ensure that the connection has been checked, API client correctly closed
|
|
# and WebSocket connection has not been initialized
|
|
assert mock_mozart_client.check_device_connection.call_count == 1
|
|
assert mock_mozart_client.close_api_client.call_count == 1
|
|
assert mock_mozart_client.connect_notifications.call_count == 0
|
|
|
|
|
|
async def test_unload_entry(
|
|
hass: HomeAssistant, mock_config_entry, mock_mozart_client
|
|
) -> None:
|
|
"""Test unload_entry."""
|
|
|
|
# Load entry
|
|
assert mock_config_entry.state == ConfigEntryState.NOT_LOADED
|
|
|
|
mock_config_entry.add_to_hass(hass)
|
|
await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
|
|
assert mock_config_entry.state == ConfigEntryState.LOADED
|
|
|
|
# Unload entry
|
|
await hass.config_entries.async_unload(mock_config_entry.entry_id)
|
|
|
|
# Ensure WebSocket notification listener and REST API client have been closed
|
|
assert mock_mozart_client.disconnect_notifications.call_count == 1
|
|
assert mock_mozart_client.close_api_client.call_count == 1
|
|
|
|
# Ensure that the entry is not loaded and has been removed from hass
|
|
assert mock_config_entry.entry_id not in hass.data[DOMAIN]
|
|
assert mock_config_entry.state == ConfigEntryState.NOT_LOADED
|