diff --git a/tests/components/alexa/test_smart_home.py b/tests/components/alexa/test_smart_home.py index 708b06bab2b..c42ea0a0f6a 100644 --- a/tests/components/alexa/test_smart_home.py +++ b/tests/components/alexa/test_smart_home.py @@ -8,7 +8,7 @@ from homeassistant.components.alexa import smart_home, state_report import homeassistant.components.camera as camera from homeassistant.components.cover import CoverDeviceClass from homeassistant.components.media_player import MediaPlayerEntityFeature -import homeassistant.components.vacuum as vacuum +from homeassistant.components.vacuum import VacuumEntityFeature from homeassistant.config import async_process_ha_core_config from homeassistant.const import STATE_UNKNOWN, UnitOfTemperature from homeassistant.core import Context, Event, HomeAssistant @@ -3872,12 +3872,12 @@ async def test_vacuum_discovery(hass: HomeAssistant) -> None: "docked", { "friendly_name": "Test vacuum 1", - "supported_features": vacuum.SUPPORT_TURN_ON - | vacuum.SUPPORT_TURN_OFF - | vacuum.SUPPORT_START - | vacuum.SUPPORT_STOP - | vacuum.SUPPORT_RETURN_HOME - | vacuum.SUPPORT_PAUSE, + "supported_features": VacuumEntityFeature.TURN_ON + | VacuumEntityFeature.TURN_OFF + | VacuumEntityFeature.START + | VacuumEntityFeature.STOP + | VacuumEntityFeature.RETURN_HOME + | VacuumEntityFeature.PAUSE, }, ) appliance = await discovery_test(device, hass) @@ -3913,12 +3913,12 @@ async def test_vacuum_fan_speed(hass: HomeAssistant) -> None: "cleaning", { "friendly_name": "Test vacuum 2", - "supported_features": vacuum.SUPPORT_TURN_ON - | vacuum.SUPPORT_TURN_OFF - | vacuum.SUPPORT_START - | vacuum.SUPPORT_STOP - | vacuum.SUPPORT_PAUSE - | vacuum.SUPPORT_FAN_SPEED, + "supported_features": VacuumEntityFeature.TURN_ON + | VacuumEntityFeature.TURN_OFF + | VacuumEntityFeature.START + | VacuumEntityFeature.STOP + | VacuumEntityFeature.PAUSE + | VacuumEntityFeature.FAN_SPEED, "fan_speed_list": ["off", "low", "medium", "high", "turbo", "super_sucker"], "fan_speed": "medium", }, @@ -4042,12 +4042,12 @@ async def test_vacuum_pause(hass: HomeAssistant) -> None: "cleaning", { "friendly_name": "Test vacuum 3", - "supported_features": vacuum.SUPPORT_TURN_ON - | vacuum.SUPPORT_TURN_OFF - | vacuum.SUPPORT_START - | vacuum.SUPPORT_STOP - | vacuum.SUPPORT_PAUSE - | vacuum.SUPPORT_FAN_SPEED, + "supported_features": VacuumEntityFeature.TURN_ON + | VacuumEntityFeature.TURN_OFF + | VacuumEntityFeature.START + | VacuumEntityFeature.STOP + | VacuumEntityFeature.PAUSE + | VacuumEntityFeature.FAN_SPEED, "fan_speed_list": ["off", "low", "medium", "high", "turbo", "super_sucker"], "fan_speed": "medium", }, @@ -4080,12 +4080,12 @@ async def test_vacuum_resume(hass: HomeAssistant) -> None: "docked", { "friendly_name": "Test vacuum 4", - "supported_features": vacuum.SUPPORT_TURN_ON - | vacuum.SUPPORT_TURN_OFF - | vacuum.SUPPORT_START - | vacuum.SUPPORT_STOP - | vacuum.SUPPORT_PAUSE - | vacuum.SUPPORT_FAN_SPEED, + "supported_features": VacuumEntityFeature.TURN_ON + | VacuumEntityFeature.TURN_OFF + | VacuumEntityFeature.START + | VacuumEntityFeature.STOP + | VacuumEntityFeature.PAUSE + | VacuumEntityFeature.FAN_SPEED, "fan_speed_list": ["off", "low", "medium", "high", "turbo", "super_sucker"], "fan_speed": "medium", }, @@ -4108,9 +4108,9 @@ async def test_vacuum_discovery_no_turn_on(hass: HomeAssistant) -> None: "cleaning", { "friendly_name": "Test vacuum 5", - "supported_features": vacuum.SUPPORT_TURN_OFF - | vacuum.SUPPORT_START - | vacuum.SUPPORT_RETURN_HOME, + "supported_features": VacuumEntityFeature.TURN_OFF + | VacuumEntityFeature.START + | VacuumEntityFeature.RETURN_HOME, }, ) appliance = await discovery_test(device, hass) @@ -4138,9 +4138,9 @@ async def test_vacuum_discovery_no_turn_off(hass: HomeAssistant) -> None: "cleaning", { "friendly_name": "Test vacuum 6", - "supported_features": vacuum.SUPPORT_TURN_ON - | vacuum.SUPPORT_START - | vacuum.SUPPORT_RETURN_HOME, + "supported_features": VacuumEntityFeature.TURN_ON + | VacuumEntityFeature.START + | VacuumEntityFeature.RETURN_HOME, }, ) appliance = await discovery_test(device, hass) @@ -4169,7 +4169,8 @@ async def test_vacuum_discovery_no_turn_on_or_off(hass: HomeAssistant) -> None: "cleaning", { "friendly_name": "Test vacuum 7", - "supported_features": vacuum.SUPPORT_START | vacuum.SUPPORT_RETURN_HOME, + "supported_features": VacuumEntityFeature.START + | VacuumEntityFeature.RETURN_HOME, }, ) appliance = await discovery_test(device, hass)