ece5587e1f
* Rename device.py to hub.py * Rename AxisNetworkDevice to AxisHub * Move hub.py into hub package * Rename get_axis_device to get_axis_api * Split out get_axis_api to its own module * Rename device object to hub * Rename device to api in config flow * Convenience method to get hub
52 lines
1.9 KiB
Python
52 lines
1.9 KiB
Python
"""Test Axis component setup process."""
|
|
from unittest.mock import AsyncMock, Mock, patch
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components import axis
|
|
from homeassistant.config_entries import ConfigEntryState
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
|
|
async def test_setup_entry(hass: HomeAssistant, setup_config_entry) -> None:
|
|
"""Test successful setup of entry."""
|
|
assert setup_config_entry.state == ConfigEntryState.LOADED
|
|
|
|
|
|
async def test_setup_entry_fails(hass: HomeAssistant, config_entry) -> None:
|
|
"""Test successful setup of entry."""
|
|
mock_device = Mock()
|
|
mock_device.async_setup = AsyncMock(return_value=False)
|
|
|
|
with patch.object(axis, "AxisHub") as mock_device_class:
|
|
mock_device_class.return_value = mock_device
|
|
|
|
assert not await hass.config_entries.async_setup(config_entry.entry_id)
|
|
|
|
assert config_entry.state == ConfigEntryState.SETUP_ERROR
|
|
|
|
|
|
async def test_unload_entry(hass: HomeAssistant, setup_config_entry) -> None:
|
|
"""Test successful unload of entry."""
|
|
assert setup_config_entry.state == ConfigEntryState.LOADED
|
|
|
|
assert await hass.config_entries.async_unload(setup_config_entry.entry_id)
|
|
assert setup_config_entry.state == ConfigEntryState.NOT_LOADED
|
|
|
|
|
|
@pytest.mark.parametrize("config_entry_version", [1])
|
|
async def test_migrate_entry(hass: HomeAssistant, config_entry) -> None:
|
|
"""Test successful migration of entry data."""
|
|
assert config_entry.version == 1
|
|
|
|
mock_device = Mock()
|
|
mock_device.async_setup = AsyncMock()
|
|
mock_device.async_update_device_registry = AsyncMock()
|
|
mock_device.api.vapix.light_control = None
|
|
mock_device.api.vapix.params.image_format = None
|
|
|
|
with patch("homeassistant.components.axis.async_setup_entry", return_value=True):
|
|
assert await hass.config_entries.async_setup(config_entry.entry_id)
|
|
|
|
assert config_entry.state == ConfigEntryState.LOADED
|
|
assert config_entry.version == 3
|