123 lines
3.8 KiB
Python
123 lines
3.8 KiB
Python
"""Tests for the Synology DSM component."""
|
|
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from synology_dsm.exceptions import SynologyDSMLoginInvalidException
|
|
|
|
from homeassistant.components.synology_dsm.const import (
|
|
CONF_BACKUP_PATH,
|
|
CONF_BACKUP_SHARE,
|
|
DEFAULT_VERIFY_SSL,
|
|
DOMAIN,
|
|
SERVICES,
|
|
)
|
|
from homeassistant.const import (
|
|
CONF_HOST,
|
|
CONF_MAC,
|
|
CONF_PASSWORD,
|
|
CONF_PORT,
|
|
CONF_SSL,
|
|
CONF_USERNAME,
|
|
CONF_VERIFY_SSL,
|
|
)
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.data_entry_flow import FlowResultType
|
|
|
|
from .consts import HOST, MACS, PASSWORD, PORT, USE_SSL, USERNAME
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
async def test_services_registered(hass: HomeAssistant, mock_dsm: MagicMock) -> None:
|
|
"""Test if all services are registered."""
|
|
with (
|
|
patch(
|
|
"homeassistant.components.synology_dsm.common.SynologyDSM",
|
|
return_value=mock_dsm,
|
|
),
|
|
patch("homeassistant.components.synology_dsm.PLATFORMS", return_value=[]),
|
|
):
|
|
entry = MockConfigEntry(
|
|
domain=DOMAIN,
|
|
data={
|
|
CONF_HOST: HOST,
|
|
CONF_PORT: PORT,
|
|
CONF_SSL: USE_SSL,
|
|
CONF_USERNAME: USERNAME,
|
|
CONF_PASSWORD: PASSWORD,
|
|
CONF_MAC: MACS[0],
|
|
},
|
|
)
|
|
entry.add_to_hass(hass)
|
|
assert await hass.config_entries.async_setup(entry.entry_id)
|
|
for service in SERVICES:
|
|
assert hass.services.has_service(DOMAIN, service)
|
|
|
|
|
|
async def test_reauth_triggered(hass: HomeAssistant) -> None:
|
|
"""Test if reauthentication flow is triggered."""
|
|
with (
|
|
patch(
|
|
"homeassistant.components.synology_dsm.SynoApi.async_setup",
|
|
side_effect=SynologyDSMLoginInvalidException(USERNAME),
|
|
),
|
|
patch(
|
|
"homeassistant.components.synology_dsm.config_flow.SynologyDSMFlowHandler.async_step_reauth",
|
|
return_value={
|
|
"type": FlowResultType.FORM,
|
|
"flow_id": "mock_flow",
|
|
"step_id": "reauth_confirm",
|
|
},
|
|
) as mock_async_step_reauth,
|
|
):
|
|
entry = MockConfigEntry(
|
|
domain=DOMAIN,
|
|
data={
|
|
CONF_HOST: HOST,
|
|
CONF_PORT: PORT,
|
|
CONF_SSL: USE_SSL,
|
|
CONF_USERNAME: USERNAME,
|
|
CONF_PASSWORD: PASSWORD,
|
|
CONF_MAC: MACS[0],
|
|
},
|
|
)
|
|
entry.add_to_hass(hass)
|
|
assert not await hass.config_entries.async_setup(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
mock_async_step_reauth.assert_called_once()
|
|
|
|
|
|
async def test_config_entry_migrations(
|
|
hass: HomeAssistant, mock_dsm: MagicMock
|
|
) -> None:
|
|
"""Test if reauthentication flow is triggered."""
|
|
with (
|
|
patch(
|
|
"homeassistant.components.synology_dsm.common.SynologyDSM",
|
|
return_value=mock_dsm,
|
|
),
|
|
patch("homeassistant.components.synology_dsm.PLATFORMS", return_value=[]),
|
|
):
|
|
entry = MockConfigEntry(
|
|
domain=DOMAIN,
|
|
data={
|
|
CONF_HOST: HOST,
|
|
CONF_PORT: PORT,
|
|
CONF_SSL: USE_SSL,
|
|
CONF_USERNAME: USERNAME,
|
|
CONF_PASSWORD: PASSWORD,
|
|
CONF_MAC: MACS[0],
|
|
},
|
|
)
|
|
entry.add_to_hass(hass)
|
|
|
|
assert CONF_VERIFY_SSL not in entry.data
|
|
assert CONF_BACKUP_SHARE not in entry.options
|
|
assert CONF_BACKUP_PATH not in entry.options
|
|
|
|
assert await hass.config_entries.async_setup(entry.entry_id)
|
|
|
|
assert entry.data[CONF_VERIFY_SSL] == DEFAULT_VERIFY_SSL
|
|
assert entry.options[CONF_BACKUP_SHARE] is None
|
|
assert entry.options[CONF_BACKUP_PATH] is None
|