Add switch as entity platform on MQTT subentries

Author: jbouwh

homeassistant/components/mqtt/config_flow.py

@@ -33,6 +33,7 @@
     SensorDeviceClass,
     SensorStateClass,
 )
+from homeassistant.components.switch import SwitchDeviceClass
 from homeassistant.config_entries import (
     SOURCE_RECONFIGURE,
     ConfigEntry,
@@ -55,6 +56,7 @@
     CONF_DISCOVERY,
     CONF_HOST,
     CONF_NAME,
+    CONF_OPTIMISTIC,
     CONF_PASSWORD,
     CONF_PAYLOAD,
     CONF_PLATFORM,
@@ -234,7 +236,7 @@
 
 # Subentry selectors
 RESET_IF_EMPTY = {CONF_OPTIONS}
-SUBENTRY_PLATFORMS = [Platform.NOTIFY, Platform.SENSOR]
+SUBENTRY_PLATFORMS = [Platform.NOTIFY, Platform.SENSOR, Platform.SWITCH]
 SUBENTRY_PLATFORM_SELECTOR = SelectSelector(
     SelectSelectorConfig(
         options=[platform.value for platform in SUBENTRY_PLATFORMS],
@@ -288,6 +290,15 @@
     NumberSelectorConfig(mode=NumberSelectorMode.BOX, min=0)
 )
 
+# Switch specific selectors
+SWITCH_DEVICE_CLASS_SELECTOR = SelectSelector(
+    SelectSelectorConfig(
+        options=[device_class.value for device_class in SwitchDeviceClass],
+        mode=SelectSelectorMode.DROPDOWN,
+        translation_key="device_class_switch",
+    )
+)
+
 
 @dataclass(frozen=True)
 class PlatformField:
@@ -343,6 +354,9 @@ def unit_of_measurement_selector(user_data: dict[str, Any | None]) -> Selector:
         ),
         CONF_OPTIONS: PlatformField(OPTIONS_SELECTOR, False, cv.ensure_list),
     },
+    Platform.SWITCH.value: {
+        CONF_DEVICE_CLASS: PlatformField(SWITCH_DEVICE_CLASS_SELECTOR, False, str),
+    },
 }
 PLATFORM_MQTT_FIELDS = {
     Platform.NOTIFY.value: {
@@ -366,12 +380,29 @@ def unit_of_measurement_selector(user_data: dict[str, Any | None]) -> Selector:
         ),
         CONF_EXPIRE_AFTER: PlatformField(EXIRE_AFTER_SELECTOR, False, cv.positive_int),
     },
+    Platform.SWITCH.value: {
+        CONF_COMMAND_TOPIC: PlatformField(
+            TEXT_SELECTOR, True, valid_publish_topic, "invalid_publish_topic"
+        ),
+        CONF_COMMAND_TEMPLATE: PlatformField(
+            TEMPLATE_SELECTOR, False, cv.template, "invalid_template"
+        ),
+        CONF_STATE_TOPIC: PlatformField(
+            TEXT_SELECTOR, False, valid_subscribe_topic, "invalid_subscribe_topic"
+        ),
+        CONF_VALUE_TEMPLATE: PlatformField(
+            TEMPLATE_SELECTOR, False, cv.template, "invalid_template"
+        ),
+        CONF_RETAIN: PlatformField(BOOLEAN_SELECTOR, False, bool),
+        CONF_OPTIMISTIC: PlatformField(BOOLEAN_SELECTOR, False, bool),
+    },
 }
 ENTITY_CONFIG_VALIDATOR: dict[
     str, Callable[[dict[str, Any], dict[str, str]], dict[str, Any]] | None
 ] = {
     Platform.NOTIFY.value: None,
     Platform.SENSOR.value: validate_sensor_state_and_device_class_config,
+    Platform.SWITCH.value: None,
 }
 
 MQTT_DEVICE_SCHEMA = vol.Schema(

homeassistant/components/mqtt/strings.json

@@ -227,6 +227,7 @@
             "last_reset_value_template": "Last reset value template",
             "expire_after": "Expire after",
             "force_update": "Force update",
+            "optimistic": "Optimistic",
             "retain": "Retain",
             "qos": "QoS"
           },
@@ -238,6 +239,7 @@
             "last_reset_value_template": "Defines a [template](https://www.home-assistant.io/docs/configuration/templating/#using-value-templates-with-mqtt) to extract the last reset. When Last reset temaplate is set, the State class option must be Total.  [Learn more..]({url}#last_reset_value_template)",
             "expire_after": "If set, it defines the number of seconds after the sensor’s state expires, if it’s not updated. After expiry, the sensor’s state becomes unavailable. Default the sensors state never expires. [Learn more..]({url}#expire_after)",
             "force_update": "Sends update events even if the value hasn’t changed. Useful if you want to have meaningful value graphs in history. [Learn more..]({url}#force_update)",
+            "optimistic": "Flag that defines if the {platform} entity works in optimistic mode. [Learn more..]({url}#optimistic)",
             "retain": "Select if values published by the {platform} entity should be retained at the MQTT broker.",
             "qos": "The QoS value {platform} entity should use."
           }
@@ -434,10 +436,17 @@
         "wind_speed": "[%key:component::sensor::entity_component::wind_speed::name%]"
       }
     },
+    "device_class_switch": {
+      "options": {
+        "outlet": "[%key:component::switch::entity_component::outlet::name%]",
+        "switch": "[%key:component::switch::title%]"
+      }
+    },
     "platform": {
       "options": {
         "notify": "Notify",
-        "sensor": "Sensor"
+        "sensor": "Sensor",
+        "switch": "Switch"
       }
     },
     "set_ca_cert": {

tests/components/mqtt/common.py

@@ -119,6 +119,20 @@
         "qos": 0,
     },
 }
+MOCK_SUBENTRY_SWITCH_COMPONENT = {
+    "3faf1318016c46c5aea26707eeb6f12e": {
+        "platform": "switch",
+        "name": "Outlet",
+        "device_class": "outlet",
+        "qos": 0,
+        "command_topic": "test-topic",
+        "state_topic": "test-topic",
+        "command_template": "{{ value }}",
+        "value_template": "{{ value_json.value }}",
+        "entity_picture": "https://example.com/3faf1318016c46c5aea26707eeb6f12e",
+        "optimistic": True,
+    },
+}
 
 # Bogus light component just for code coverage
 # Note that light cannot be setup through the UI yet
@@ -207,6 +221,17 @@
     },
     "components": MOCK_SUBENTRY_SENSOR_COMPONENT_STATE_CLASS,
 }
+MOCK_SWITCH_SUBENTRY_DATA_SINGLE_STATE_CLASS = {
+    "device": {
+        "name": "Test switch",
+        "sw_version": "1.0",
+        "hw_version": "2.1 rev a",
+        "model": "Model XL",
+        "model_id": "mn002",
+        "configuration_url": "https://example.com",
+    },
+    "components": MOCK_SUBENTRY_SWITCH_COMPONENT,
+}
 
 MOCK_SUBENTRY_DATA_BAD_COMPONENT_SCHEMA = {
     "device": {
@@ -230,7 +255,8 @@
     },
     "components": MOCK_SUBENTRY_NOTIFY_COMPONENT1
     | MOCK_SUBENTRY_NOTIFY_COMPONENT2
-    | MOCK_SUBENTRY_LIGHT_COMPONENT,
+    | MOCK_SUBENTRY_LIGHT_COMPONENT
+    | MOCK_SUBENTRY_SWITCH_COMPONENT,
 } | MOCK_SUBENTRY_AVAILABILITY_DATA
 _SENTINEL = object()
 

tests/components/mqtt/test_config_flow.py

@@ -38,6 +38,7 @@
     MOCK_NOTIFY_SUBENTRY_DATA_SINGLE,
     MOCK_SENSOR_SUBENTRY_DATA_SINGLE,
     MOCK_SENSOR_SUBENTRY_DATA_SINGLE_STATE_CLASS,
+    MOCK_SWITCH_SUBENTRY_DATA_SINGLE_STATE_CLASS,
 )
 
 from tests.common import MockConfigEntry, MockMqttReasonCode
@@ -2725,12 +2726,41 @@ async def test_migrate_of_incompatible_config_entry(
             ),
             "Test sensor Energy",
         ),
+        (
+            MOCK_SWITCH_SUBENTRY_DATA_SINGLE_STATE_CLASS,
+            {"name": "Outlet"},
+            {"device_class": "outlet"},
+            (),
+            {
+                "command_topic": "test-topic",
+                "command_template": "{{ value }}",
+                "state_topic": "test-topic",
+                "value_template": "{{ value_json.value }}",
+                "qos": 0,
+                "optimistic": True,
+            },
+            (
+                (
+                    {"command_topic": "test-topic#invalid"},
+                    {"command_topic": "invalid_publish_topic"},
+                ),
+                (
+                    {
+                        "command_topic": "test-topic",
+                        "state_topic": "test-topic#invalid",
+                    },
+                    {"state_topic": "invalid_subscribe_topic"},
+                ),
+            ),
+            "Test switch Outlet",
+        ),
     ],
     ids=[
         "notify_with_entity_name",
         "notify_no_entity_name",
         "sensor_options",
         "sensor_total",
+        "switch",
     ],
 )
 async def test_subentry_configflow(