Add model ID to Wemo (home-assistant#125368)

eduherraiz · Sep 5, 2024 · 0677a25 · 0677a25
1 parent 97ffbf5
commit 0677a25
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 0 deletions.
diff --git a/homeassistant/components/wemo/coordinator.py b/homeassistant/components/wemo/coordinator.py
@@ -275,6 +275,7 @@ def _device_info(wemo: WeMoDevice) -> DeviceInfo:
         identifiers={(DOMAIN, wemo.serial_number)},
         manufacturer="Belkin",
         model=wemo.model_name,
+        model_id=wemo.model,
         name=wemo.name,
         sw_version=wemo.firmware_version,
     )

diff --git a/tests/components/wemo/conftest.py b/tests/components/wemo/conftest.py
@@ -65,6 +65,7 @@ def create_pywemo_device(
     device.name = MOCK_NAME
     device.serial_number = MOCK_SERIAL_NUMBER
     device.model_name = pywemo_model.replace("LongPress", "")
+    device.model = device.model_name
     device.udn = f"uuid:{device.model_name}-1_0-{device.serial_number}"
     device.firmware_version = MOCK_FIRMWARE_VERSION
     device.get_state.return_value = 0  # Default to Off

diff --git a/tests/components/wemo/test_coordinator.py b/tests/components/wemo/test_coordinator.py
@@ -178,6 +178,7 @@ async def test_device_info(
     }
     assert device_entries[0].manufacturer == "Belkin"
     assert device_entries[0].model == "LightSwitch"
+    assert device_entries[0].model_id == "LightSwitch"
     assert device_entries[0].sw_version == MOCK_FIRMWARE_VERSION
 
 

diff --git a/tests/components/wemo/test_init.py b/tests/components/wemo/test_init.py
@@ -201,6 +201,7 @@ def create_device(counter):
         device.name = f"{MOCK_NAME}_{counter}"
         device.serial_number = f"{MOCK_SERIAL_NUMBER}_{counter}"
         device.model_name = "Motion"
+        device.model = "Motion"
         device.udn = f"uuid:{device.model_name}-1_0-{device.serial_number}"
         device.firmware_version = MOCK_FIRMWARE_VERSION
         device.get_state.return_value = 0  # Default to Off