"""Tests for Midea sensor.py.""" from collections.abc import Callable from unittest.mock import patch from freezegun.api import FrozenDateTimeFactory from midealocal.const import DeviceType from midealocal.devices.ac import DeviceAttributes as ACAttributes from midealocal.devices.c3 import DeviceAttributes as C3Attributes from midealocal.devices.ca import MideaCADevice from midealocal.devices.db import DeviceAttributes as DBAttributes, MideaDBDevice from midealocal.devices.e8 import DeviceAttributes as E8Attributes from midealocal.devices.ea import DeviceAttributes as EAAttributes, MideaEADevice from midealocal.devices.ec import DeviceAttributes as ECAttributes, MideaECDevice from midealocal.devices.ed import DeviceAttributes as EDAttributes import pytest from syrupy.assertion import SnapshotAssertion from homeassistant.components.midea.sensor import SENSOR_ENTITIES 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: 60, 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, freezer: FrozenDateTimeFactory, device: DummyDevice, mock_config_entry: Callable[[DummyDevice], MockConfigEntry], snapshot: SnapshotAssertion, entity_registry: er.EntityRegistry, ) -> None: """Test sensor entities are created.""" freezer.move_to("2026-01-09 12:00:00+00:00") 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: 45, 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 == "45" await set_device_attribute(device, ACAttributes.indoor_humidity, None) state = hass.states.get(entity_entry.entity_id) assert state is not None assert state.state == "unknown" async def test_time_remaining_sensor_as_timestamp( hass: HomeAssistant, freezer: FrozenDateTimeFactory, set_device_attribute: SetDeviceAttribute, mock_config_entry: Callable[[DummyDevice], MockConfigEntry], ) -> None: """Test time_remaining is reported as an absolute timestamp in minutes.""" freezer.move_to("2026-01-09 12:00:00+00:00") device = DummyDevice( DeviceType.E8, attributes={E8Attributes.time_remaining: 90}, ) config_entry = mock_config_entry(device) with patch("homeassistant.components.midea._PLATFORMS", [Platform.SENSOR]): await setup_integration(hass, config_entry, device) entity_entry = entity_entries(hass, config_entry)[ f"{TEST_DEVICE_ID}_time_remaining" ] state = hass.states.get(entity_entry.entity_id) assert state is not None assert state.state == "2026-01-09T13:30:00+00:00" await set_device_attribute(device, E8Attributes.time_remaining, 0) state = hass.states.get(entity_entry.entity_id) assert state is not None assert state.state == "unknown" await set_device_attribute(device, E8Attributes.time_remaining, None) state = hass.states.get(entity_entry.entity_id) assert state is not None assert state.state == "unknown" @pytest.mark.parametrize( ("model", "expected_options"), [ pytest.param(DeviceType.DB, MideaDBDevice.mode_options(), id="db"), pytest.param(DeviceType.EA, MideaEADevice.mode_options(), id="ea"), pytest.param(DeviceType.EC, MideaECDevice.mode_options(), id="ec"), ], ) def test_mode_options_match_library( model: DeviceType, expected_options: list[str], ) -> None: """Guard against drift between the hardcoded mode options and midealocal's tables.""" description = next( description for description in SENSOR_ENTITIES if description.key == "mode" and description.models == [model] ) assert description.options == expected_options def test_variable_mode_options_match_library() -> None: """Guard against drift between the hardcoded variable_mode options and midealocal's table.""" description = next( description for description in SENSOR_ENTITIES if description.key == "variable_mode" ) assert description.options == MideaCADevice.mode_options()