tuya-local/tests/devices/test_saswell_t29utk_thermostat.py

from homeassistant.components.climate.const import (
    ClimateEntityFeature,
    HVACAction,
    HVACMode,
    FAN_AUTO,
    FAN_ON,
    PRESET_AWAY,
    PRESET_HOME,
)
from homeassistant.const import UnitOfTemperature

from ..const import SASWELL_T29UTK_THERMOSTAT_PAYLOAD
from ..helpers import assert_device_properties_set
from ..mixins.climate import TargetTemperatureTests
from ..mixins.select import MultiSelectTests
from .base_device_tests import TuyaDeviceTestCase

POWER_DPS = "1"
TEMPERATURE_DPS = "2"
CURRENTTEMP_DPS = "3"
HVACMODE_DPS = "4"
FAN_DPS = "5"
UNITS_DPS = "19"
AWAY_DPS = "101"
PROGRAM_DPS = "102"
HVACACTION_DPS = "103"
CONFIG_DPS = "112"
UNKNOWN113_DPS = "113"
TEMPC_DPS = "114"
CURTEMPC_DPS = "115"
TEMPF_DPS = "116"
CURTEMPF_DPS = "117"


class TestSaswellT29UTKThermostat(
    MultiSelectTests, TargetTemperatureTests, TuyaDeviceTestCase
):
    __test__ = True

    def setUp(self):
        self.setUpForConfig(
            "saswell_t29utk_thermostat.yaml", SASWELL_T29UTK_THERMOSTAT_PAYLOAD
        )
        self.subject = self.entities.get("climate")
        self.setUpTargetTemperature(
            TEMPERATURE_DPS,
            self.subject,
            min=5.0,
            max=35.0,
            scale=10,
            step=5,
        )
        self.setUpMultiSelect(
            [
                {
                    "name": "select_temperature_unit",
                    "dps": UNITS_DPS,
                    "options": {
                        "C": "Celsius",
                        "F": "Fahrenheit",
                    },
                },
                {
                    "name": "select_configuration",
                    "dps": CONFIG_DPS,
                    "options": {
                        "1": "cooling",
                        "2": "heating",
                        "3": "heat/cool",
                        "5": "heatpump",
                    },
                },
            ]
        )
        self.mark_secondary(["select_configuration", "select_temperature_unit"])

    def test_supported_features(self):
        self.assertEqual(
            self.subject.supported_features,
            (
                ClimateEntityFeature.TARGET_TEMPERATURE
                | ClimateEntityFeature.FAN_MODE
                | ClimateEntityFeature.PRESET_MODE
            ),
        )

    def test_icon(self):
        self.dps[HVACMODE_DPS] = "cold"
        self.assertEqual(self.subject.icon, "mdi:thermometer-minus")

        self.dps[HVACMODE_DPS] = "hot"
        self.assertEqual(self.subject.icon, "mdi:thermometer-plus")

        self.dps[HVACMODE_DPS] = "off"
        self.assertEqual(self.subject.icon, "mdi:thermometer-off")

    def test_temperature_unit(self):
        self.dps[UNITS_DPS] = "C"
        self.assertEqual(self.subject.temperature_unit, UnitOfTemperature.CELSIUS)

        self.dps[UNITS_DPS] = "F"
        self.assertEqual(self.subject.temperature_unit, UnitOfTemperature.FAHRENHEIT)

    def test_target_temperature_step_f(self):
        self.dps[UNITS_DPS] = "F"
        self.assertEqual(self.subject.target_temperature_step, 1)

    def test_minimum_target_temperature_f(self):
        self.dps[UNITS_DPS] = "F"
        self.assertEqual(self.subject.min_temp, 41)

    def test_maximum_target_temperature_f(self):
        self.dps[UNITS_DPS] = "F"
        self.assertEqual(self.subject.max_temp, 95)

    async def test_set_target_temperature_f(self):
        self.dps[UNITS_DPS] = "F"
        async with assert_device_properties_set(
            self.subject._device,
            {TEMPERATURE_DPS: 740},
        ):
            await self.subject.async_set_target_temperature(74)

    def test_current_temperature(self):
        self.dps[CURRENTTEMP_DPS] = 250
        self.assertEqual(self.subject.current_temperature, 25.0)

    def test_hvac_mode(self):
        self.dps[HVACMODE_DPS] = "off"
        self.assertEqual(self.subject.hvac_mode, HVACMode.OFF)

        self.dps[HVACMODE_DPS] = "cold"
        self.assertEqual(self.subject.hvac_mode, HVACMode.COOL)

        self.dps[HVACMODE_DPS] = "hot"
        self.assertEqual(self.subject.hvac_mode, HVACMode.HEAT)

    def test_hvac_modes(self):
        self.assertCountEqual(
            self.subject.hvac_modes,
            [HVACMode.COOL, HVACMode.HEAT, HVACMode.OFF],
        )

    async def test_turn_off(self):
        async with assert_device_properties_set(
            self.subject._device, {HVACMODE_DPS: "off"}
        ):
            await self.subject.async_set_hvac_mode(HVACMode.OFF)

    async def test_set_hvac_mode_cool(self):
        async with assert_device_properties_set(
            self.subject._device, {HVACMODE_DPS: "cold"}
        ):
            await self.subject.async_set_hvac_mode(HVACMode.COOL)

    async def test_set_hvac_mode_heat(self):
        async with assert_device_properties_set(
            self.subject._device, {HVACMODE_DPS: "hot"}
        ):
            await self.subject.async_set_hvac_mode(HVACMode.HEAT)

    async def test_set_hvac_mode_heat_fails_in_cooling_config(self):
        self.dps[CONFIG_DPS] = "1"
        with self.assertRaisesRegex(
            AttributeError, "hvac_mode cannot be set at this time"
        ):
            await self.subject.async_set_hvac_mode(HVACMode.HEAT)

    async def test_set_hvac_mode_cool_fails_in_heating_config(self):
        self.dps[CONFIG_DPS] = "2"
        with self.assertRaisesRegex(
            AttributeError, "hvac_mode cannot be set at this time"
        ):
            await self.subject.async_set_hvac_mode(HVACMode.COOL)

    def test_hvac_action(self):
        self.dps[POWER_DPS] = True
        self.dps[HVACACTION_DPS] = "cold"
        self.assertEqual(self.subject.hvac_action, HVACAction.COOLING)

        self.dps[HVACACTION_DPS] = "hot"
        self.assertEqual(self.subject.hvac_action, HVACAction.HEATING)

        self.dps[POWER_DPS] = False
        self.assertEqual(self.subject.hvac_action, HVACAction.OFF)

    def test_fan_mode(self):
        self.dps[FAN_DPS] = "auto"
        self.assertEqual(self.subject.fan_mode, FAN_AUTO)
        self.dps[FAN_DPS] = "on"
        self.assertEqual(self.subject.fan_mode, FAN_ON)

    def test_fan_modes(self):
        self.assertCountEqual(self.subject.fan_modes, [FAN_AUTO, FAN_ON])

    async def test_set_fan_on(self):
        async with assert_device_properties_set(
            self.subject._device,
            {FAN_DPS: "on"},
        ):
            await self.subject.async_set_fan_mode(FAN_ON)

    async def test_set_fan_auto(self):
        async with assert_device_properties_set(
            self.subject._device,
            {FAN_DPS: "auto"},
        ):
            await self.subject.async_set_fan_mode(FAN_AUTO)

    def test_preset_mode(self):
        self.dps[AWAY_DPS] = False
        self.dps[PROGRAM_DPS] = False
        self.assertEqual(self.subject.preset_mode, PRESET_HOME)
        self.dps[PROGRAM_DPS] = True
        self.assertEqual(self.subject.preset_mode, "Program")
        self.dps[AWAY_DPS] = True
        self.assertEqual(self.subject.preset_mode, PRESET_AWAY)
        self.dps[PROGRAM_DPS] = False
        self.assertEqual(self.subject.preset_mode, PRESET_AWAY)

    def test_preset_modes(self):
        self.assertCountEqual(
            self.subject.preset_modes,
            [PRESET_HOME, PRESET_AWAY, "Program"],
        )

    async def test_set_preset_to_away(self):
        async with assert_device_properties_set(
            self.subject._device,
            {AWAY_DPS: True},
        ):
            await self.subject.async_set_preset_mode(PRESET_AWAY)

    async def test_set_preset_to_home(self):
        async with assert_device_properties_set(
            self.subject._device,
            {AWAY_DPS: False, PROGRAM_DPS: False},
        ):
            await self.subject.async_set_preset_mode(PRESET_HOME)

    async def test_set_preset_to_program(self):
        async with assert_device_properties_set(
            self.subject._device,
            {AWAY_DPS: False, PROGRAM_DPS: True},
        ):
            await self.subject.async_set_preset_mode("Program")

    def test_extra_state_attributes(self):
        self.dps[POWER_DPS] = True
        self.dps[CONFIG_DPS] = "2"
        self.dps[UNKNOWN113_DPS] = 113
        self.dps[TEMPC_DPS] = 114
        self.dps[CURTEMPC_DPS] = 115
        self.dps[TEMPF_DPS] = 116
        self.dps[CURTEMPF_DPS] = 117

        self.assertDictEqual(
            self.subject.extra_state_attributes,
            {
                "power": True,
                "configuration": "heating",
                "unknown_113": 113,
                "temperature_c": 114,
                "current_temperature_c": 115,
                "temperature_f": 116,
                "current_temperature_f": 117,
            },
        )