save capabilities for "main" component too

andrewsayre · Sep 8, 2023 · 84166bd · 84166bd
1 parent ca4917e
commit 84166bd
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/pysmartthings/device.py b/pysmartthings/device.py
@@ -122,8 +122,7 @@ def apply_data(self, data: dict):
                 component_id = component["id"]
                 if component_id == "main":
                     self._capabilities.extend(capabilities)
-                else:
-                    self._components[component_id] = capabilities
+                self._components[component_id] = capabilities
 
         if self._type == DEVICE_TYPE_DTH:
             dth = data.get("dth")

diff --git a/tests/test_device.py b/tests/test_device.py
@@ -58,6 +58,16 @@ def test_apply_data():
             "light",
         ]
         assert device.components == {
+            "main": [
+                "switch",
+                "switchLevel",
+                "refresh",
+                "indicator",
+                "sensor",
+                "actuator",
+                "healthCheck",
+                "light",
+            ],
             "bottomButton": ["button"],
             "topButton": ["button"],
         }