317 lines
9.7 KiB
Python
317 lines
9.7 KiB
Python
"""Setup mocks for the Plugwise integration tests."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from collections.abc import Generator
|
|
import json
|
|
from typing import Any
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
from packaging.version import Version
|
|
import pytest
|
|
|
|
from homeassistant.components.plugwise.const import DOMAIN
|
|
from homeassistant.const import (
|
|
CONF_HOST,
|
|
CONF_MAC,
|
|
CONF_PASSWORD,
|
|
CONF_PORT,
|
|
CONF_USERNAME,
|
|
)
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from tests.common import MockConfigEntry, load_fixture
|
|
|
|
|
|
def _read_json(environment: str, call: str) -> dict[str, Any]:
|
|
"""Undecode the json data."""
|
|
fixture = load_fixture(f"plugwise/{environment}/{call}.json")
|
|
return json.loads(fixture)
|
|
|
|
|
|
@pytest.fixture
|
|
def cooling_present(request: pytest.FixtureRequest) -> str:
|
|
"""Pass the cooling_present boolean.
|
|
|
|
Used with fixtures that require parametrization of the cooling capability.
|
|
"""
|
|
return request.param
|
|
|
|
|
|
@pytest.fixture
|
|
def chosen_env(request: pytest.FixtureRequest) -> str:
|
|
"""Pass the chosen_env string.
|
|
|
|
Used with fixtures that require parametrization of the user-data fixture.
|
|
"""
|
|
return request.param
|
|
|
|
|
|
@pytest.fixture
|
|
def gateway_id(request: pytest.FixtureRequest) -> str:
|
|
"""Pass the gateway_id string.
|
|
|
|
Used with fixtures that require parametrization of the gateway_id.
|
|
"""
|
|
return request.param
|
|
|
|
|
|
@pytest.fixture
|
|
def heater_id(request: pytest.FixtureRequest) -> str:
|
|
"""Pass the heater_idstring.
|
|
|
|
Used with fixtures that require parametrization of the heater_id.
|
|
"""
|
|
return request.param
|
|
|
|
|
|
@pytest.fixture
|
|
def reboot(request: pytest.FixtureRequest) -> str:
|
|
"""Pass the reboot boolean.
|
|
|
|
Used with fixtures that require parametrization of the reboot capability.
|
|
"""
|
|
return request.param
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_config_entry() -> MockConfigEntry:
|
|
"""Return the default mocked config entry."""
|
|
return MockConfigEntry(
|
|
title="My Plugwise",
|
|
domain=DOMAIN,
|
|
data={
|
|
CONF_HOST: "127.0.0.1",
|
|
CONF_MAC: "AA:BB:CC:DD:EE:FF",
|
|
CONF_PASSWORD: "test-password",
|
|
CONF_PORT: 80,
|
|
CONF_USERNAME: "smile",
|
|
},
|
|
unique_id="smile98765",
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_setup_entry() -> Generator[AsyncMock]:
|
|
"""Mock setting up a config entry."""
|
|
with patch(
|
|
"homeassistant.components.plugwise.async_setup_entry", return_value=True
|
|
) as mock_setup:
|
|
yield mock_setup
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_smile_config_flow() -> Generator[MagicMock]:
|
|
"""Return a mocked Smile client."""
|
|
with patch(
|
|
"homeassistant.components.plugwise.config_flow.Smile",
|
|
autospec=True,
|
|
) as smile_mock:
|
|
smile = smile_mock.return_value
|
|
|
|
smile.connect.return_value = Version("4.3.2")
|
|
smile.smile_hostname = "smile12345"
|
|
smile.smile_model = "Test Model"
|
|
smile.smile_model_id = "Test Model ID"
|
|
smile.smile_name = "Test Smile Name"
|
|
smile.smile_version = "4.3.2"
|
|
|
|
yield smile
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_smile_adam() -> Generator[MagicMock]:
|
|
"""Create a Mock Adam environment for testing exceptions."""
|
|
chosen_env = "m_adam_multiple_devices_per_zone"
|
|
data = _read_json(chosen_env, "data")
|
|
with (
|
|
patch(
|
|
"homeassistant.components.plugwise.coordinator.Smile", autospec=True
|
|
) as smile_mock,
|
|
patch(
|
|
"homeassistant.components.plugwise.config_flow.Smile",
|
|
new=smile_mock,
|
|
),
|
|
):
|
|
smile = smile_mock.return_value
|
|
|
|
smile.async_update.return_value = data
|
|
smile.cooling_present = False
|
|
smile.connect.return_value = Version("3.0.15")
|
|
smile.gateway_id = "fe799307f1624099878210aa0b9f1475"
|
|
smile.heater_id = "90986d591dcd426cae3ec3e8111ff730"
|
|
smile.reboot = True
|
|
smile.smile_hostname = "smile98765"
|
|
smile.smile_model = "Gateway"
|
|
smile.smile_model_id = "smile_open_therm"
|
|
smile.smile_name = "Adam"
|
|
smile.smile_type = "thermostat"
|
|
smile.smile_version = "3.0.15"
|
|
|
|
yield smile
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_smile_adam_heat_cool(
|
|
chosen_env: str, cooling_present: bool
|
|
) -> Generator[MagicMock]:
|
|
"""Create a special base Mock Adam type for testing with different datasets."""
|
|
data = _read_json(chosen_env, "data")
|
|
with patch(
|
|
"homeassistant.components.plugwise.coordinator.Smile", autospec=True
|
|
) as smile_mock:
|
|
smile = smile_mock.return_value
|
|
|
|
smile.async_update.return_value = data
|
|
smile.connect.return_value = Version("3.6.4")
|
|
smile.cooling_present = cooling_present
|
|
smile.gateway_id = "da224107914542988a88561b4452b0f6"
|
|
smile.heater_id = "056ee145a816487eaa69243c3280f8bf"
|
|
smile.reboot = True
|
|
smile.smile_hostname = "smile98765"
|
|
smile.smile_model = "Gateway"
|
|
smile.smile_model_id = "smile_open_therm"
|
|
smile.smile_name = "Adam"
|
|
smile.smile_type = "thermostat"
|
|
smile.smile_version = "3.6.4"
|
|
|
|
yield smile
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_smile_adam_jip() -> Generator[MagicMock]:
|
|
"""Create a Mock adam-jip type for testing exceptions."""
|
|
chosen_env = "m_adam_jip"
|
|
data = _read_json(chosen_env, "data")
|
|
with patch(
|
|
"homeassistant.components.plugwise.coordinator.Smile", autospec=True
|
|
) as smile_mock:
|
|
smile = smile_mock.return_value
|
|
|
|
smile.async_update.return_value = data
|
|
smile.connect.return_value = Version("3.2.8")
|
|
smile.cooling_present = False
|
|
smile.gateway_id = "b5c2386c6f6342669e50fe49dd05b188"
|
|
smile.heater_id = "e4684553153b44afbef2200885f379dc"
|
|
smile.reboot = True
|
|
smile.smile_hostname = "smile98765"
|
|
smile.smile_model = "Gateway"
|
|
smile.smile_model_id = "smile_open_therm"
|
|
smile.smile_name = "Adam"
|
|
smile.smile_type = "thermostat"
|
|
smile.smile_version = "3.2.8"
|
|
|
|
yield smile
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_smile_anna(chosen_env: str, cooling_present: bool) -> Generator[MagicMock]:
|
|
"""Create a Mock Anna type for testing."""
|
|
data = _read_json(chosen_env, "data")
|
|
with patch(
|
|
"homeassistant.components.plugwise.coordinator.Smile", autospec=True
|
|
) as smile_mock:
|
|
smile = smile_mock.return_value
|
|
|
|
smile.async_update.return_value = data
|
|
smile.connect.return_value = Version("4.0.15")
|
|
smile.cooling_present = cooling_present
|
|
smile.gateway_id = "015ae9ea3f964e668e490fa39da3870b"
|
|
smile.heater_id = "1cbf783bb11e4a7c8a6843dee3a86927"
|
|
smile.reboot = True
|
|
smile.smile_hostname = "smile98765"
|
|
smile.smile_model = "Gateway"
|
|
smile.smile_model_id = "smile_thermo"
|
|
smile.smile_name = "Smile Anna"
|
|
smile.smile_type = "thermostat"
|
|
smile.smile_version = "4.0.15"
|
|
|
|
yield smile
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_smile_p1(chosen_env: str, gateway_id: str) -> Generator[MagicMock]:
|
|
"""Create a base Mock P1 type for testing with different datasets and gateway-ids."""
|
|
data = _read_json(chosen_env, "data")
|
|
with patch(
|
|
"homeassistant.components.plugwise.coordinator.Smile", autospec=True
|
|
) as smile_mock:
|
|
smile = smile_mock.return_value
|
|
|
|
smile.async_update.return_value = data
|
|
smile.connect.return_value = Version("4.4.2")
|
|
smile.gateway_id = gateway_id
|
|
smile.heater_id = None
|
|
smile.reboot = True
|
|
smile.smile_hostname = "smile98765"
|
|
smile.smile_model = "Gateway"
|
|
smile.smile_model_id = "smile"
|
|
smile.smile_name = "Smile P1"
|
|
smile.smile_type = "power"
|
|
smile.smile_version = "4.4.2"
|
|
|
|
yield smile
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_smile_legacy_anna() -> Generator[MagicMock]:
|
|
"""Create a Mock legacy Anna environment for testing exceptions."""
|
|
chosen_env = "legacy_anna"
|
|
data = _read_json(chosen_env, "data")
|
|
with patch(
|
|
"homeassistant.components.plugwise.coordinator.Smile", autospec=True
|
|
) as smile_mock:
|
|
smile = smile_mock.return_value
|
|
|
|
smile.async_update.return_value = data
|
|
smile.connect.return_value = Version("1.8.22")
|
|
smile.gateway_id = "0000aaaa0000aaaa0000aaaa0000aa00"
|
|
smile.heater_id = "04e4cbfe7f4340f090f85ec3b9e6a950"
|
|
smile.reboot = False
|
|
smile.smile_hostname = "smile98765"
|
|
smile.smile_model = "Gateway"
|
|
smile.smile_model_id = None
|
|
smile.smile_name = "Smile Anna"
|
|
smile.smile_type = "thermostat"
|
|
smile.smile_version = "1.8.22"
|
|
|
|
yield smile
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_stretch() -> Generator[MagicMock]:
|
|
"""Create a Mock Stretch environment for testing exceptions."""
|
|
chosen_env = "stretch_v31"
|
|
data = _read_json(chosen_env, "data")
|
|
with patch(
|
|
"homeassistant.components.plugwise.coordinator.Smile", autospec=True
|
|
) as smile_mock:
|
|
smile = smile_mock.return_value
|
|
|
|
smile.async_update.return_value = data
|
|
smile.connect.return_value = Version("3.1.11")
|
|
smile.gateway_id = "259882df3c05415b99c2d962534ce820"
|
|
smile.heater_id = None
|
|
smile.reboot = False
|
|
smile.smile_hostname = "stretch98765"
|
|
smile.smile_model = "Gateway"
|
|
smile.smile_model_id = None
|
|
smile.smile_name = "Stretch"
|
|
smile.smile_type = "stretch"
|
|
smile.smile_version = "3.1.11"
|
|
|
|
yield smile
|
|
|
|
|
|
@pytest.fixture
|
|
async def init_integration(
|
|
hass: HomeAssistant, mock_config_entry: MockConfigEntry
|
|
) -> MockConfigEntry:
|
|
"""Set up the Plugwise integration for testing."""
|
|
mock_config_entry.add_to_hass(hass)
|
|
|
|
await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
return mock_config_entry
|