mirror of
https://github.com/home-assistant/core.git
synced 2025-05-02 13:17:53 +00:00
93d1961aae
* Initial commit * Revert "Initial commit" This reverts commit 452027f1a3c1be186cedd4115cea6928917c9467. * Change ezviz to token auth * Bump API version. * Add fix for token expired. Fix options update and unload. * Fix tests (PLATFORM to PLATFORM_BY_TYPE) * Uses and stores token only, added reauth step when token expires. * Add tests MFA code exceptions. * Fix tests. * Remove redundant try/except blocks. * Rebase fixes. * Fix errors in reauth config flow * Implement recommendations * Fix typing error in config_flow * Fix tests after rebase, readd camera check on init * Change to platform setup * Cleanup init. * Test for MFA required under user form * Remove useless if block. * Fix formating after rebase * Fix formating. * No longer stored in the repository --------- Co-authored-by: Paulus Schoutsen <paulus@home-assistant.io>
108 lines
2.4 KiB
Python
108 lines
2.4 KiB
Python
"""Tests for the EZVIZ integration."""
|
|
from unittest.mock import patch
|
|
|
|
from homeassistant.components.ezviz.const import (
|
|
ATTR_SERIAL,
|
|
ATTR_TYPE_CAMERA,
|
|
ATTR_TYPE_CLOUD,
|
|
CONF_FFMPEG_ARGUMENTS,
|
|
CONF_RFSESSION_ID,
|
|
CONF_SESSION_ID,
|
|
DEFAULT_FFMPEG_ARGUMENTS,
|
|
DEFAULT_TIMEOUT,
|
|
DOMAIN,
|
|
)
|
|
from homeassistant.const import (
|
|
CONF_IP_ADDRESS,
|
|
CONF_PASSWORD,
|
|
CONF_TIMEOUT,
|
|
CONF_TYPE,
|
|
CONF_URL,
|
|
CONF_USERNAME,
|
|
)
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
ENTRY_CONFIG = {
|
|
CONF_SESSION_ID: "test-username",
|
|
CONF_RFSESSION_ID: "test-password",
|
|
CONF_URL: "apiieu.ezvizlife.com",
|
|
CONF_TYPE: ATTR_TYPE_CLOUD,
|
|
}
|
|
|
|
ENTRY_OPTIONS = {
|
|
CONF_FFMPEG_ARGUMENTS: DEFAULT_FFMPEG_ARGUMENTS,
|
|
CONF_TIMEOUT: DEFAULT_TIMEOUT,
|
|
}
|
|
|
|
USER_INPUT_VALIDATE = {
|
|
CONF_USERNAME: "test-username",
|
|
CONF_PASSWORD: "test-password",
|
|
CONF_URL: "apiieu.ezvizlife.com",
|
|
}
|
|
|
|
USER_INPUT = {
|
|
CONF_USERNAME: "test-username",
|
|
CONF_PASSWORD: "test-password",
|
|
CONF_URL: "apiieu.ezvizlife.com",
|
|
CONF_TYPE: ATTR_TYPE_CLOUD,
|
|
}
|
|
|
|
USER_INPUT_CAMERA_VALIDATE = {
|
|
ATTR_SERIAL: "C666666",
|
|
CONF_PASSWORD: "test-password",
|
|
CONF_USERNAME: "test-username",
|
|
}
|
|
|
|
USER_INPUT_CAMERA = {
|
|
CONF_PASSWORD: "test-password",
|
|
CONF_USERNAME: "test-username",
|
|
CONF_TYPE: ATTR_TYPE_CAMERA,
|
|
}
|
|
|
|
DISCOVERY_INFO = {
|
|
ATTR_SERIAL: "C666666",
|
|
CONF_USERNAME: None,
|
|
CONF_PASSWORD: None,
|
|
CONF_IP_ADDRESS: "127.0.0.1",
|
|
}
|
|
|
|
TEST = {
|
|
CONF_USERNAME: None,
|
|
CONF_PASSWORD: None,
|
|
CONF_IP_ADDRESS: "127.0.0.1",
|
|
}
|
|
|
|
API_LOGIN_RETURN_VALIDATE = {
|
|
CONF_SESSION_ID: "fake_token",
|
|
CONF_RFSESSION_ID: "fake_rf_token",
|
|
CONF_URL: "apiieu.ezvizlife.com",
|
|
CONF_TYPE: ATTR_TYPE_CLOUD,
|
|
}
|
|
|
|
|
|
def _patch_async_setup_entry(return_value=True):
|
|
return patch(
|
|
"homeassistant.components.ezviz.async_setup_entry",
|
|
return_value=return_value,
|
|
)
|
|
|
|
|
|
async def init_integration(
|
|
hass: HomeAssistant,
|
|
*,
|
|
data: dict = ENTRY_CONFIG,
|
|
options: dict = ENTRY_OPTIONS,
|
|
skip_entry_setup: bool = False,
|
|
) -> MockConfigEntry:
|
|
"""Set up the EZVIZ integration in Home Assistant."""
|
|
entry = MockConfigEntry(domain=DOMAIN, data=data, options=options)
|
|
entry.add_to_hass(hass)
|
|
|
|
if not skip_entry_setup:
|
|
await hass.config_entries.async_setup(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
return entry
|