mirror of https://github.com/home-assistant/core
69 lines
2.2 KiB
Python
69 lines
2.2 KiB
Python
"""Tests for the nexia integration."""
|
|
|
|
from unittest.mock import patch
|
|
import uuid
|
|
|
|
from nexia.home import NexiaHome
|
|
|
|
from homeassistant.components.nexia.const import DOMAIN
|
|
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from tests.common import MockConfigEntry, load_fixture
|
|
from tests.test_util.aiohttp import mock_aiohttp_client
|
|
|
|
|
|
async def async_init_integration(
|
|
hass: HomeAssistant,
|
|
skip_setup: bool = False,
|
|
exception: Exception | None = None,
|
|
) -> MockConfigEntry:
|
|
"""Set up the nexia integration in Home Assistant."""
|
|
|
|
house_fixture = "nexia/mobile_houses_123456.json"
|
|
session_fixture = "nexia/session_123456.json"
|
|
sign_in_fixture = "nexia/sign_in.json"
|
|
set_fan_speed_fixture = "nexia/set_fan_speed_2293892.json"
|
|
with (
|
|
mock_aiohttp_client() as mock_session,
|
|
patch("nexia.home.load_or_create_uuid", return_value=uuid.uuid4()),
|
|
):
|
|
nexia = NexiaHome(mock_session)
|
|
if exception:
|
|
|
|
async def _raise_exception(*args, **kwargs):
|
|
raise exception
|
|
|
|
mock_session.post(
|
|
nexia.API_MOBILE_SESSION_URL, side_effect=_raise_exception
|
|
)
|
|
else:
|
|
mock_session.post(
|
|
nexia.API_MOBILE_SESSION_URL, text=load_fixture(session_fixture)
|
|
)
|
|
mock_session.get(
|
|
nexia.API_MOBILE_HOUSES_URL.format(house_id=123456),
|
|
text=load_fixture(house_fixture),
|
|
)
|
|
mock_session.post(
|
|
nexia.API_MOBILE_ACCOUNTS_SIGN_IN_URL,
|
|
text=load_fixture(sign_in_fixture),
|
|
)
|
|
mock_session.post(
|
|
"https://www.mynexia.com/mobile/xxl_thermostats/2293892/fan_speed",
|
|
text=load_fixture(set_fan_speed_fixture),
|
|
)
|
|
entry = MockConfigEntry(
|
|
domain=DOMAIN,
|
|
data={CONF_USERNAME: "mock", CONF_PASSWORD: "mock"},
|
|
minor_version=2,
|
|
unique_id="123456",
|
|
)
|
|
entry.add_to_hass(hass)
|
|
|
|
if not skip_setup:
|
|
await hass.config_entries.async_setup(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
return entry
|