hass-core/tests/components/apsystems/conftest.py
Simon Hörrle d246d02ab8
Add apsystems diagnostic binary sensors (#123045)
* add diagnostic sensors

* select output_data from data

* split sensor and binary_sensor configurations

* adjust module description

* convert values to bool

* add strings

* add tests

* add tests

* update translations

* remove already available _attr_has_entity_name

* use dataclass instead of TypedDict

* Update tests/components/apsystems/test_binary_sensor.py

---------

Co-authored-by: Joost Lekkerkerker <joostlek@outlook.com>
2024-08-05 11:08:27 +02:00

74 lines
2 KiB
Python

"""Common fixtures for the APsystems Local API tests."""
from collections.abc import Generator
from unittest.mock import AsyncMock, MagicMock, patch
from APsystemsEZ1 import ReturnAlarmInfo, ReturnDeviceInfo, ReturnOutputData, Status
import pytest
from homeassistant.components.apsystems.const import DOMAIN
from homeassistant.const import CONF_IP_ADDRESS
from tests.common import MockConfigEntry
@pytest.fixture
def mock_setup_entry() -> Generator[AsyncMock]:
"""Override async_setup_entry."""
with patch(
"homeassistant.components.apsystems.async_setup_entry",
return_value=True,
) as mock_setup_entry:
yield mock_setup_entry
@pytest.fixture
def mock_apsystems() -> Generator[MagicMock]:
"""Mock APSystems lib."""
with (
patch(
"homeassistant.components.apsystems.APsystemsEZ1M",
autospec=True,
) as mock_client,
patch(
"homeassistant.components.apsystems.config_flow.APsystemsEZ1M",
new=mock_client,
),
):
mock_api = mock_client.return_value
mock_api.get_device_info.return_value = ReturnDeviceInfo(
deviceId="MY_SERIAL_NUMBER",
devVer="1.0.0",
ssid="MY_SSID",
ipAddr="127.0.01",
minPower=0,
maxPower=1000,
)
mock_api.get_output_data.return_value = ReturnOutputData(
p1=2.0,
e1=3.0,
te1=4.0,
p2=5.0,
e2=6.0,
te2=7.0,
)
mock_api.get_alarm_info.return_value = ReturnAlarmInfo(
og=Status.normal,
isce1=Status.alarm,
isce2=Status.normal,
oe=Status.alarm,
)
mock_api.get_device_power_status.return_value = Status.normal
yield mock_api
@pytest.fixture
def mock_config_entry() -> MockConfigEntry:
"""Mock config entry."""
return MockConfigEntry(
domain=DOMAIN,
data={
CONF_IP_ADDRESS: "127.0.0.1",
},
unique_id="MY_SERIAL_NUMBER",
)