Support MyStrom switch 120 (#96535)

home-assistant · Jul 14, 2023 · 1b7632a · 1b7632a
1 parent 357af58
commit 1b7632a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/homeassistant/components/mystrom/__init__.py b/homeassistant/components/mystrom/__init__.py
@@ -53,7 +53,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     info.setdefault("type", 101)
 
     device_type = info["type"]
-    if device_type in [101, 106, 107]:
+    if device_type in [101, 106, 107, 120]:
         device = _get_mystrom_switch(host)
         platforms = PLATFORMS_SWITCH
         await _async_get_device_state(device, info["ip"])
@@ -86,7 +86,7 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     """Unload a config entry."""
     device_type = hass.data[DOMAIN][entry.entry_id].info["type"]
     platforms = []
-    if device_type in [101, 106, 107]:
+    if device_type in [101, 106, 107, 120]:
         platforms.extend(PLATFORMS_SWITCH)
     elif device_type in [102, 105]:
         platforms.extend(PLATFORMS_BULB)

diff --git a/tests/components/mystrom/test_init.py b/tests/components/mystrom/test_init.py
@@ -68,7 +68,7 @@ async def test_init_switch_and_unload(
         (110, "sensor", ConfigEntryState.SETUP_ERROR, True),
         (113, "switch", ConfigEntryState.SETUP_ERROR, True),
         (118, "button", ConfigEntryState.SETUP_ERROR, True),
-        (120, "switch", ConfigEntryState.SETUP_ERROR, True),
+        (120, "switch", ConfigEntryState.LOADED, False),
     ],
 )
 async def test_init_bulb(