SukramJ
8bbf261302
* Refactor home to hap for Homematic IP Cloud * Add some tests * Rename ha_entity --> ha_state * use asynctest.Mock
103 lines
3.0 KiB
Python
103 lines
3.0 KiB
Python
"""Initializer helpers for HomematicIP fake server."""
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from homematicip.aio.connection import AsyncConnection
|
|
import pytest
|
|
|
|
from homeassistant import config_entries
|
|
from homeassistant.components.homematicip_cloud import (
|
|
CONF_ACCESSPOINT,
|
|
CONF_AUTHTOKEN,
|
|
DOMAIN as HMIPC_DOMAIN,
|
|
async_setup as hmip_async_setup,
|
|
const as hmipc,
|
|
hap as hmip_hap,
|
|
)
|
|
from homeassistant.const import CONF_NAME
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from .helper import AUTH_TOKEN, HAPID, HomeTemplate
|
|
|
|
from tests.common import MockConfigEntry, mock_coro
|
|
|
|
|
|
@pytest.fixture(name="mock_connection")
|
|
def mock_connection_fixture():
|
|
"""Return a mocked connection."""
|
|
connection = MagicMock(spec=AsyncConnection)
|
|
|
|
def _rest_call_side_effect(path, body=None):
|
|
return path, body
|
|
|
|
connection._restCall.side_effect = _rest_call_side_effect # pylint: disable=W0212
|
|
connection.api_call.return_value = mock_coro(True)
|
|
|
|
return connection
|
|
|
|
|
|
@pytest.fixture(name="default_mock_home")
|
|
def default_mock_home_fixture(mock_connection):
|
|
"""Create a fake homematic async home."""
|
|
return HomeTemplate(connection=mock_connection).init_home().get_async_home_mock()
|
|
|
|
|
|
@pytest.fixture(name="hmip_config_entry")
|
|
def hmip_config_entry_fixture():
|
|
"""Create a mock config entriy for homematic ip cloud."""
|
|
entry_data = {
|
|
hmipc.HMIPC_HAPID: HAPID,
|
|
hmipc.HMIPC_AUTHTOKEN: AUTH_TOKEN,
|
|
hmipc.HMIPC_NAME: "",
|
|
}
|
|
config_entry = MockConfigEntry(
|
|
version=1,
|
|
domain=HMIPC_DOMAIN,
|
|
title=HAPID,
|
|
data=entry_data,
|
|
source="import",
|
|
connection_class=config_entries.CONN_CLASS_CLOUD_PUSH,
|
|
system_options={"disable_new_entities": False},
|
|
)
|
|
|
|
return config_entry
|
|
|
|
|
|
@pytest.fixture(name="default_mock_hap")
|
|
async def default_mock_hap_fixture(
|
|
hass: HomeAssistant, default_mock_home, hmip_config_entry
|
|
):
|
|
"""Create a fake homematic access point."""
|
|
hass.config.components.add(HMIPC_DOMAIN)
|
|
hap = hmip_hap.HomematicipHAP(hass, hmip_config_entry)
|
|
with patch.object(hap, "get_hap", return_value=mock_coro(default_mock_home)):
|
|
assert await hap.async_setup() is True
|
|
default_mock_home.on_update(hap.async_update)
|
|
default_mock_home.on_create(hap.async_create_entity)
|
|
|
|
hass.data[HMIPC_DOMAIN] = {HAPID: hap}
|
|
|
|
await hass.async_block_till_done()
|
|
|
|
return hap
|
|
|
|
|
|
@pytest.fixture(name="hmip_config")
|
|
def hmip_config_fixture():
|
|
"""Create a config for homematic ip cloud."""
|
|
|
|
entry_data = {CONF_ACCESSPOINT: HAPID, CONF_AUTHTOKEN: AUTH_TOKEN, CONF_NAME: ""}
|
|
|
|
return {hmipc.DOMAIN: [entry_data]}
|
|
|
|
|
|
@pytest.fixture(name="mock_hap_with_service")
|
|
async def mock_hap_with_service_fixture(
|
|
hass: HomeAssistant, default_mock_hap, hmip_config
|
|
):
|
|
"""Create a fake homematic access point with hass services."""
|
|
|
|
await hmip_async_setup(hass, hmip_config)
|
|
await hass.async_block_till_done()
|
|
hass.data[HMIPC_DOMAIN] = {HAPID: default_mock_hap}
|
|
return default_mock_hap
|