mirror of
https://github.com/home-assistant/core.git
synced 2026-02-04 22:35:58 +01:00
146 lines
4.6 KiB
Python
146 lines
4.6 KiB
Python
"""Test the Blink init."""
|
|
|
|
from unittest.mock import AsyncMock, MagicMock
|
|
|
|
from aiohttp import ClientError
|
|
from blinkpy.auth import LoginError
|
|
import pytest
|
|
|
|
from homeassistant.components.blink.const import DOMAIN
|
|
from homeassistant.components.blink.services import SERVICE_SAVE_VIDEO
|
|
from homeassistant.config_entries import ConfigEntryState
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
CAMERA_NAME = "Camera 1"
|
|
FILENAME = "blah"
|
|
PIN = "1234"
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("the_error", "available"),
|
|
[(ClientError, False), (TimeoutError, False), (None, False)],
|
|
)
|
|
async def test_setup_not_ready(
|
|
hass: HomeAssistant,
|
|
mock_blink_api: MagicMock,
|
|
mock_blink_auth_api: MagicMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
the_error,
|
|
available,
|
|
) -> None:
|
|
"""Test setup failed because we can't connect to the Blink system."""
|
|
|
|
mock_blink_api.start = AsyncMock(side_effect=the_error)
|
|
mock_blink_api.available = available
|
|
|
|
mock_config_entry.add_to_hass(hass)
|
|
assert not await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
assert mock_config_entry.state is ConfigEntryState.SETUP_RETRY
|
|
|
|
|
|
async def test_setup_not_ready_authkey_required(
|
|
hass: HomeAssistant,
|
|
mock_blink_api: MagicMock,
|
|
mock_blink_auth_api: MagicMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Test setup failed because 2FA is needed to connect to the Blink system."""
|
|
|
|
mock_blink_api.start = AsyncMock(side_effect=LoginError)
|
|
|
|
mock_config_entry.add_to_hass(hass)
|
|
|
|
assert not await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert mock_config_entry.state is ConfigEntryState.SETUP_ERROR
|
|
|
|
|
|
async def test_unload_entry(
|
|
hass: HomeAssistant,
|
|
mock_blink_api: MagicMock,
|
|
mock_blink_auth_api: MagicMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Test unload doesn't un-register services."""
|
|
|
|
mock_config_entry.add_to_hass(hass)
|
|
assert await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
assert mock_config_entry.state is ConfigEntryState.LOADED
|
|
assert await hass.config_entries.async_unload(mock_config_entry.entry_id)
|
|
assert mock_config_entry.state is ConfigEntryState.NOT_LOADED
|
|
assert hass.services.has_service(DOMAIN, SERVICE_SAVE_VIDEO)
|
|
|
|
|
|
async def test_migrate_V0(
|
|
hass: HomeAssistant,
|
|
mock_blink_api: MagicMock,
|
|
mock_blink_auth_api: MagicMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Test migration script version 0."""
|
|
mock_config_entry.add_to_hass(hass)
|
|
|
|
hass.config_entries.async_update_entry(mock_config_entry, version=0)
|
|
|
|
assert await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
entry = hass.config_entries.async_get_entry(mock_config_entry.entry_id)
|
|
assert entry.state is ConfigEntryState.LOADED
|
|
|
|
|
|
@pytest.mark.parametrize(("version"), [1, 2])
|
|
async def test_migrate(
|
|
hass: HomeAssistant,
|
|
mock_blink_api: MagicMock,
|
|
mock_blink_auth_api: MagicMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
version,
|
|
) -> None:
|
|
"""Test migration scripts."""
|
|
mock_config_entry.add_to_hass(hass)
|
|
|
|
hass.config_entries.async_update_entry(
|
|
mock_config_entry,
|
|
version=version,
|
|
data={**mock_config_entry.data, "login_response": "Blah"},
|
|
)
|
|
|
|
assert not await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
entry = hass.config_entries.async_get_entry(mock_config_entry.entry_id)
|
|
assert entry.state is ConfigEntryState.MIGRATION_ERROR
|
|
|
|
|
|
async def test_migrate_v3_to_v4(
|
|
hass: HomeAssistant,
|
|
mock_blink_api: MagicMock,
|
|
mock_blink_auth_api: MagicMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Test migration from version 3 to 4 (device_id to hardware_id)."""
|
|
mock_config_entry.add_to_hass(hass)
|
|
|
|
# Set up v3 config entry with device_id
|
|
data = {**mock_config_entry.data}
|
|
data.pop("hardware_id", None)
|
|
data["device_id"] = "Home Assistant"
|
|
hass.config_entries.async_update_entry(
|
|
mock_config_entry,
|
|
version=3,
|
|
data=data,
|
|
)
|
|
|
|
assert await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
entry = hass.config_entries.async_get_entry(mock_config_entry.entry_id)
|
|
assert entry.state is ConfigEntryState.LOADED
|
|
assert entry.version == 4
|
|
assert "hardware_id" in entry.data
|
|
assert "device_id" not in entry.data
|
|
assert entry.data["hardware_id"] == "Home Assistant"
|