Adapt tests to newly added switch platform

home-assistant · Jan 11, 2023 · 45d75ed · 45d75ed
1 parent 7f6d0b7
commit 45d75ed
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/components/devolo_home_network/test_init.py b/tests/components/devolo_home_network/test_init.py
@@ -8,6 +8,7 @@
 from homeassistant.components.device_tracker import DOMAIN as DEVICE_TRACKER
 from homeassistant.components.devolo_home_network.const import DOMAIN
 from homeassistant.components.sensor import DOMAIN as SENSOR
+from homeassistant.components.switch import DOMAIN as SWITCH
 from homeassistant.config_entries import ConfigEntryState
 from homeassistant.const import CONF_IP_ADDRESS, EVENT_HOMEASSISTANT_STOP
 from homeassistant.core import HomeAssistant
@@ -82,9 +83,9 @@ async def test_hass_stop(hass: HomeAssistant, mock_device: MockDevice):
 @pytest.mark.parametrize(
     "device, expected_platforms",
     [
-        ["mock_device", (BINARY_SENSOR, DEVICE_TRACKER, SENSOR)],
-        ["mock_repeater_device", (DEVICE_TRACKER, SENSOR)],
-        ["mock_nonwifi_device", (BINARY_SENSOR, SENSOR)],
+        ["mock_device", (BINARY_SENSOR, DEVICE_TRACKER, SENSOR, SWITCH)],
+        ["mock_repeater_device", (DEVICE_TRACKER, SENSOR, SWITCH)],
+        ["mock_nonwifi_device", (BINARY_SENSOR, SENSOR, SWITCH)],
     ],
 )
 async def test_platforms(