mirror of
https://github.com/home-assistant/core.git
synced 2026-08-22 06:11:36 +01:00
73 lines
2.3 KiB
Python
73 lines
2.3 KiB
Python
"""Tests for the Electra Smart integration setup."""
|
|
|
|
from unittest.mock import AsyncMock, Mock, patch
|
|
|
|
from electrasmart.device import OperationMode
|
|
import pytest
|
|
from syrupy.assertion import SnapshotAssertion
|
|
|
|
from homeassistant.components.electrasmart.const import (
|
|
CONF_IMEI,
|
|
CONF_PHONE_NUMBER,
|
|
DOMAIN,
|
|
)
|
|
from homeassistant.const import CONF_TOKEN
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import device_registry as dr
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
@pytest.fixture(name="mock_device")
|
|
def mock_device_fixture() -> Mock:
|
|
"""Return a mocked Electra AC device."""
|
|
device = Mock(
|
|
mac="a8032ab12345",
|
|
model="Electra A/C",
|
|
manufactor="Electra",
|
|
features=[],
|
|
is_disconnected=Mock(return_value=False),
|
|
is_on=Mock(return_value=False),
|
|
is_horizontal_swing=Mock(return_value=False),
|
|
is_vertical_swing=Mock(return_value=False),
|
|
get_fan_speed=Mock(return_value=OperationMode.FAN_SPEED_AUTO),
|
|
get_mode=Mock(return_value=OperationMode.MODE_COOL),
|
|
get_sensor_temperature=Mock(return_value=24),
|
|
get_temperature=Mock(return_value=22),
|
|
get_shabat_mode=Mock(return_value=False),
|
|
)
|
|
# `name` is a reserved Mock kwarg, so it must be set after construction.
|
|
device.name = "Living Room"
|
|
return device
|
|
|
|
|
|
async def test_device_registry(
|
|
hass: HomeAssistant,
|
|
device_registry: dr.DeviceRegistry,
|
|
mock_device: Mock,
|
|
snapshot: SnapshotAssertion,
|
|
) -> None:
|
|
"""Test the device registry entry, including the network MAC connection."""
|
|
entry = MockConfigEntry(
|
|
domain=DOMAIN,
|
|
unique_id="0521234567",
|
|
data={
|
|
CONF_TOKEN: "token",
|
|
CONF_IMEI: "2b950000024051000000000000000000",
|
|
CONF_PHONE_NUMBER: "0521234567",
|
|
},
|
|
)
|
|
entry.add_to_hass(hass)
|
|
|
|
mock_api = Mock(devices=[mock_device], fetch_devices=AsyncMock())
|
|
with patch(
|
|
"homeassistant.components.electrasmart.ElectraAPI", return_value=mock_api
|
|
):
|
|
assert await hass.config_entries.async_setup(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
device_entry = device_registry.async_get_device_by_identifier(
|
|
(DOMAIN, "a8032ab12345"), entry.entry_id
|
|
)
|
|
assert device_entry == snapshot
|