Update vacuum entity constants for Alexa tests (#98336)

* Update vacuum entity constants for Alexa tests

* Import VacuumEntityFeature
This commit is contained in:
Jan Bouwhuis 2023-08-13 15:35:00 +02:00 committed by GitHub
parent b36681b318
commit fa6ffd994a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

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