Add initial support for Tuya sjz category (electric desk) (home-assistant#152036)

Author: epenet

homeassistant/components/tuya/select.py

@@ -248,6 +248,14 @@
             entity_category=EntityCategory.CONFIG,
         ),
     ),
+    # Electric desk
+    "sjz": (
+        SelectEntityDescription(
+            key=DPCode.LEVEL,
+            translation_key="desk_level",
+            entity_category=EntityCategory.CONFIG,
+        ),
+    ),
     # Smart Camera
     # https://developer.tuya.com/en/docs/iot/categorysp?id=Kaiuz35leyo12
     "sp": (

homeassistant/components/tuya/strings.json

@@ -509,6 +509,15 @@
           "interim": "Interim"
         }
       },
+      "desk_level": {
+        "name": "Level",
+        "state": {
+          "level_1": "Level 1",
+          "level_2": "Level 2",
+          "level_3": "Level 3",
+          "level_4": "Level 4"
+        }
+      },
       "inverter_work_mode": {
         "name": "Inverter work mode",
         "state": {

homeassistant/components/tuya/switch.py

@@ -662,6 +662,14 @@
             entity_category=EntityCategory.CONFIG,
         ),
     ),
+    # Electric desk
+    "sjz": (
+        SwitchEntityDescription(
+            key=DPCode.CHILD_LOCK,
+            translation_key="child_lock",
+            entity_category=EntityCategory.CONFIG,
+        ),
+    ),
     # Smart Camera
     # https://developer.tuya.com/en/docs/iot/categorysp?id=Kaiuz35leyo12
     "sp": (

tests/components/tuya/snapshots/test_init.ambr

@@ -6716,7 +6716,7 @@
     'labels': set({
     }),
     'manufacturer': 'Tuya',
-    'model': 'geniodesk (unsupported)',
+    'model': 'geniodesk',
     'model_id': 'ftbc8rp8ipksdfpv',
     'name': 'mesa',
     'name_by_user': None,

tests/components/tuya/snapshots/test_select.ambr

@@ -3136,6 +3136,67 @@
     'state': 'power_on',
   })
 # ---
+# name: test_platform_setup_and_discovery[select.mesa_level-entry]
+  EntityRegistryEntrySnapshot({
+    'aliases': set({
+    }),
+    'area_id': None,
+    'capabilities': dict({
+      'options': list([
+        'level_1',
+        'level_2',
+        'level_3',
+        'level_4',
+      ]),
+    }),
+    'config_entry_id': <ANY>,
+    'config_subentry_id': <ANY>,
+    'device_class': None,
+    'device_id': <ANY>,
+    'disabled_by': None,
+    'domain': 'select',
+    'entity_category': <EntityCategory.CONFIG: 'config'>,
+    'entity_id': 'select.mesa_level',
+    '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': 'Level',
+    'platform': 'tuya',
+    'previous_unique_id': None,
+    'suggested_object_id': None,
+    'supported_features': 0,
+    'translation_key': 'desk_level',
+    'unique_id': 'tuya.vpfdskpi8pr8cbtfzjslevel',
+    'unit_of_measurement': None,
+  })
+# ---
+# name: test_platform_setup_and_discovery[select.mesa_level-state]
+  StateSnapshot({
+    'attributes': ReadOnlyDict({
+      'friendly_name': 'mesa Level',
+      'options': list([
+        'level_1',
+        'level_2',
+        'level_3',
+        'level_4',
+      ]),
+    }),
+    'context': <ANY>,
+    'entity_id': 'select.mesa_level',
+    'last_changed': <ANY>,
+    'last_reported': <ANY>,
+    'last_updated': <ANY>,
+    'state': 'level_1',
+  })
+# ---
 # name: test_platform_setup_and_discovery[select.mirilla_puerta_anti_flicker-entry]
   EntityRegistryEntrySnapshot({
     'aliases': set({

tests/components/tuya/snapshots/test_switch.ambr

@@ -5513,6 +5513,54 @@
     'state': 'on',
   })
 # ---
+# name: test_platform_setup_and_discovery[switch.mesa_child_lock-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': <EntityCategory.CONFIG: 'config'>,
+    'entity_id': 'switch.mesa_child_lock',
+    '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': 'Child lock',
+    'platform': 'tuya',
+    'previous_unique_id': None,
+    'suggested_object_id': None,
+    'supported_features': 0,
+    'translation_key': 'child_lock',
+    'unique_id': 'tuya.vpfdskpi8pr8cbtfzjschild_lock',
+    'unit_of_measurement': None,
+  })
+# ---
+# name: test_platform_setup_and_discovery[switch.mesa_child_lock-state]
+  StateSnapshot({
+    'attributes': ReadOnlyDict({
+      'friendly_name': 'mesa Child lock',
+    }),
+    'context': <ANY>,
+    'entity_id': 'switch.mesa_child_lock',
+    'last_changed': <ANY>,
+    'last_reported': <ANY>,
+    'last_updated': <ANY>,
+    'state': 'off',
+  })
+# ---
 # name: test_platform_setup_and_discovery[switch.mesh_gateway_mute-entry]
   EntityRegistryEntrySnapshot({
     'aliases': set({