Add support for Walkingpad A1 (ksmb.walkingpad.v3)

docs/api/miio.gateway.rst

@@ -1,5 +1,29 @@
-miio.gateway module
-===================
+miio.gateway package
+====================
+
+Subpackages
+-----------
+
+.. toctree::
+   :maxdepth: 4
+
+   miio.gateway.devices
+
+Submodules
+----------
+
+.. toctree::
+   :maxdepth: 4
+
+   miio.gateway.alarm
+   miio.gateway.gateway
+   miio.gateway.gatewaydevice
+   miio.gateway.light
+   miio.gateway.radio
+   miio.gateway.zigbee
+
+Module contents
+---------------
 
 .. automodule:: miio.gateway
    :members:

docs/api/miio.rst

@@ -1,12 +1,21 @@
 miio package
 ============
 
+Subpackages
+-----------
+
+.. toctree::
+   :maxdepth: 4
+
+   miio.gateway
+
 Submodules
 ----------
 
 .. toctree::
    :maxdepth: 4
 
+   miio.airconditioner_miot
    miio.airconditioningcompanion
    miio.airconditioningcompanionMCN
    miio.airdehumidifier
@@ -18,8 +27,10 @@ Submodules
    miio.airhumidifier_miot
    miio.airhumidifier_mjjsq
    miio.airpurifier
+   miio.airpurifier_airdog
    miio.airpurifier_miot
    miio.airqualitymonitor
+   miio.airqualitymonitor_miot
    miio.alarmclock
    miio.aqaracamera
    miio.ceil
@@ -30,15 +41,19 @@ Submodules
    miio.cli
    miio.click_common
    miio.cooker
+   miio.curtain_youpin
    miio.device
    miio.discovery
+   miio.dreamevacuum_miot
    miio.exceptions
    miio.extract_tokens
    miio.fan
    miio.fan_common
+   miio.fan_leshow
    miio.fan_miot
-   miio.gateway
    miio.heater
+   miio.heater_miot
+   miio.huizuo
    miio.miioprotocol
    miio.miot_device
    miio.philips_bulb
@@ -50,17 +65,22 @@ Submodules
    miio.powerstrip
    miio.protocol
    miio.pwzn_relay
+   miio.scishare_coffeemaker
    miio.toiletlid
    miio.updater
    miio.utils
    miio.vacuum
    miio.vacuum_cli
+   miio.vacuum_tui
    miio.vacuumcontainers
    miio.viomivacuum
+   miio.walkingpad
    miio.waterpurifier
+   miio.waterpurifier_yunmi
    miio.wifirepeater
    miio.wifispeaker
    miio.yeelight
+   miio.yeelight_dual_switch
 
 Module contents
 ---------------

miio/__init__.py

@@ -62,6 +62,7 @@
     VacuumStatus,
 )
 from miio.viomivacuum import ViomiVacuum
+from miio.walkingpad import Walkingpad
 from miio.waterpurifier import WaterPurifier
 from miio.waterpurifier_yunmi import WaterPurifierYunmi
 from miio.wifirepeater import WifiRepeater

miio/tests/test_walkingpad.py

@@ -0,0 +1,116 @@
+from unittest import TestCase
+
+import pytest
+
+from miio import Walkingpad
+from miio.walkingpad import OperationMode, WalkingpadException, WalkingpadStatus
+
+from .dummies import DummyDevice
+
+
+class DummyWalkingpad(DummyDevice, Walkingpad):
+    def __init__(self, *args, **kwargs):
+        self.state = {
+            "power": "on",
+            "mode": 1,
+            "time": 1387,
+            "step": 2117,
+            "dist": 1150,
+            "sp": 3.05,
+            "cal": 71710,
+            "all": [
+                "mode:1",
+                "time:1387",
+                "sp:3.05",
+                "dist:1150",
+                "cal:71710",
+                "step:2117",
+            ],
+        }
+        self.return_values = {
+            "get_prop": self._get_state,
+            "set_power": lambda x: self._set_state("power", x),
+            "set_mode": lambda x: self._set_state("mode", x),
+            "set_speed": lambda x: (
+                self._set_state(
+                    "all",
+                    [
+                        "mode:1",
+                        "time:1387",
+                        "sp:" + str(x),
+                        "dist:1150",
+                        "cal:71710",
+                        "step:2117",
+                    ],
+                ),
+                self._set_state("sp", x),
+            ),
+            "set_step": lambda x: self._set_state("step", x),
+            "set_time": lambda x: self._set_state("time", x),
+            "set_distance": lambda x: self._set_state("dist", x),
+        }
+        super().__init__(args, kwargs)
+
+
+@pytest.fixture(scope="class")
+def walkingpad(request):
+    request.cls.device = DummyWalkingpad()
+    # TODO add ability to test on a real device
+
+
+@pytest.mark.usefixtures("walkingpad")
+class TestWalkingpad(TestCase):
+    def is_on(self):
+        return self.device.status().is_on
+
+    def state(self):
+        return self.device.status()
+
+    def test_on(self):
+        self.device.off()  # ensure off
+        assert self.is_on() is False
+
+        self.device.on()
+        assert self.is_on() is True
+
+    def test_off(self):
+        self.device.on()  # ensure on
+        assert self.is_on() is True
+
+        self.device.off()
+        assert self.is_on() is False
+
+    def test_status(self):
+        self.device._reset_state()
+
+        # Because we get a set of status values from the walkingpad via 'all' the dummy device doesnt work for testing.
+        # TODO Need to figure out how to test this properly.
+
+        assert repr(self.state()) == repr(WalkingpadStatus(self.device.start_state))
+
+        assert self.is_on() is True
+        assert self.state().power == self.device.start_state["power"]
+        assert self.state().mode == self.device.start_state["mode"]
+        # assert self.state().speed == self.device.start_state["speed"]
+        # assert self.state().step == self.device.start_state["step"]
+        # assert self.state().distance == self.device.start_state["dist"]
+        # assert self.state().time == self.device.start_state["time"]
+
+    def test_set_mode(self):
+        def mode():
+            return self.device.status().mode
+
+        self.device.set_mode(OperationMode.Auto)
+        assert mode() == OperationMode.Auto.value
+
+        self.device.set_mode(OperationMode.Manual)
+        assert mode() == OperationMode.Manual.value
+
+        with pytest.raises(WalkingpadException):
+            self.device.set_mode(-1)
+
+        with pytest.raises(WalkingpadException):
+            self.device.set_mode(3)
+
+        with pytest.raises(WalkingpadException):
+            self.device.set_mode("blah")