evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use light enums in tests (#70801)

Browse source
This commit is contained in:
epenet 2022-04-26 16:32:00 +02:00 • committed by GitHub
parent cad2ba4d44
commit 7d2363ad22
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
19 changed files with 225 additions and 258 deletions
Download patch file Download diff file Expand all files Collapse all files

9
tests/components/blebox/test_light.py
View file

@ -9,8 +9,7 @@ from homeassistant.components.light import (
ATTR_BRIGHTNESS,
ATTR_RGBW_COLOR,
ATTR_SUPPORTED_COLOR_MODES,
COLOR_MODE_BRIGHTNESS,
COLOR_MODE_RGBW,
ColorMode,
)
from homeassistant.const import (
SERVICE_TURN_OFF,
@ -58,7 +57,7 @@ async def test_dimmer_init(dimmer, hass, config):
assert state.name == "dimmerBox-brightness"
color_modes = state.attributes[ATTR_SUPPORTED_COLOR_MODES]
assert color_modes == [COLOR_MODE_BRIGHTNESS]
assert color_modes == [ColorMode.BRIGHTNESS]
assert state.attributes[ATTR_BRIGHTNESS] == 65
assert state.state == STATE_ON
@ -229,7 +228,7 @@ async def test_wlightbox_s_init(wlightbox_s, hass, config):
assert state.name == "wLightBoxS-color"
color_modes = state.attributes[ATTR_SUPPORTED_COLOR_MODES]
assert color_modes == [COLOR_MODE_BRIGHTNESS]
assert color_modes == [ColorMode.BRIGHTNESS]
assert ATTR_BRIGHTNESS not in state.attributes
assert state.state == STATE_UNKNOWN
@ -329,7 +328,7 @@ async def test_wlightbox_init(wlightbox, hass, config):
assert state.name == "wLightBox-color"
color_modes = state.attributes[ATTR_SUPPORTED_COLOR_MODES]
assert color_modes == [COLOR_MODE_RGBW]
assert color_modes == [ColorMode.RGBW]
assert ATTR_BRIGHTNESS not in state.attributes
assert ATTR_RGBW_COLOR not in state.attributes