starred/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2026-08-29 02:35:05 -05:00
Code Issues Packages Projects Releases Wiki Activity
Files
rc
core/tests/components/neopool/conftest.py
T

280 lines
8.5 KiB
Python
Raw Permalink Blame History

"""Common fixtures for the NeoPool tests."""
from collections.abc import Generator
from typing import Any
from unittest.mock import AsyncMock, MagicMock, patch
import pytest
from homeassistant.components.neopool.const import (
CONF_USE_AUX1,
CONF_USE_AUX2,
CONF_USE_AUX3,
CONF_USE_AUX4,
CONF_USE_COVER_SENSOR,
CONF_USE_LIGHT,
CURRENT_VERSION,
DEFAULT_PORT,
DEFAULT_UNIT_ID,
DOMAIN,
)
from homeassistant.const import CONF_HOST, CONF_NAME, CONF_PORT
from tests.common import MockConfigEntry
MOCK_HOST = "192.0.2.1"
MOCK_PORT = DEFAULT_PORT
MOCK_NAME = "Pool"
MOCK_SERIAL = "1234567890"
MOCK_POOL_DATA: dict[str, Any] = {
"MBF_POWER_MODULE_VERSION": 0x1234,
"MBF_PAR_VERSION": 0x100,
"MBF_PAR_MODEL": 0x0003,
"MBF_PAR_SERNUM": int(MOCK_SERIAL),
"MBF_PAR_FILTRATION_CONF": 1,
"MBF_PAR_FILT_GPIO": 1,
"MBF_PAR_LIGHTING_GPIO": 2,
"MBF_PAR_HEATING_GPIO": 3,
"MBF_PAR_PH_ACID_RELAY_GPIO": 4,
"MBF_PAR_PH_BASE_RELAY_GPIO": 5,
"MBF_PAR_RX_RELAY_GPIO": 6,
"MBF_PAR_CL_RELAY_GPIO": 7,
"MBF_PAR_CD_RELAY_GPIO": 0,
"MBF_PAR_UV_RELAY_GPIO": 1,
"MBF_PAR_FILTVALVE_GPIO": 1,
"MBF_PAR_FILTVALVE_ENABLE": 1,
"MBF_PAR_TEMPERATURE_ACTIVE": 1,
"MBF_PAR_UICFG_MACHINE": 0,
"MBF_PAR_RELAY_PH": 0,
"Hydrolysis module detected": True,
"Redox measurement module detected": True,
"pH measurement module detected": True,
"Chlorine measurement module detected": True,
"Conductivity measurement module detected": True,
"Ionization module detected": True,
"MBF_PAR_FILT_MODE": 0,
"filtration_mode": "manual",
"filtration_speed_state": "off",
"MBF_MEASURE_TEMPERATURE": 250,
"MBF_MEASURE_PH": 720,
"MBF_MEASURE_RX": 650,
"MBF_MEASURE_CL": 120,
"MBF_MEASURE_CONDUCTIVITY": 45,
"MBF_HIDRO_CURRENT": 70,
"MBF_HIDRO_VOLTAGE": 24,
"MBF_ION_CURRENT": 50,
"MBF_PAR_INTELLIGENT_INTERVALS": 4,
"MBF_PAR_INTELLIGENT_TT_NEXT_INTERVAL": 7200,
"MBF_PAR_FILTVALVE_REMAINING": 0,
"MBF_PAR_FILTVALVE_INTERVAL": 150,
"MBF_PAR_FILTVALVE_MODE": 0,
"HIDRO_POLARITY": 0,
"ION_POLARITY": 0,
"PH_PUMP_STATUS": "off",
"HIDRO in Pol1": False,
"HIDRO in Pol2": False,
"HIDRO in dead time": False,
"ION in Pol1": False,
"ION in Pol2": False,
"ION in dead time": False,
"pH control module": True,
"pH pump active": False,
"pH acid pump active": False,
"Filtration Pump": False,
"MBF_PAR_HIDRO_COVER_REDUCTION": 0x0C19,
"MBF_PAR_HIDRO_COVER_ENABLE": 0x0000,
"Pool Cover": 0,
"CELL_RUNTIME_TOTAL": 0x00010000,
"CELL_RUNTIME_PART": 0x00000E10,
"CELL_RUNTIME_POLA": 0x00000708,
"CELL_RUNTIME_POLB": 0x00000708,
"CELL_RUNTIME_POL_CHANGES": 0x00000007,
}
# Aux timer blocks default to manual mode (enable=4) so switch writes pass the
# guard; auto-mode tests override the returned mode per case.
def _timer_block(enable: int = 4) -> dict[str, Any]:
"""Return a timer block dict for the mock read_all_timers."""
return {
"enable": enable,
"on": 0,
"interval": 0,
"period": 0,
"countdown": 0,
"stop": None,
}
MOCK_TIMER_BLOCKS: dict[str, dict[str, Any]] = {
"relay_aux1": _timer_block(),
"relay_aux2": _timer_block(),
"relay_aux3": _timer_block(),
"relay_aux4": _timer_block(),
}
def _read_all_timers(
enabled_timers: list[str] | None = None, **_kwargs: Any
) -> dict[str, dict[str, Any]]:
"""Return the requested timer blocks, mirroring the library contract."""
if enabled_timers is None:
return {name: dict(block) for name, block in MOCK_TIMER_BLOCKS.items()}
return {
name: dict(MOCK_TIMER_BLOCKS[name])
for name in enabled_timers
if name in MOCK_TIMER_BLOCKS
}
@pytest.fixture
def mock_setup_entry() -> Generator[AsyncMock]:
"""Override async_setup_entry."""
with patch(
"homeassistant.components.neopool.async_setup_entry", return_value=True
) as mock_setup_entry:
yield mock_setup_entry
@pytest.fixture
def mock_config_entry() -> MockConfigEntry:
"""Return a config entry with a bare-serial unique_id."""
return MockConfigEntry(
domain=DOMAIN,
title=MOCK_NAME,
unique_id=MOCK_SERIAL,
version=CURRENT_VERSION,
data={
CONF_HOST: MOCK_HOST,
CONF_PORT: MOCK_PORT,
CONF_NAME: MOCK_NAME,
"unit_id": DEFAULT_UNIT_ID,
"modbus_framer": "tcp",
},
)
@pytest.fixture
def mock_config_entry_light() -> MockConfigEntry:
"""Return a config entry with the pool light option enabled."""
return MockConfigEntry(
domain=DOMAIN,
title=MOCK_NAME,
unique_id=MOCK_SERIAL,
version=CURRENT_VERSION,
data={
CONF_HOST: MOCK_HOST,
CONF_PORT: MOCK_PORT,
CONF_NAME: MOCK_NAME,
"unit_id": DEFAULT_UNIT_ID,
"modbus_framer": "tcp",
},
options={CONF_USE_LIGHT: True},
)
@pytest.fixture
def mock_config_entry_switch() -> MockConfigEntry:
"""Return a config entry with the option-gated switches enabled."""
return MockConfigEntry(
domain=DOMAIN,
title=MOCK_NAME,
unique_id=MOCK_SERIAL,
version=CURRENT_VERSION,
data={
CONF_HOST: MOCK_HOST,
CONF_PORT: MOCK_PORT,
CONF_NAME: MOCK_NAME,
"unit_id": DEFAULT_UNIT_ID,
"modbus_framer": "tcp",
},
options={
CONF_USE_COVER_SENSOR: True,
CONF_USE_AUX1: True,
CONF_USE_AUX2: True,
CONF_USE_AUX3: True,
CONF_USE_AUX4: True,
},
)
@pytest.fixture
def mock_neopool_client() -> Generator[MagicMock]:
"""Patch the NeoPoolModbusClient and return a configurable mock instance."""
with (
patch(
"homeassistant.components.neopool.NeoPoolModbusClient",
autospec=True,
) as mock_client_cls,
patch(
"homeassistant.components.neopool.config_flow.async_probe_serial",
new=AsyncMock(return_value=MOCK_SERIAL),
),
):
mock_client = mock_client_cls.return_value
mock_client.async_read_all = AsyncMock(return_value=dict(MOCK_POOL_DATA))
mock_client.read_all_timers = AsyncMock(side_effect=_read_all_timers)
mock_client.connection_stats = {
"host": MOCK_HOST,
"port": MOCK_PORT,
"unit_id": DEFAULT_UNIT_ID,
"connected": True,
"total_operations": 10,
"successful_operations": 10,
"success_rate_percent": 100.0,
}
mock_client.async_set_relay_state = AsyncMock(return_value={})
mock_client.async_set_manual_filtration = AsyncMock(return_value={})
mock_client.async_set_binary_flag = AsyncMock(return_value={})
mock_client.async_set_bitmask_flag = AsyncMock(return_value={})
mock_client.async_start_backwash = AsyncMock(return_value=None)
mock_client.async_stop_backwash = AsyncMock(return_value=None)
mock_client.close = AsyncMock()
yield mock_client
@pytest.fixture
def minimal_pool_data() -> dict[str, Any]:
"""Pool data with all optional capability flags off.
Used to drive the 'should-skip' branches for supported_fn gating.
"""
return {
"MBF_POWER_MODULE_VERSION": 0x1234,
"MBF_PAR_VERSION": 0x100,
"MBF_PAR_MODEL": 0,
"MBF_PAR_SERNUM": int(MOCK_SERIAL),
"MBF_PAR_FILTRATION_CONF": 0,
"MBF_PAR_FILT_GPIO": 0,
"MBF_PAR_LIGHTING_GPIO": 0,
"MBF_PAR_HEATING_GPIO": 0,
"MBF_PAR_PH_ACID_RELAY_GPIO": 0,
"MBF_PAR_PH_BASE_RELAY_GPIO": 0,
"MBF_PAR_RX_RELAY_GPIO": 0,
"MBF_PAR_CL_RELAY_GPIO": 0,
"MBF_PAR_CD_RELAY_GPIO": 0,
"MBF_PAR_UV_RELAY_GPIO": 0,
"MBF_PAR_FILTVALVE_GPIO": 0,
"MBF_PAR_FILTVALVE_ENABLE": 0,
"MBF_PAR_TEMPERATURE_ACTIVE": 0,
"Hydrolysis module detected": False,
"Redox measurement module detected": False,
"pH measurement module detected": False,
"MBF_PAR_FILT_MODE": 0,
"filtration_mode": "manual",
"filtration_speed_state": "off",
"Filtration Pump": False,
}
@pytest.fixture
def mock_socket_connection() -> Generator[AsyncMock]:
"""Patch the lib probe in config_flow so we don't hit the network."""
with patch(
"homeassistant.components.neopool.config_flow.async_probe_serial",
new=AsyncMock(return_value=MOCK_SERIAL),
) as mock:
yield mock
Reference in New Issue View Git Blame Copy Permalink