tuya-local/tests/devices/test_blitzwolf_bsh2_humidifier.py

from homeassistant.components.fan import FanEntityFeature

from ..const import BLITZWOLF_BWSH2_PAYLOAD
from ..helpers import assert_device_properties_set
from ..mixins.select import MultiSelectTests
from .base_device_tests import TuyaDeviceTestCase

SWITCH_DP = "1"
SPEED_DP = "3"
LIGHT_DP = "6"
TIMER_DP = "19"


class TestBlitzwolfSH2Humidifier(MultiSelectTests, TuyaDeviceTestCase):
    __test__ = True

    def setUp(self):
        self.setUpForConfig(
            "blitzwolf_bwsh2_humidifier.yaml",
            BLITZWOLF_BWSH2_PAYLOAD,
        )
        self.subject = self.entities.get("fan")
        self.light = self.entities.get("light")
        self.setUpMultiSelect(
            [
                {
                    "name": "select_light",
                    "dps": LIGHT_DP,
                    "options": {
                        "close": "Off",
                        "purple": "Purple",
                        "blue": "Blue",
                        "cyan": "Cyan",
                        "green": "Green",
                        "yellow": "Yellow",
                        "orange": "Orange",
                        "red": "Red",
                        "colour": "Colorful",
                    },
                },
                {
                    "name": "select_timer",
                    "dps": TIMER_DP,
                    "options": {
                        "cancel": "cancel",
                        "2h": "2h",
                        "4h": "4h",
                        "6h": "6h",
                        "8h": "8h",
                        "10h": "10h",
                        "12h": "12h",
                    },
                },
            ]
        )
        self.mark_secondary(["select_light", "select_timer"])

    def test_supported_features(self):
        self.assertEqual(
            self.subject.supported_features,
            FanEntityFeature.SET_SPEED
            | FanEntityFeature.TURN_OFF
            | FanEntityFeature.TURN_ON,
        )

    def test_speed(self):
        self.dps[SPEED_DP] = "sleep"
        self.assertEqual(self.subject.percentage, 10)
        self.dps[SPEED_DP] = "grade1"
        self.assertEqual(self.subject.percentage, 25)
        self.dps[SPEED_DP] = "grade2"
        self.assertEqual(self.subject.percentage, 40)
        self.dps[SPEED_DP] = "grade3"
        self.assertEqual(self.subject.percentage, 55)
        self.dps[SPEED_DP] = "grade4"
        self.assertEqual(self.subject.percentage, 70)
        self.dps[SPEED_DP] = "grade5"
        self.assertEqual(self.subject.percentage, 85)
        self.dps[SPEED_DP] = "grade6"
        self.assertEqual(self.subject.percentage, 100)

    async def test_set_speed_snaps(self):
        async with assert_device_properties_set(
            self.subject._device,
            {SPEED_DP: "grade3"},
        ):
            await self.subject.async_set_percentage(50)

    def test_light_named_color(self):
        self.dps[LIGHT_DP] = "colour"
        self.assertEqual(self.light.hs_color, (0, 0))
        self.dps[LIGHT_DP] = "red"
        self.assertEqual(self.light.hs_color, (0, 100))

    async def test_async_set_light_color(self):
        self.dps[LIGHT_DP] = "red"
        async with assert_device_properties_set(
            self.light._device,
            {LIGHT_DP: "green"},
        ):
            await self.light.async_turn_on(hs_color=(120, 100))

    async def test_async_light_turn_off(self):
        async with assert_device_properties_set(
            self.light._device,
            {LIGHT_DP: "close"},
        ):
            await self.light.async_turn_off()

    async def test_async_light_turn_on(self):
        async with assert_device_properties_set(
            self.light._device,
            {LIGHT_DP: "colour"},
        ):
            await self.light.async_turn_on()