Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 12:59:34 +00:00
Code Activity

Cleanup EntityFeature in tests (#78859)

Browse Source
This commit is contained in:
epenet
2022-09-21 10:48:55 +02:00
committed by GitHub
parent c88a874063
commit 0ac581a0b1
37 changed files with 472 additions and 472 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
tests/components/blebox/test_cover.py
View File

@@ -12,11 +12,8 @@ from homeassistant.components.cover import (
STATE_CLOSING,
STATE_OPEN,
STATE_OPENING,
SUPPORT_CLOSE,
SUPPORT_OPEN,
SUPPORT_SET_POSITION,
SUPPORT_STOP,
CoverDeviceClass,
CoverEntityFeature,
)
from homeassistant.const import (
ATTR_DEVICE_CLASS,
@@ -107,11 +104,11 @@ async def test_init_gatecontroller(gatecontroller, hass, config):
assert state.attributes[ATTR_DEVICE_CLASS] == CoverDeviceClass.GATE
supported_features = state.attributes[ATTR_SUPPORTED_FEATURES]
assert supported_features & SUPPORT_OPEN
assert supported_features & SUPPORT_CLOSE
assert supported_features & SUPPORT_STOP
assert supported_features & CoverEntityFeature.OPEN
assert supported_features & CoverEntityFeature.CLOSE
assert supported_features & CoverEntityFeature.STOP
assert supported_features & SUPPORT_SET_POSITION
assert supported_features & CoverEntityFeature.SET_POSITION
assert ATTR_CURRENT_POSITION not in state.attributes
assert state.state == STATE_UNKNOWN
@@ -137,11 +134,11 @@ async def test_init_shutterbox(shutterbox, hass, config):
assert entry.original_device_class == CoverDeviceClass.SHUTTER
supported_features = state.attributes[ATTR_SUPPORTED_FEATURES]
assert supported_features & SUPPORT_OPEN
assert supported_features & SUPPORT_CLOSE
assert supported_features & SUPPORT_STOP
assert supported_features & CoverEntityFeature.OPEN
assert supported_features & CoverEntityFeature.CLOSE
assert supported_features & CoverEntityFeature.STOP
assert supported_features & SUPPORT_SET_POSITION
assert supported_features & CoverEntityFeature.SET_POSITION
assert ATTR_CURRENT_POSITION not in state.attributes
assert state.state == STATE_UNKNOWN
@@ -167,13 +164,13 @@ async def test_init_gatebox(gatebox, hass, config):
assert state.attributes[ATTR_DEVICE_CLASS] == CoverDeviceClass.DOOR
supported_features = state.attributes[ATTR_SUPPORTED_FEATURES]
assert supported_features & SUPPORT_OPEN
assert supported_features & SUPPORT_CLOSE
assert supported_features & CoverEntityFeature.OPEN
assert supported_features & CoverEntityFeature.CLOSE
# Not available during init since requires fetching state to detect
assert not supported_features & SUPPORT_STOP
assert not supported_features & CoverEntityFeature.STOP
assert not supported_features & SUPPORT_SET_POSITION
assert not supported_features & CoverEntityFeature.SET_POSITION
assert ATTR_CURRENT_POSITION not in state.attributes
assert state.state == STATE_UNKNOWN
@@ -350,7 +347,7 @@ async def test_with_stop(gatebox, hass, config):
state = hass.states.get(entity_id)
supported_features = state.attributes[ATTR_SUPPORTED_FEATURES]
assert supported_features & SUPPORT_STOP
assert supported_features & CoverEntityFeature.STOP
async def test_with_no_stop(gatebox, hass, config):
@@ -364,7 +361,7 @@ async def test_with_no_stop(gatebox, hass, config):
state = hass.states.get(entity_id)
supported_features = state.attributes[ATTR_SUPPORTED_FEATURES]
assert not supported_features & SUPPORT_STOP
assert not supported_features & CoverEntityFeature.STOP
@pytest.mark.parametrize("feature", ALL_COVER_FIXTURES, indirect=["feature"])