mirror of
https://github.com/home-assistant/core.git
synced 2026-08-28 18:24:50 -05:00
7c662cdfd1
Co-authored-by: Simone Chemelli <simone.chemelli@gmail.com> Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com> Co-authored-by: J. Nick Koston <nick@koston.org>
248 lines
9.0 KiB
Python
248 lines
9.0 KiB
Python
"""Tests for Midea sensor.py."""
|
|
|
|
from collections.abc import Callable
|
|
from unittest.mock import patch
|
|
|
|
from midealocal.const import DeviceType
|
|
from midealocal.devices.ac import DeviceAttributes as ACAttributes
|
|
from midealocal.devices.c3 import DeviceAttributes as C3Attributes
|
|
from midealocal.devices.db import DeviceAttributes as DBAttributes
|
|
from midealocal.devices.e8 import DeviceAttributes as E8Attributes
|
|
from midealocal.devices.ea import DeviceAttributes as EAAttributes
|
|
from midealocal.devices.ec import DeviceAttributes as ECAttributes
|
|
from midealocal.devices.ed import DeviceAttributes as EDAttributes
|
|
import pytest
|
|
from syrupy.assertion import SnapshotAssertion
|
|
|
|
from homeassistant.const import Platform
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import entity_registry as er
|
|
|
|
from . import setup_integration
|
|
from .conftest import DummyDevice, SetDeviceAttribute, entity_entries
|
|
from .const import TEST_DEVICE_ID
|
|
|
|
from tests.common import MockConfigEntry, snapshot_platform
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"device",
|
|
[
|
|
pytest.param(
|
|
DummyDevice(
|
|
DeviceType.AC,
|
|
attributes={
|
|
ACAttributes.power: True,
|
|
ACAttributes.mode: 1,
|
|
ACAttributes.target_temperature: 22.0,
|
|
ACAttributes.indoor_temperature: 21.0,
|
|
ACAttributes.comfort_mode: False,
|
|
ACAttributes.eco_mode: False,
|
|
ACAttributes.boost_mode: False,
|
|
ACAttributes.sleep_mode: False,
|
|
ACAttributes.frost_protect: False,
|
|
ACAttributes.fan_speed: 103,
|
|
ACAttributes.swing_vertical: True,
|
|
ACAttributes.swing_horizontal: True,
|
|
ACAttributes.indoor_humidity: 50,
|
|
},
|
|
),
|
|
id="ac",
|
|
),
|
|
pytest.param(
|
|
DummyDevice(
|
|
DeviceType.C3,
|
|
attributes={
|
|
C3Attributes.zone_temp_type: [True, False],
|
|
C3Attributes.temperature_min: [16, 17],
|
|
C3Attributes.temperature_max: [30, 29],
|
|
C3Attributes.mode: 1,
|
|
C3Attributes.zone1_power: True,
|
|
C3Attributes.zone2_power: False,
|
|
C3Attributes.target_temperature: [22, 23],
|
|
C3Attributes.temp_tw_out: 21.5,
|
|
},
|
|
),
|
|
id="c3",
|
|
),
|
|
pytest.param(
|
|
DummyDevice(
|
|
DeviceType.E8,
|
|
attributes={
|
|
E8Attributes.status: 1,
|
|
E8Attributes.time_remaining: 3600,
|
|
E8Attributes.keep_warm_remaining: 1800,
|
|
E8Attributes.working_time: 7200,
|
|
E8Attributes.target_temperature: 22.0,
|
|
E8Attributes.current_temperature: 21.0,
|
|
E8Attributes.finished: False,
|
|
E8Attributes.water_shortage: False,
|
|
},
|
|
),
|
|
id="e8",
|
|
),
|
|
pytest.param(
|
|
DummyDevice(
|
|
DeviceType.DB,
|
|
attributes={
|
|
DBAttributes.power: True,
|
|
DBAttributes.mode: "normal",
|
|
DBAttributes.temperature: 22.0,
|
|
DBAttributes.wash_time: 65,
|
|
DBAttributes.dehydration_time: 30,
|
|
DBAttributes.program: "cotton",
|
|
},
|
|
),
|
|
id="db",
|
|
),
|
|
pytest.param(
|
|
DummyDevice(
|
|
DeviceType.DB,
|
|
attributes={
|
|
DBAttributes.power: False,
|
|
DBAttributes.mode: "unknown",
|
|
DBAttributes.temperature: 22.0,
|
|
DBAttributes.wash_time: 65,
|
|
DBAttributes.dehydration_time: 30,
|
|
DBAttributes.program: "unknown",
|
|
},
|
|
),
|
|
id="db_unknown_attributes",
|
|
),
|
|
pytest.param(
|
|
DummyDevice(
|
|
DeviceType.EA,
|
|
attributes={
|
|
EAAttributes.bottom_temperature: 180,
|
|
EAAttributes.cooking: 1,
|
|
EAAttributes.keep_warm: True,
|
|
EAAttributes.mode: "heat_rice",
|
|
EAAttributes.progress: 50,
|
|
EAAttributes.keep_warm_time: 30,
|
|
EAAttributes.time_remaining: 5,
|
|
EAAttributes.top_temperature: 200,
|
|
},
|
|
),
|
|
id="ea",
|
|
),
|
|
pytest.param(
|
|
DummyDevice(
|
|
DeviceType.EC,
|
|
attributes={
|
|
ECAttributes.cooking: 1,
|
|
ECAttributes.mode: "diy",
|
|
ECAttributes.progress: 50,
|
|
ECAttributes.keep_warm_time: 30,
|
|
ECAttributes.time_remaining: 5,
|
|
ECAttributes.top_temperature: 200,
|
|
ECAttributes.with_pressure: True,
|
|
},
|
|
),
|
|
id="ec",
|
|
),
|
|
pytest.param(
|
|
DummyDevice(
|
|
DeviceType.ED,
|
|
attributes={
|
|
EDAttributes.keep_warm: True,
|
|
EDAttributes.boil_temperature: 100,
|
|
EDAttributes.boiling: True,
|
|
EDAttributes.keep_warm_time: 30,
|
|
EDAttributes.child_lock: True,
|
|
EDAttributes.cl_sterilization: True,
|
|
EDAttributes.cooling: True,
|
|
EDAttributes.current_temperature: 150,
|
|
EDAttributes.dispensing: True,
|
|
EDAttributes.error: 3,
|
|
EDAttributes.filter1: 30,
|
|
EDAttributes.filter2: 20,
|
|
EDAttributes.filter3: 100,
|
|
EDAttributes.flushing_days: 50,
|
|
EDAttributes.heating: True,
|
|
EDAttributes.hot_water_dispensing: True,
|
|
EDAttributes.in_tds: 200,
|
|
EDAttributes.keep_warm_remaining: 20,
|
|
EDAttributes.lack_water: True,
|
|
EDAttributes.leak_water: True,
|
|
EDAttributes.out_tds: 100,
|
|
EDAttributes.power: True,
|
|
EDAttributes.water_consumption: 500,
|
|
EDAttributes.water_consumption_big: 6000,
|
|
EDAttributes.water_consumption_average: 30,
|
|
},
|
|
),
|
|
id="ed",
|
|
),
|
|
],
|
|
)
|
|
async def test_all_entities(
|
|
hass: HomeAssistant,
|
|
device: DummyDevice,
|
|
mock_config_entry: Callable[[DummyDevice], MockConfigEntry],
|
|
snapshot: SnapshotAssertion,
|
|
entity_registry: er.EntityRegistry,
|
|
) -> None:
|
|
"""Test sensor entities are created."""
|
|
config_entry = mock_config_entry(device)
|
|
with patch("homeassistant.components.midea._PLATFORMS", [Platform.SENSOR]):
|
|
await setup_integration(hass, config_entry, device)
|
|
|
|
await snapshot_platform(hass, entity_registry, snapshot, config_entry.entry_id)
|
|
|
|
|
|
async def test_sensor_state_update(
|
|
hass: HomeAssistant,
|
|
set_device_attribute: SetDeviceAttribute,
|
|
mock_config_entry: Callable[[DummyDevice], MockConfigEntry],
|
|
) -> None:
|
|
"""Test sensor state follows push updates from the device."""
|
|
device = DummyDevice(
|
|
DeviceType.AC,
|
|
attributes={
|
|
ACAttributes.power: True,
|
|
ACAttributes.mode: 1,
|
|
ACAttributes.target_temperature: 22.0,
|
|
ACAttributes.indoor_temperature: 21.0,
|
|
ACAttributes.indoor_humidity: 0,
|
|
ACAttributes.full_dust: False,
|
|
ACAttributes.outdoor_temperature: "unknown",
|
|
},
|
|
)
|
|
config_entry = mock_config_entry(device)
|
|
with patch("homeassistant.components.midea._PLATFORMS", [Platform.SENSOR]):
|
|
await setup_integration(hass, config_entry, device)
|
|
|
|
assert len(entity_entries(hass, config_entry)) == 3
|
|
entity_entry = entity_entries(hass, config_entry)[
|
|
f"{TEST_DEVICE_ID}_indoor_temperature"
|
|
]
|
|
|
|
state = hass.states.get(entity_entry.entity_id)
|
|
assert state is not None
|
|
assert state.state == "21.0"
|
|
|
|
await set_device_attribute(device, ACAttributes.indoor_temperature, 19.5)
|
|
|
|
state = hass.states.get(entity_entry.entity_id)
|
|
assert state is not None
|
|
assert state.state == "19.5"
|
|
|
|
entity_entry = entity_entries(hass, config_entry)[
|
|
f"{TEST_DEVICE_ID}_outdoor_temperature"
|
|
]
|
|
state = hass.states.get(entity_entry.entity_id)
|
|
assert state is not None
|
|
assert state.state == "unknown"
|
|
|
|
entity_entry = entity_entries(hass, config_entry)[
|
|
f"{TEST_DEVICE_ID}_indoor_humidity"
|
|
]
|
|
state = hass.states.get(entity_entry.entity_id)
|
|
assert state is not None
|
|
assert state.state == "unknown"
|
|
|
|
await set_device_attribute(device, ACAttributes.indoor_humidity, 255)
|
|
state = hass.states.get(entity_entry.entity_id)
|
|
assert state is not None
|
|
assert state.state == "unknown"
|