tuya-local/tests/devices/test_qoto_03_sprinkler.py

"""Tests for the Quto 03 Sprinkler."""

from homeassistant.components.binary_sensor import BinarySensorDeviceClass
from homeassistant.const import PERCENTAGE, UnitOfTime

from ..const import QOTO_SPRINKLER_PAYLOAD
from ..mixins.binary_sensor import BasicBinarySensorTests
from ..mixins.number import MultiNumberTests
from ..mixins.sensor import MultiSensorTests
from .base_device_tests import TuyaDeviceTestCase

TARGET_DPS = "102"
CURRENT_DPS = "103"
COUNTDOWN_DPS = "104"
TIMER_DPS = "105"
ERROR_DPS = "108"


class TestQotoSprinkler(
    BasicBinarySensorTests,
    MultiNumberTests,
    MultiSensorTests,
    TuyaDeviceTestCase,
):
    __test__ = True

    def setUp(self):
        self.setUpForConfig("qoto_03_sprinkler.yaml", QOTO_SPRINKLER_PAYLOAD)
        self.setUpBasicBinarySensor(
            ERROR_DPS,
            self.entities.get("binary_sensor_error"),
            device_class=BinarySensorDeviceClass.PROBLEM,
            testdata=(1, 0),
        )
        self.setUpMultiNumber(
            [
                {
                    "name": "number",
                    "dps": TARGET_DPS,
                    "max": 100,
                    "step": 5,
                    "unit": PERCENTAGE,
                },
                {
                    "name": "number_timer",
                    "dps": TIMER_DPS,
                    "max": 86399,
                    "unit": UnitOfTime.SECONDS,
                },
            ]
        )
        self.setUpMultiSensors(
            [
                {
                    "name": "sensor_open",
                    "dps": CURRENT_DPS,
                    "unit": "%",
                },
                {
                    "name": "sensor_timer",
                    "dps": COUNTDOWN_DPS,
                    "device_class": "duration",
                    "unit": "s",
                },
            ]
        )
        self.mark_secondary(
            [
                "binary_sensor_error",
                "number_timer",
                "sensor_open",
                "sensor_timer",
            ]
        )