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

Cleanup ColorMode in tests (#78807)

Browse source
This commit is contained in:
epenet 2022-09-20 18:33:45 +02:00 • committed by GitHub
parent 41d2ac3943
commit 6b3c91bd6a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 48 additions and 53 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -8,9 +8,8 @@ from homeassistant.components.light import (
ATTR_COLOR_TEMP, ATTR_COLOR_TEMP,
ATTR_RGB_COLOR, ATTR_RGB_COLOR,
ATTR_SUPPORTED_COLOR_MODES, ATTR_SUPPORTED_COLOR_MODES,
COLOR_MODE_COLOR_TEMP,
COLOR_MODE_HS,
DOMAIN as LIGHT_DOMAIN, DOMAIN as LIGHT_DOMAIN,
ColorMode,
) )
from homeassistant.const import ( from homeassistant.const import (
ATTR_ENTITY_ID, ATTR_ENTITY_ID,
@ -52,10 +51,10 @@ async def test_attributes(hass: HomeAssistant) -> None:
assert state.attributes.get("device_type") == "RGB Dimmer" assert state.attributes.get("device_type") == "RGB Dimmer"
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Living Room Lamp" assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Living Room Lamp"
assert state.attributes.get(ATTR_SUPPORTED_FEATURES) == 0 assert state.attributes.get(ATTR_SUPPORTED_FEATURES) == 0
assert state.attributes.get(ATTR_COLOR_MODE) == COLOR_MODE_HS assert state.attributes.get(ATTR_COLOR_MODE) == ColorMode.HS
assert state.attributes.get(ATTR_SUPPORTED_COLOR_MODES) == [ assert state.attributes.get(ATTR_SUPPORTED_COLOR_MODES) == [
COLOR_MODE_COLOR_TEMP, ColorMode.COLOR_TEMP,
COLOR_MODE_HS, ColorMode.HS,
] ]

50
tests/components/homekit/test_type_lights.py
View file

@ -27,12 +27,8 @@ from homeassistant.components.light import (
ATTR_RGBWW_COLOR, ATTR_RGBWW_COLOR,
ATTR_SUPPORTED_COLOR_MODES, ATTR_SUPPORTED_COLOR_MODES,
ATTR_WHITE, ATTR_WHITE,
COLOR_MODE_COLOR_TEMP,
COLOR_MODE_RGB,
COLOR_MODE_RGBW,
COLOR_MODE_RGBWW,
COLOR_MODE_WHITE,
DOMAIN, DOMAIN,
ColorMode,
) )
from homeassistant.const import ( from homeassistant.const import (
ATTR_ENTITY_ID, ATTR_ENTITY_ID,
@ -583,24 +579,24 @@ async def test_light_restore(hass, hk_driver, events):
"supported_color_modes, state_props, turn_on_props_with_brightness", "supported_color_modes, state_props, turn_on_props_with_brightness",
[ [
[ [
[COLOR_MODE_COLOR_TEMP, COLOR_MODE_RGBW], [ColorMode.COLOR_TEMP, ColorMode.RGBW],
{ {
ATTR_RGBW_COLOR: (128, 50, 0, 255), ATTR_RGBW_COLOR: (128, 50, 0, 255),
ATTR_RGB_COLOR: (128, 50, 0), ATTR_RGB_COLOR: (128, 50, 0),
ATTR_HS_COLOR: (23.438, 100.0), ATTR_HS_COLOR: (23.438, 100.0),
ATTR_BRIGHTNESS: 255, ATTR_BRIGHTNESS: 255,
ATTR_COLOR_MODE: COLOR_MODE_RGBW, ATTR_COLOR_MODE: ColorMode.RGBW,
}, },
{ATTR_HS_COLOR: (145, 75), ATTR_BRIGHTNESS_PCT: 25}, {ATTR_HS_COLOR: (145, 75), ATTR_BRIGHTNESS_PCT: 25},
], ],
[ [
[COLOR_MODE_COLOR_TEMP, COLOR_MODE_RGBWW], [ColorMode.COLOR_TEMP, ColorMode.RGBWW],
{ {
ATTR_RGBWW_COLOR: (128, 50, 0, 255, 255), ATTR_RGBWW_COLOR: (128, 50, 0, 255, 255),
ATTR_RGB_COLOR: (128, 50, 0), ATTR_RGB_COLOR: (128, 50, 0),
ATTR_HS_COLOR: (23.438, 100.0), ATTR_HS_COLOR: (23.438, 100.0),
ATTR_BRIGHTNESS: 255, ATTR_BRIGHTNESS: 255,
ATTR_COLOR_MODE: COLOR_MODE_RGBWW, ATTR_COLOR_MODE: ColorMode.RGBWW,
}, },
{ATTR_HS_COLOR: (145, 75), ATTR_BRIGHTNESS_PCT: 25}, {ATTR_HS_COLOR: (145, 75), ATTR_BRIGHTNESS_PCT: 25},
], ],
@ -703,24 +699,24 @@ async def test_light_rgb_with_color_temp(
"supported_color_modes, state_props, turn_on_props_with_brightness", "supported_color_modes, state_props, turn_on_props_with_brightness",
[ [
[ [
[COLOR_MODE_RGBW], [ColorMode.RGBW],
{ {
ATTR_RGBW_COLOR: (128, 50, 0, 255), ATTR_RGBW_COLOR: (128, 50, 0, 255),
ATTR_RGB_COLOR: (128, 50, 0), ATTR_RGB_COLOR: (128, 50, 0),
ATTR_HS_COLOR: (23.438, 100.0), ATTR_HS_COLOR: (23.438, 100.0),
ATTR_BRIGHTNESS: 255, ATTR_BRIGHTNESS: 255,
ATTR_COLOR_MODE: COLOR_MODE_RGBW, ATTR_COLOR_MODE: ColorMode.RGBW,
}, },
{ATTR_RGBW_COLOR: (0, 0, 0, 191)}, {ATTR_RGBW_COLOR: (0, 0, 0, 191)},
], ],
[ [
[COLOR_MODE_RGBWW], [ColorMode.RGBWW],
{ {
ATTR_RGBWW_COLOR: (128, 50, 0, 255, 255), ATTR_RGBWW_COLOR: (128, 50, 0, 255, 255),
ATTR_RGB_COLOR: (128, 50, 0), ATTR_RGB_COLOR: (128, 50, 0),
ATTR_HS_COLOR: (23.438, 100.0), ATTR_HS_COLOR: (23.438, 100.0),
ATTR_BRIGHTNESS: 255, ATTR_BRIGHTNESS: 255,
ATTR_COLOR_MODE: COLOR_MODE_RGBWW, ATTR_COLOR_MODE: ColorMode.RGBWW,
}, },
{ATTR_RGBWW_COLOR: (0, 0, 0, 165, 26)}, {ATTR_RGBWW_COLOR: (0, 0, 0, 165, 26)},
], ],
@ -800,12 +796,12 @@ async def test_light_rgb_or_w_lights(
entity_id, entity_id,
STATE_ON, STATE_ON,
{ {
ATTR_SUPPORTED_COLOR_MODES: [COLOR_MODE_RGB, COLOR_MODE_WHITE], ATTR_SUPPORTED_COLOR_MODES: [ColorMode.RGB, ColorMode.WHITE],
ATTR_RGBW_COLOR: (128, 50, 0, 255), ATTR_RGBW_COLOR: (128, 50, 0, 255),
ATTR_RGB_COLOR: (128, 50, 0), ATTR_RGB_COLOR: (128, 50, 0),
ATTR_HS_COLOR: (23.438, 100.0), ATTR_HS_COLOR: (23.438, 100.0),
ATTR_BRIGHTNESS: 255, ATTR_BRIGHTNESS: 255,
ATTR_COLOR_MODE: COLOR_MODE_RGB, ATTR_COLOR_MODE: ColorMode.RGB,
}, },
) )
await hass.async_block_till_done() await hass.async_block_till_done()
@ -884,9 +880,9 @@ async def test_light_rgb_or_w_lights(
entity_id, entity_id,
STATE_ON, STATE_ON,
{ {
ATTR_SUPPORTED_COLOR_MODES: [COLOR_MODE_RGB, COLOR_MODE_WHITE], ATTR_SUPPORTED_COLOR_MODES: [ColorMode.RGB, ColorMode.WHITE],
ATTR_BRIGHTNESS: 255, ATTR_BRIGHTNESS: 255,
ATTR_COLOR_MODE: COLOR_MODE_WHITE, ATTR_COLOR_MODE: ColorMode.WHITE,
}, },
) )
await hass.async_block_till_done() await hass.async_block_till_done()
@ -900,23 +896,23 @@ async def test_light_rgb_or_w_lights(
"supported_color_modes, state_props", "supported_color_modes, state_props",
[ [
[ [
[COLOR_MODE_COLOR_TEMP, COLOR_MODE_RGBW], [ColorMode.COLOR_TEMP, ColorMode.RGBW],
{ {
ATTR_RGBW_COLOR: (128, 50, 0, 255), ATTR_RGBW_COLOR: (128, 50, 0, 255),
ATTR_RGB_COLOR: (128, 50, 0), ATTR_RGB_COLOR: (128, 50, 0),
ATTR_HS_COLOR: (23.438, 100.0), ATTR_HS_COLOR: (23.438, 100.0),
ATTR_BRIGHTNESS: 255, ATTR_BRIGHTNESS: 255,
ATTR_COLOR_MODE: COLOR_MODE_RGBW, ATTR_COLOR_MODE: ColorMode.RGBW,
}, },
], ],
[ [
[COLOR_MODE_COLOR_TEMP, COLOR_MODE_RGBWW], [ColorMode.COLOR_TEMP, ColorMode.RGBWW],
{ {
ATTR_RGBWW_COLOR: (128, 50, 0, 255, 255), ATTR_RGBWW_COLOR: (128, 50, 0, 255, 255),
ATTR_RGB_COLOR: (128, 50, 0), ATTR_RGB_COLOR: (128, 50, 0),
ATTR_HS_COLOR: (23.438, 100.0), ATTR_HS_COLOR: (23.438, 100.0),
ATTR_BRIGHTNESS: 255, ATTR_BRIGHTNESS: 255,
ATTR_COLOR_MODE: COLOR_MODE_RGBWW, ATTR_COLOR_MODE: ColorMode.RGBWW,
}, },
], ],
], ],
@ -1013,12 +1009,12 @@ async def test_light_rgbww_with_color_temp_conversion(
entity_id, entity_id,
STATE_ON, STATE_ON,
{ {
ATTR_SUPPORTED_COLOR_MODES: [COLOR_MODE_RGBWW], ATTR_SUPPORTED_COLOR_MODES: [ColorMode.RGBWW],
ATTR_RGBWW_COLOR: (128, 50, 0, 255, 255), ATTR_RGBWW_COLOR: (128, 50, 0, 255, 255),
ATTR_RGB_COLOR: (128, 50, 0), ATTR_RGB_COLOR: (128, 50, 0),
ATTR_HS_COLOR: (23.438, 100.0), ATTR_HS_COLOR: (23.438, 100.0),
ATTR_BRIGHTNESS: 255, ATTR_BRIGHTNESS: 255,
ATTR_COLOR_MODE: COLOR_MODE_RGBWW, ATTR_COLOR_MODE: ColorMode.RGBWW,
}, },
) )
await hass.async_block_till_done() await hass.async_block_till_done()
@ -1091,12 +1087,12 @@ async def test_light_rgbww_with_color_temp_conversion(
entity_id, entity_id,
STATE_ON, STATE_ON,
{ {
ATTR_SUPPORTED_COLOR_MODES: [COLOR_MODE_RGBWW], ATTR_SUPPORTED_COLOR_MODES: [ColorMode.RGBWW],
ATTR_RGBWW_COLOR: (0, 0, 0, 128, 255), ATTR_RGBWW_COLOR: (0, 0, 0, 128, 255),
ATTR_RGB_COLOR: (255, 163, 79), ATTR_RGB_COLOR: (255, 163, 79),
ATTR_HS_COLOR: (28.636, 69.02), ATTR_HS_COLOR: (28.636, 69.02),
ATTR_BRIGHTNESS: 180, ATTR_BRIGHTNESS: 180,
ATTR_COLOR_MODE: COLOR_MODE_RGBWW, ATTR_COLOR_MODE: ColorMode.RGBWW,
}, },
) )
await hass.async_block_till_done() await hass.async_block_till_done()
@ -1134,12 +1130,12 @@ async def test_light_rgbw_with_color_temp_conversion(
entity_id, entity_id,
STATE_ON, STATE_ON,
{ {
ATTR_SUPPORTED_COLOR_MODES: [COLOR_MODE_RGBW], ATTR_SUPPORTED_COLOR_MODES: [ColorMode.RGBW],
ATTR_RGBWW_COLOR: (128, 50, 0, 255, 255), ATTR_RGBWW_COLOR: (128, 50, 0, 255, 255),
ATTR_RGB_COLOR: (128, 50, 0), ATTR_RGB_COLOR: (128, 50, 0),
ATTR_HS_COLOR: (23.438, 100.0), ATTR_HS_COLOR: (23.438, 100.0),
ATTR_BRIGHTNESS: 255, ATTR_BRIGHTNESS: 255,
ATTR_COLOR_MODE: COLOR_MODE_RGBW, ATTR_COLOR_MODE: ColorMode.RGBW,
}, },
) )
await hass.async_block_till_done() await hass.async_block_till_done()

14
tests/components/hue/test_light_v1.py
View file

@ -7,7 +7,7 @@ import aiohue
from homeassistant.components import hue from homeassistant.components import hue
from homeassistant.components.hue.const import CONF_ALLOW_HUE_GROUPS from homeassistant.components.hue.const import CONF_ALLOW_HUE_GROUPS
from homeassistant.components.hue.v1 import light as hue_light from homeassistant.components.hue.v1 import light as hue_light
from homeassistant.components.light import COLOR_MODE_COLOR_TEMP, COLOR_MODE_HS from homeassistant.components.light import ColorMode
from homeassistant.helpers import device_registry as dr, entity_registry as er from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.util import color from homeassistant.util import color
@ -237,10 +237,10 @@ async def test_lights_color_mode(hass, mock_bridge_v1):
assert lamp_1.attributes["brightness"] == 145 assert lamp_1.attributes["brightness"] == 145
assert lamp_1.attributes["hs_color"] == (36.067, 69.804) assert lamp_1.attributes["hs_color"] == (36.067, 69.804)
assert "color_temp" not in lamp_1.attributes assert "color_temp" not in lamp_1.attributes
assert lamp_1.attributes["color_mode"] == COLOR_MODE_HS assert lamp_1.attributes["color_mode"] == ColorMode.HS
assert lamp_1.attributes["supported_color_modes"] == [ assert lamp_1.attributes["supported_color_modes"] == [
COLOR_MODE_COLOR_TEMP, ColorMode.COLOR_TEMP,
COLOR_MODE_HS, ColorMode.HS,
] ]
new_light1_on = LIGHT_1_ON.copy() new_light1_on = LIGHT_1_ON.copy()
@ -262,10 +262,10 @@ async def test_lights_color_mode(hass, mock_bridge_v1):
assert lamp_1.attributes["brightness"] == 145 assert lamp_1.attributes["brightness"] == 145
assert lamp_1.attributes["color_temp"] == 467 assert lamp_1.attributes["color_temp"] == 467
assert "hs_color" in lamp_1.attributes assert "hs_color" in lamp_1.attributes
assert lamp_1.attributes["color_mode"] == COLOR_MODE_COLOR_TEMP assert lamp_1.attributes["color_mode"] == ColorMode.COLOR_TEMP
assert lamp_1.attributes["supported_color_modes"] == [ assert lamp_1.attributes["supported_color_modes"] == [
COLOR_MODE_COLOR_TEMP, ColorMode.COLOR_TEMP,
COLOR_MODE_HS, ColorMode.HS,
] ]

28
tests/components/hue/test_light_v2.py
View file

@ -1,6 +1,6 @@
"""Philips Hue lights platform tests for V2 bridge/api.""" """Philips Hue lights platform tests for V2 bridge/api."""
from homeassistant.components.light import COLOR_MODE_COLOR_TEMP, COLOR_MODE_XY from homeassistant.components.light import ColorMode
from homeassistant.helpers import entity_registry as er from homeassistant.helpers import entity_registry as er
from .conftest import setup_platform from .conftest import setup_platform
@ -27,10 +27,10 @@ async def test_lights(hass, mock_bridge_v2, v2_resources_test_data):
assert light_1.state == "on" assert light_1.state == "on"
assert light_1.attributes["brightness"] == int(46.85 / 100 * 255) assert light_1.attributes["brightness"] == int(46.85 / 100 * 255)
assert light_1.attributes["mode"] == "normal" assert light_1.attributes["mode"] == "normal"
assert light_1.attributes["color_mode"] == COLOR_MODE_XY assert light_1.attributes["color_mode"] == ColorMode.XY
assert set(light_1.attributes["supported_color_modes"]) == { assert set(light_1.attributes["supported_color_modes"]) == {
COLOR_MODE_COLOR_TEMP, ColorMode.COLOR_TEMP,
COLOR_MODE_XY, ColorMode.XY,
} }
assert light_1.attributes["xy_color"] == (0.5614, 0.4058) assert light_1.attributes["xy_color"] == (0.5614, 0.4058)
assert light_1.attributes["min_mireds"] == 153 assert light_1.attributes["min_mireds"] == 153
@ -47,7 +47,7 @@ async def test_lights(hass, mock_bridge_v2, v2_resources_test_data):
) )
assert light_2.state == "off" assert light_2.state == "off"
assert light_2.attributes["mode"] == "normal" assert light_2.attributes["mode"] == "normal"
assert light_2.attributes["supported_color_modes"] == [COLOR_MODE_COLOR_TEMP] assert light_2.attributes["supported_color_modes"] == [ColorMode.COLOR_TEMP]
assert light_2.attributes["min_mireds"] == 153 assert light_2.attributes["min_mireds"] == 153
assert light_2.attributes["max_mireds"] == 454 assert light_2.attributes["max_mireds"] == 454
assert light_2.attributes["dynamics"] == "none" assert light_2.attributes["dynamics"] == "none"
@ -60,8 +60,8 @@ async def test_lights(hass, mock_bridge_v2, v2_resources_test_data):
assert light_3.state == "on" assert light_3.state == "on"
assert light_3.attributes["brightness"] == 128 assert light_3.attributes["brightness"] == 128
assert light_3.attributes["mode"] == "normal" assert light_3.attributes["mode"] == "normal"
assert light_3.attributes["supported_color_modes"] == [COLOR_MODE_XY] assert light_3.attributes["supported_color_modes"] == [ColorMode.XY]
assert light_3.attributes["color_mode"] == COLOR_MODE_XY assert light_3.attributes["color_mode"] == ColorMode.XY
assert light_3.attributes["dynamics"] == "dynamic_palette" assert light_3.attributes["dynamics"] == "dynamic_palette"
# test light which supports on/off only # test light which supports on/off only
@ -113,8 +113,8 @@ async def test_light_turn_on_service(hass, mock_bridge_v2, v2_resources_test_dat
assert test_light is not None assert test_light is not None
assert test_light.state == "on" assert test_light.state == "on"
assert test_light.attributes["mode"] == "normal" assert test_light.attributes["mode"] == "normal"
assert test_light.attributes["supported_color_modes"] == [COLOR_MODE_COLOR_TEMP] assert test_light.attributes["supported_color_modes"] == [ColorMode.COLOR_TEMP]
assert test_light.attributes["color_mode"] == COLOR_MODE_COLOR_TEMP assert test_light.attributes["color_mode"] == ColorMode.COLOR_TEMP
assert test_light.attributes["brightness"] == 255 assert test_light.attributes["brightness"] == 255
# test again with sending transition with 250ms which should round up to 200ms # test again with sending transition with 250ms which should round up to 200ms
@ -344,10 +344,10 @@ async def test_grouped_lights(hass, mock_bridge_v2, v2_resources_test_data):
assert test_entity.attributes["friendly_name"] == "Test Zone" assert test_entity.attributes["friendly_name"] == "Test Zone"
assert test_entity.state == "on" assert test_entity.state == "on"
assert test_entity.attributes["brightness"] == 119 assert test_entity.attributes["brightness"] == 119
assert test_entity.attributes["color_mode"] == COLOR_MODE_XY assert test_entity.attributes["color_mode"] == ColorMode.XY
assert set(test_entity.attributes["supported_color_modes"]) == { assert set(test_entity.attributes["supported_color_modes"]) == {
COLOR_MODE_COLOR_TEMP, ColorMode.COLOR_TEMP,
COLOR_MODE_XY, ColorMode.XY,
} }
assert test_entity.attributes["min_mireds"] == 153 assert test_entity.attributes["min_mireds"] == 153
assert test_entity.attributes["max_mireds"] == 500 assert test_entity.attributes["max_mireds"] == 500
@ -365,7 +365,7 @@ async def test_grouped_lights(hass, mock_bridge_v2, v2_resources_test_data):
assert test_entity is not None assert test_entity is not None
assert test_entity.attributes["friendly_name"] == "Test Room" assert test_entity.attributes["friendly_name"] == "Test Room"
assert test_entity.state == "off" assert test_entity.state == "off"
assert test_entity.attributes["supported_color_modes"] == [COLOR_MODE_COLOR_TEMP] assert test_entity.attributes["supported_color_modes"] == [ColorMode.COLOR_TEMP]
assert test_entity.attributes["min_mireds"] == 153 assert test_entity.attributes["min_mireds"] == 153
assert test_entity.attributes["max_mireds"] == 454 assert test_entity.attributes["max_mireds"] == 454
assert test_entity.attributes["is_hue_group"] is True assert test_entity.attributes["is_hue_group"] is True
@ -417,7 +417,7 @@ async def test_grouped_lights(hass, mock_bridge_v2, v2_resources_test_data):
test_light = hass.states.get(test_light_id) test_light = hass.states.get(test_light_id)
assert test_light is not None assert test_light is not None
assert test_light.state == "on" assert test_light.state == "on"
assert test_light.attributes["color_mode"] == COLOR_MODE_XY assert test_light.attributes["color_mode"] == ColorMode.XY
assert test_light.attributes["brightness"] == 255 assert test_light.attributes["brightness"] == 255
assert test_light.attributes["xy_color"] == (0.123, 0.123) assert test_light.attributes["xy_color"] == (0.123, 0.123)