Add 'internal_state' to return the results from the last update query (python-kasa#306)

rytilahti · web-flow · commit b61c0feea9fa · 2022-02-07T09:13:47.000+01:00
This can be useful for debugging purposes, e.g., for homeassistant diagnostics
diff --git a/kasa/smartdevice.py b/kasa/smartdevice.py
@@ -721,6 +721,15 @@ def is_color(self) -> bool:
         """Return True if the device supports color changes."""
         return False
 
+    @property
+    def internal_state(self) -> Any:
+        """Return the internal state of the instance.
+
+        The returned object contains the raw results from the last update call.
+        This should only be used for debugging purposes.
+        """
+        return self._last_update
+
     def __repr__(self):
         if self._last_update is None:
             return f"<{self._device_type} at {self.host} - update() needed>"
diff --git a/kasa/tests/test_smartdevice.py b/kasa/tests/test_smartdevice.py
@@ -134,3 +134,8 @@ async def test_childrens(dev):
         assert len(dev.children) > 0
     else:
         assert len(dev.children) == 0
+
+
+async def test_internal_state(dev):
+    """Make sure the internal state returns the last update results."""
+    assert dev.internal_state == dev._last_update
