97 lines
3.0 KiB
Python
97 lines
3.0 KiB
Python
"""Tests for AVM Fritz!Box."""
|
|
from unittest.mock import patch
|
|
|
|
from fritzconnection.core.exceptions import FritzSecurityError
|
|
import pytest
|
|
|
|
from homeassistant.components.device_tracker.const import (
|
|
CONF_CONSIDER_HOME,
|
|
DEFAULT_CONSIDER_HOME,
|
|
)
|
|
from homeassistant.components.fritz.const import DOMAIN, FRITZ_EXCEPTIONS
|
|
from homeassistant.config_entries import ConfigEntryState
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from .const import MOCK_USER_DATA
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
async def test_setup(hass: HomeAssistant, fc_class_mock, fh_class_mock):
|
|
"""Test setup and unload of Fritz!Tools."""
|
|
|
|
entry = MockConfigEntry(domain=DOMAIN, data=MOCK_USER_DATA)
|
|
entry.add_to_hass(hass)
|
|
|
|
assert await async_setup_component(hass, DOMAIN, {})
|
|
await hass.async_block_till_done()
|
|
assert entry.state == ConfigEntryState.LOADED
|
|
|
|
await hass.config_entries.async_unload(entry.entry_id)
|
|
assert entry.state == ConfigEntryState.NOT_LOADED
|
|
|
|
|
|
async def test_options_reload(hass: HomeAssistant, fc_class_mock, fh_class_mock):
|
|
"""Test reload of Fritz!Tools, when options changed."""
|
|
|
|
entry = MockConfigEntry(
|
|
domain=DOMAIN,
|
|
data=MOCK_USER_DATA,
|
|
options={CONF_CONSIDER_HOME: DEFAULT_CONSIDER_HOME.total_seconds()},
|
|
)
|
|
entry.add_to_hass(hass)
|
|
|
|
with patch(
|
|
"homeassistant.config_entries.ConfigEntries.async_reload",
|
|
return_value=None,
|
|
) as mock_reload:
|
|
assert await async_setup_component(hass, DOMAIN, {})
|
|
await hass.async_block_till_done()
|
|
assert entry.state == ConfigEntryState.LOADED
|
|
|
|
result = await hass.config_entries.options.async_init(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={CONF_CONSIDER_HOME: 60},
|
|
)
|
|
await hass.async_block_till_done()
|
|
mock_reload.assert_called_once()
|
|
|
|
|
|
async def test_setup_auth_fail(hass: HomeAssistant):
|
|
"""Test starting a flow by user with an already configured device."""
|
|
|
|
entry = MockConfigEntry(domain=DOMAIN, data=MOCK_USER_DATA)
|
|
entry.add_to_hass(hass)
|
|
|
|
with patch(
|
|
"homeassistant.components.fritz.common.FritzConnection",
|
|
side_effect=FritzSecurityError,
|
|
):
|
|
assert await async_setup_component(hass, DOMAIN, {})
|
|
await hass.async_block_till_done()
|
|
|
|
assert entry.state == ConfigEntryState.SETUP_ERROR
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"error",
|
|
FRITZ_EXCEPTIONS,
|
|
)
|
|
async def test_setup_fail(hass: HomeAssistant, error):
|
|
"""Test starting a flow by user with an already configured device."""
|
|
|
|
entry = MockConfigEntry(domain=DOMAIN, data=MOCK_USER_DATA)
|
|
entry.add_to_hass(hass)
|
|
|
|
with patch(
|
|
"homeassistant.components.fritz.common.FritzConnection",
|
|
side_effect=error,
|
|
):
|
|
assert await async_setup_component(hass, DOMAIN, {})
|
|
await hass.async_block_till_done()
|
|
|
|
assert entry.state == ConfigEntryState.SETUP_RETRY
|