Fix bot error message handling (#7)

sucks/__init__.py

@@ -93,6 +93,15 @@
     'dust_case_heap': COMPONENT_FILTER
 }
 
+ERROR_CODES = {
+    "100": "Robot is operational",
+    "101": "Low battery",
+    "102": "Robot is stuck",
+    "103": "Wheels are not moving as expected",
+    "104": "Down sensor is getting abnormal values",
+    "110": "Dust Bin Not installed",
+}
+
 class EcoVacsAPI:
     CLIENT_KEY = "eJUWrzRv34qFSaYk"
     SECRET = "Cyu5jcR4zyK6QEPn1hdIGXB5QIDAQABMA0GC"
@@ -289,7 +298,9 @@ def _handle_ctl(self, ctl):
             getattr(self, method)(ctl)
 
     def _handle_error(self, event):
-        error = event['error']
+        errno = event.get('errno')
+        if not errno or (error := ERROR_CODES.get(errno)) is None:
+            error = event.get('error', 'unknown')
         self.errorEvents.notify(error)
         _LOGGER.debug("*** error = " + error)
 

tests/test_vacbot.py

@@ -275,6 +275,16 @@ def test_error_event_subscription():
     v._handle_ctl({"event": "error", "error": "an_error_name"})
     mock.assert_called_once_with("an_error_name")
 
+    for err_code, err_msg in ERROR_CODES.items():
+        v._handle_ctl({"event": "error", "errno": err_code})
+        mock.assert_called_with(err_msg)
+
+    v._handle_ctl({"event": "error", "errno": "999"})
+    mock.assert_called_with("unknown")
+
+    v._handle_ctl({"event": "error"})
+    mock.assert_called_with("unknown")
+
     # Test unsubscribe
     mock = Mock()
     subscription = v.errorEvents.subscribe(mock)