diff --git a/homeassistant/components/wemo/__init__.py b/homeassistant/components/wemo/__init__.py index 24cf7d66a0a..8d75b9bddae 100644 --- a/homeassistant/components/wemo/__init__.py +++ b/homeassistant/components/wemo/__init__.py @@ -9,13 +9,8 @@ import pywemo import voluptuous as vol from homeassistant import config_entries -from homeassistant.components.binary_sensor import DOMAIN as BINARY_SENSOR_DOMAIN -from homeassistant.components.fan import DOMAIN as FAN_DOMAIN -from homeassistant.components.light import DOMAIN as LIGHT_DOMAIN -from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN -from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN from homeassistant.config_entries import ConfigEntry -from homeassistant.const import CONF_DISCOVERY, EVENT_HOMEASSISTANT_STOP +from homeassistant.const import CONF_DISCOVERY, EVENT_HOMEASSISTANT_STOP, Platform from homeassistant.core import CALLBACK_TYPE, Event, HomeAssistant, callback from homeassistant.helpers import config_validation as cv from homeassistant.helpers.dispatcher import async_dispatcher_send @@ -32,17 +27,17 @@ MAX_CONCURRENCY = 3 # Mapping from Wemo model_name to domain. WEMO_MODEL_DISPATCH = { - "Bridge": [LIGHT_DOMAIN], - "CoffeeMaker": [SWITCH_DOMAIN], - "Dimmer": [LIGHT_DOMAIN], - "Humidifier": [FAN_DOMAIN], - "Insight": [BINARY_SENSOR_DOMAIN, SENSOR_DOMAIN, SWITCH_DOMAIN], - "LightSwitch": [SWITCH_DOMAIN], - "Maker": [BINARY_SENSOR_DOMAIN, SWITCH_DOMAIN], - "Motion": [BINARY_SENSOR_DOMAIN], - "OutdoorPlug": [SWITCH_DOMAIN], - "Sensor": [BINARY_SENSOR_DOMAIN], - "Socket": [SWITCH_DOMAIN], + "Bridge": [Platform.LIGHT], + "CoffeeMaker": [Platform.SWITCH], + "Dimmer": [Platform.LIGHT], + "Humidifier": [Platform.FAN], + "Insight": [Platform.BINARY_SENSOR, Platform.SENSOR, Platform.SWITCH], + "LightSwitch": [Platform.SWITCH], + "Maker": [Platform.BINARY_SENSOR, Platform.SWITCH], + "Motion": [Platform.BINARY_SENSOR], + "OutdoorPlug": [Platform.SWITCH], + "Sensor": [Platform.BINARY_SENSOR], + "Socket": [Platform.SWITCH], } _LOGGER = logging.getLogger(__name__) @@ -155,7 +150,7 @@ class WemoDispatcher: return coordinator = await async_register_device(hass, self._config_entry, wemo) - for component in WEMO_MODEL_DISPATCH.get(wemo.model_name, [SWITCH_DOMAIN]): + for component in WEMO_MODEL_DISPATCH.get(wemo.model_name, [Platform.SWITCH]): # Three cases: # - First time we see component, we need to load it and initialize the backlog # - Component is being loaded, add to backlog diff --git a/tests/components/wemo/test_fan.py b/tests/components/wemo/test_fan.py index dc450311e6a..a00d5523678 100644 --- a/tests/components/wemo/test_fan.py +++ b/tests/components/wemo/test_fan.py @@ -3,14 +3,13 @@ import pytest from pywemo.exceptions import ActionException -from homeassistant.components.fan import DOMAIN as FAN_DOMAIN from homeassistant.components.homeassistant import ( DOMAIN as HA_DOMAIN, SERVICE_UPDATE_ENTITY, ) from homeassistant.components.wemo import fan from homeassistant.components.wemo.const import DOMAIN -from homeassistant.const import ATTR_ENTITY_ID, STATE_OFF, STATE_ON +from homeassistant.const import ATTR_ENTITY_ID, STATE_OFF, STATE_ON, Platform from homeassistant.setup import async_setup_component from . import entity_test_helpers @@ -85,7 +84,7 @@ async def test_available_after_update( pywemo_device.set_state.side_effect = ActionException pywemo_device.get_state.return_value = 1 await entity_test_helpers.test_avaliable_after_update( - hass, pywemo_registry, pywemo_device, wemo_entity, FAN_DOMAIN + hass, pywemo_registry, pywemo_device, wemo_entity, Platform.FAN ) diff --git a/tests/components/wemo/test_light_bridge.py b/tests/components/wemo/test_light_bridge.py index 9f622a22c1d..1d316d93c8f 100644 --- a/tests/components/wemo/test_light_bridge.py +++ b/tests/components/wemo/test_light_bridge.py @@ -8,8 +8,8 @@ from homeassistant.components.homeassistant import ( DOMAIN as HA_DOMAIN, SERVICE_UPDATE_ENTITY, ) -from homeassistant.components.light import ATTR_COLOR_TEMP, DOMAIN as LIGHT_DOMAIN -from homeassistant.const import ATTR_ENTITY_ID, STATE_OFF, STATE_ON +from homeassistant.components.light import ATTR_COLOR_TEMP +from homeassistant.const import ATTR_ENTITY_ID, STATE_OFF, STATE_ON, Platform from homeassistant.setup import async_setup_component from . import entity_test_helpers @@ -76,7 +76,7 @@ async def test_available_after_update( pywemo_bridge_light.turn_on.side_effect = pywemo.exceptions.ActionException pywemo_bridge_light.state["onoff"] = 1 await entity_test_helpers.test_avaliable_after_update( - hass, pywemo_registry, pywemo_device, wemo_entity, LIGHT_DOMAIN + hass, pywemo_registry, pywemo_device, wemo_entity, Platform.LIGHT ) diff --git a/tests/components/wemo/test_light_dimmer.py b/tests/components/wemo/test_light_dimmer.py index 830eb6dbdf4..0460994aa86 100644 --- a/tests/components/wemo/test_light_dimmer.py +++ b/tests/components/wemo/test_light_dimmer.py @@ -7,8 +7,7 @@ from homeassistant.components.homeassistant import ( DOMAIN as HA_DOMAIN, SERVICE_UPDATE_ENTITY, ) -from homeassistant.components.light import DOMAIN as LIGHT_DOMAIN -from homeassistant.const import ATTR_ENTITY_ID, STATE_OFF, STATE_ON +from homeassistant.const import ATTR_ENTITY_ID, STATE_OFF, STATE_ON, Platform from homeassistant.setup import async_setup_component from . import entity_test_helpers @@ -41,7 +40,7 @@ async def test_available_after_update( pywemo_device.on.side_effect = ActionException pywemo_device.get_state.return_value = 1 await entity_test_helpers.test_avaliable_after_update( - hass, pywemo_registry, pywemo_device, wemo_entity, LIGHT_DOMAIN + hass, pywemo_registry, pywemo_device, wemo_entity, Platform.LIGHT ) diff --git a/tests/components/wemo/test_switch.py b/tests/components/wemo/test_switch.py index 1023498c792..963c662b124 100644 --- a/tests/components/wemo/test_switch.py +++ b/tests/components/wemo/test_switch.py @@ -7,8 +7,7 @@ from homeassistant.components.homeassistant import ( DOMAIN as HA_DOMAIN, SERVICE_UPDATE_ENTITY, ) -from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN -from homeassistant.const import ATTR_ENTITY_ID, STATE_OFF, STATE_ON +from homeassistant.const import ATTR_ENTITY_ID, STATE_OFF, STATE_ON, Platform from homeassistant.setup import async_setup_component from . import entity_test_helpers @@ -83,5 +82,5 @@ async def test_available_after_update( pywemo_device.on.side_effect = ActionException pywemo_device.get_state.return_value = 1 await entity_test_helpers.test_avaliable_after_update( - hass, pywemo_registry, pywemo_device, wemo_entity, SWITCH_DOMAIN + hass, pywemo_registry, pywemo_device, wemo_entity, Platform.SWITCH )