Add boost switches for Shelly cury component (home-assistant#154387)

bieniu · web-flow · commit 02e05643f15e · 2025-10-14T11:21:55.000+02:00
diff --git a/homeassistant/components/shelly/icons.json b/homeassistant/components/shelly/icons.json
@@ -68,6 +68,9 @@
           "on": "mdi:valve-open"
         }
       },
+      "cury_boost": {
+        "default": "mdi:rocket-launch"
+      },
       "cury_slot": {
         "default": "mdi:scent",
         "state": {
diff --git a/homeassistant/components/shelly/switch.py b/homeassistant/components/shelly/switch.py
@@ -243,6 +243,19 @@ class RpcSwitchDescription(RpcEntityDescription, SwitchEntityDescription):
         available=lambda status: (left := status["left"]) is not None
         and left.get("vial", {}).get("level", -1) != -1,
     ),
+    "cury_left_boost": RpcSwitchDescription(
+        key="cury",
+        sub_key="slots",
+        name="Left slot boost",
+        translation_key="cury_boost",
+        is_on=lambda status: status["slots"]["left"]["boost"] is not None,
+        method_on="cury_boost",
+        method_off="cury_stop_boost",
+        method_params_fn=lambda id, _: (id, "left"),
+        entity_registry_enabled_default=True,
+        available=lambda status: (left := status["left"]) is not None
+        and left.get("vial", {}).get("level", -1) != -1,
+    ),
     "cury_right": RpcSwitchDescription(
         key="cury",
         sub_key="slots",
@@ -256,6 +269,19 @@ class RpcSwitchDescription(RpcEntityDescription, SwitchEntityDescription):
         available=lambda status: (right := status["right"]) is not None
         and right.get("vial", {}).get("level", -1) != -1,
     ),
+    "cury_right_boost": RpcSwitchDescription(
+        key="cury",
+        sub_key="slots",
+        name="Right slot boost",
+        translation_key="cury_boost",
+        is_on=lambda status: status["slots"]["right"]["boost"] is not None,
+        method_on="cury_boost",
+        method_off="cury_stop_boost",
+        method_params_fn=lambda id, _: (id, "right"),
+        entity_registry_enabled_default=True,
+        available=lambda status: (right := status["right"]) is not None
+        and right.get("vial", {}).get("level", -1) != -1,
+    ),
 }
 
 
diff --git a/tests/components/shelly/snapshots/test_switch.ambr b/tests/components/shelly/snapshots/test_switch.ambr
@@ -47,6 +47,54 @@
     'state': 'on',
   })
 # ---
+# name: test_cury_switch_entity[switch.test_name_left_slot_boost-entry]
+  EntityRegistryEntrySnapshot({
+    'aliases': set({
+    }),
+    'area_id': None,
+    'capabilities': None,
+    'config_entry_id': <ANY>,
+    'config_subentry_id': <ANY>,
+    'device_class': None,
+    'device_id': <ANY>,
+    'disabled_by': None,
+    'domain': 'switch',
+    'entity_category': None,
+    'entity_id': 'switch.test_name_left_slot_boost',
+    'has_entity_name': True,
+    'hidden_by': None,
+    'icon': None,
+    'id': <ANY>,
+    'labels': set({
+    }),
+    'name': None,
+    'options': dict({
+    }),
+    'original_device_class': None,
+    'original_icon': None,
+    'original_name': 'Left slot boost',
+    'platform': 'shelly',
+    'previous_unique_id': None,
+    'suggested_object_id': None,
+    'supported_features': 0,
+    'translation_key': 'cury_boost',
+    'unique_id': '123456789ABC-cury:0-cury_left_boost',
+    'unit_of_measurement': None,
+  })
+# ---
+# name: test_cury_switch_entity[switch.test_name_left_slot_boost-state]
+  StateSnapshot({
+    'attributes': ReadOnlyDict({
+      'friendly_name': 'Test name Left slot boost',
+    }),
+    'context': <ANY>,
+    'entity_id': 'switch.test_name_left_slot_boost',
+    'last_changed': <ANY>,
+    'last_reported': <ANY>,
+    'last_updated': <ANY>,
+    'state': 'on',
+  })
+# ---
 # name: test_cury_switch_entity[switch.test_name_right_slot-entry]
   EntityRegistryEntrySnapshot({
     'aliases': set({
@@ -95,3 +143,51 @@
     'state': 'off',
   })
 # ---
+# name: test_cury_switch_entity[switch.test_name_right_slot_boost-entry]
+  EntityRegistryEntrySnapshot({
+    'aliases': set({
+    }),
+    'area_id': None,
+    'capabilities': None,
+    'config_entry_id': <ANY>,
+    'config_subentry_id': <ANY>,
+    'device_class': None,
+    'device_id': <ANY>,
+    'disabled_by': None,
+    'domain': 'switch',
+    'entity_category': None,
+    'entity_id': 'switch.test_name_right_slot_boost',
+    'has_entity_name': True,
+    'hidden_by': None,
+    'icon': None,
+    'id': <ANY>,
+    'labels': set({
+    }),
+    'name': None,
+    'options': dict({
+    }),
+    'original_device_class': None,
+    'original_icon': None,
+    'original_name': 'Right slot boost',
+    'platform': 'shelly',
+    'previous_unique_id': None,
+    'suggested_object_id': None,
+    'supported_features': 0,
+    'translation_key': 'cury_boost',
+    'unique_id': '123456789ABC-cury:0-cury_right_boost',
+    'unit_of_measurement': None,
+  })
+# ---
+# name: test_cury_switch_entity[switch.test_name_right_slot_boost-state]
+  StateSnapshot({
+    'attributes': ReadOnlyDict({
+      'friendly_name': 'Test name Right slot boost',
+    }),
+    'context': <ANY>,
+    'entity_id': 'switch.test_name_right_slot_boost',
+    'last_changed': <ANY>,
+    'last_reported': <ANY>,
+    'last_updated': <ANY>,
+    'state': 'off',
+  })
+# ---
diff --git a/tests/components/shelly/test_switch.py b/tests/components/shelly/test_switch.py
@@ -832,11 +832,13 @@ async def test_cury_switch_entity(
                 "left": {
                     "intensity": 70,
                     "on": True,
+                    "boost": {"started_at": 1760365354, "duration": 1800},
                     "vial": {"level": 27, "name": "Forest Dream"},
                 },
                 "right": {
                     "intensity": 70,
                     "on": False,
+                    "boost": None,
                     "vial": {"level": 84, "name": "Velvet Rose"},
                 },
             },
@@ -845,7 +847,7 @@ async def test_cury_switch_entity(
     monkeypatch.setattr(mock_rpc_device, "status", status)
     await init_integration(hass, 3)
 
-    for entity in ("left_slot", "right_slot"):
+    for entity in ("left_slot", "left_slot_boost", "right_slot", "right_slot_boost"):
         entity_id = f"{SWITCH_DOMAIN}.test_name_{entity}"
 
         state = hass.states.get(entity_id)
@@ -872,6 +874,24 @@ async def test_cury_switch_entity(
     mock_rpc_device.mock_update()
     mock_rpc_device.cury_set.assert_called_with(0, "right", True)
 
+    await hass.services.async_call(
+        SWITCH_DOMAIN,
+        SERVICE_TURN_OFF,
+        {ATTR_ENTITY_ID: "switch.test_name_left_slot_boost"},
+        blocking=True,
+    )
+    mock_rpc_device.mock_update()
+    mock_rpc_device.cury_stop_boost.assert_called_with(0, "left")
+
+    await hass.services.async_call(
+        SWITCH_DOMAIN,
+        SERVICE_TURN_ON,
+        {ATTR_ENTITY_ID: "switch.test_name_right_slot_boost"},
+        blocking=True,
+    )
+    mock_rpc_device.mock_update()
+    mock_rpc_device.cury_boost.assert_called_with(0, "right")
+
 
 async def test_cury_switch_availability(
     hass: HomeAssistant,
@@ -883,11 +903,13 @@ async def test_cury_switch_availability(
         "left": {
             "intensity": 70,
             "on": True,
+            "boost": None,
             "vial": {"level": 27, "name": "Forest Dream"},
         },
         "right": {
             "intensity": 70,
             "on": False,
+            "boost": None,
             "vial": {"level": 84, "name": "Velvet Rose"},
         },
     }
@@ -924,6 +946,7 @@ async def test_cury_switch_availability(
     slots["left"] = {
         "intensity": 70,
         "on": True,
+        "boost": None,
         "vial": {"level": 27, "name": "Forest Dream"},
     }
     mutate_rpc_device_status(monkeypatch, mock_rpc_device, "cury:0", "slots", slots)
