Raise DeviceError on unsupported methods for tests

rytilahti · Feb 3, 2023 · df9de6c · df9de6c
1 parent 94cae65
commit df9de6c
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/miio/tests/dummies.py b/miio/tests/dummies.py
@@ -1,3 +1,6 @@
+from miio import DeviceError
+
+
 class DummyMiIOProtocol:
     """DummyProtocol allows you mock MiIOProtocol."""
 
@@ -8,7 +11,10 @@ def __init__(self, dummy_device):
 
     def send(self, command: str, parameters=None, retry_count=3, extra_parameters=None):
         """Overridden send() to return values from `self.return_values`."""
-        return self.dummy_device.return_values[command](parameters)
+        try:
+            return self.dummy_device.return_values[command](parameters)
+        except KeyError:
+            raise DeviceError({"code": -32601, "message": "Method not found."})
 
 
 class DummyDevice: