74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
"""Test Axis component setup process."""
|
|
from unittest.mock import AsyncMock, Mock, patch
|
|
|
|
from homeassistant.components import axis
|
|
from homeassistant.components.axis.const import DOMAIN as AXIS_DOMAIN
|
|
from homeassistant.const import CONF_MAC
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from .test_device import setup_axis_integration
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
async def test_setup_no_config(hass):
|
|
"""Test setup without configuration."""
|
|
assert await async_setup_component(hass, AXIS_DOMAIN, {})
|
|
assert AXIS_DOMAIN not in hass.data
|
|
|
|
|
|
async def test_setup_entry(hass):
|
|
"""Test successful setup of entry."""
|
|
config_entry = await setup_axis_integration(hass)
|
|
assert len(hass.data[AXIS_DOMAIN]) == 1
|
|
assert config_entry.entry_id in hass.data[AXIS_DOMAIN]
|
|
|
|
|
|
async def test_setup_entry_fails(hass):
|
|
"""Test successful setup of entry."""
|
|
config_entry = MockConfigEntry(domain=AXIS_DOMAIN, data={CONF_MAC: "0123"})
|
|
config_entry.add_to_hass(hass)
|
|
|
|
mock_device = Mock()
|
|
mock_device.async_setup = AsyncMock(return_value=False)
|
|
|
|
with patch.object(axis, "AxisNetworkDevice") as mock_device_class:
|
|
mock_device_class.return_value = mock_device
|
|
|
|
assert not await hass.config_entries.async_setup(config_entry.entry_id)
|
|
|
|
assert not hass.data[AXIS_DOMAIN]
|
|
|
|
|
|
async def test_unload_entry(hass):
|
|
"""Test successful unload of entry."""
|
|
config_entry = await setup_axis_integration(hass)
|
|
assert hass.data[AXIS_DOMAIN]
|
|
|
|
assert await hass.config_entries.async_unload(config_entry.entry_id)
|
|
assert not hass.data[AXIS_DOMAIN]
|
|
|
|
|
|
async def test_migrate_entry(hass):
|
|
"""Test successful migration of entry data."""
|
|
config_entry = MockConfigEntry(domain=AXIS_DOMAIN, version=1)
|
|
config_entry.add_to_hass(hass)
|
|
|
|
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.object(axis, "get_axis_device"), patch.object(
|
|
axis, "AxisNetworkDevice"
|
|
) as mock_device_class:
|
|
mock_device_class.return_value = mock_device
|
|
|
|
assert await hass.config_entries.async_setup(config_entry.entry_id)
|
|
|
|
assert hass.data[AXIS_DOMAIN]
|
|
assert config_entry.version == 3
|