mirror of
https://github.com/home-assistant/core
synced 2024-07-12 07:21:24 +02:00
Rewrite tests for Template Light (#41163)
This commit is contained in:
sycx2
GitHub
parent
6f7f4955a3
commit
009e44ed9b
@ -4,17 +4,23 @@ import logging
|
||||
import pytest
|
||||
|
||||
from homeassistant import setup
|
||||
import homeassistant.components.light as light
|
||||
from homeassistant.components.light import (
|
||||
ATTR_BRIGHTNESS,
|
||||
ATTR_COLOR_TEMP,
|
||||
ATTR_HS_COLOR,
|
||||
ATTR_WHITE_VALUE,
|
||||
)
|
||||
from homeassistant.const import STATE_OFF, STATE_ON, STATE_UNAVAILABLE
|
||||
from homeassistant.core import callback
|
||||
from homeassistant.const import (
|
||||
ATTR_ENTITY_ID,
|
||||
SERVICE_TURN_OFF,
|
||||
SERVICE_TURN_ON,
|
||||
STATE_OFF,
|
||||
STATE_ON,
|
||||
STATE_UNAVAILABLE,
|
||||
)
|
||||
|
||||
from tests.common import assert_setup_component, get_test_home_assistant
|
||||
from tests.components.light import common
|
||||
from tests.common import assert_setup_component, async_mock_service
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
@ -22,35 +28,18 @@ _LOGGER = logging.getLogger(__name__)
|
||||
_STATE_AVAILABILITY_BOOLEAN = "availability_boolean.state"
|
||||
|
||||
|
||||
class TestTemplateLight:
|
||||
"""Test the Template light."""
|
||||
@pytest.fixture(name="calls")
|
||||
def fixture_calls(hass):
|
||||
"""Track calls to a mock service."""
|
||||
return async_mock_service(hass, "test", "automation")
|
||||
|
||||
hass = None
|
||||
calls = None
|
||||
# pylint: disable=invalid-name
|
||||
|
||||
def setup_method(self, method):
|
||||
"""Set up things to be run when tests are started."""
|
||||
self.hass = get_test_home_assistant()
|
||||
self.calls = []
|
||||
|
||||
@callback
|
||||
def record_call(service):
|
||||
"""Track function calls."""
|
||||
self.calls.append(service)
|
||||
|
||||
self.hass.services.register("test", "automation", record_call)
|
||||
|
||||
def teardown_method(self, method):
|
||||
"""Stop everything that was started."""
|
||||
self.hass.stop()
|
||||
|
||||
def test_template_state_invalid(self):
|
||||
async def test_template_state_invalid(hass):
|
||||
"""Test template state with render error."""
|
||||
with assert_setup_component(1, "light"):
|
||||
assert setup.setup_component(
|
||||
self.hass,
|
||||
"light",
|
||||
with assert_setup_component(1, light.DOMAIN):
|
||||
assert await setup.async_setup_component(
|
||||
hass,
|
||||
light.DOMAIN,
|
||||
{
|
||||
"light": {
|
||||
"platform": "template",
|
||||
@ -78,19 +67,20 @@ class TestTemplateLight:
|
||||
},
|
||||
)
|
||||
|
||||
self.hass.block_till_done()
|
||||
self.hass.start()
|
||||
self.hass.block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_start()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
state = hass.states.get("light.test_template_light")
|
||||
assert state.state == STATE_OFF
|
||||
|
||||
def test_template_state_text(self):
|
||||
|
||||
async def test_template_state_text(hass):
|
||||
"""Test the state text of a template."""
|
||||
with assert_setup_component(1, "light"):
|
||||
assert setup.setup_component(
|
||||
self.hass,
|
||||
"light",
|
||||
with assert_setup_component(1, light.DOMAIN):
|
||||
assert await setup.async_setup_component(
|
||||
hass,
|
||||
light.DOMAIN,
|
||||
{
|
||||
"light": {
|
||||
"platform": "template",
|
||||
@ -118,32 +108,33 @@ class TestTemplateLight:
|
||||
},
|
||||
)
|
||||
|
||||
self.hass.block_till_done()
|
||||
self.hass.start()
|
||||
self.hass.block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_start()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = self.hass.states.set("light.test_state", STATE_ON)
|
||||
self.hass.block_till_done()
|
||||
state = hass.states.async_set("light.test_state", STATE_ON)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
state = hass.states.get("light.test_template_light")
|
||||
assert state.state == STATE_ON
|
||||
|
||||
state = self.hass.states.set("light.test_state", STATE_OFF)
|
||||
self.hass.block_till_done()
|
||||
state = hass.states.async_set("light.test_state", STATE_OFF)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
state = hass.states.get("light.test_template_light")
|
||||
assert state.state == STATE_OFF
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"expected_state,template",
|
||||
[(STATE_ON, "{{ 1 == 1 }}"), (STATE_OFF, "{{ 1 == 2 }}")],
|
||||
)
|
||||
def test_template_state_boolean(self, expected_state, template):
|
||||
)
|
||||
async def test_template_state_boolean(hass, expected_state, template):
|
||||
"""Test the setting of the state with boolean on."""
|
||||
with assert_setup_component(1, "light"):
|
||||
assert setup.setup_component(
|
||||
self.hass,
|
||||
"light",
|
||||
with assert_setup_component(1, light.DOMAIN):
|
||||
assert await setup.async_setup_component(
|
||||
hass,
|
||||
light.DOMAIN,
|
||||
{
|
||||
"light": {
|
||||
"platform": "template",
|
||||
@ -171,19 +162,20 @@ class TestTemplateLight:
|
||||
},
|
||||
)
|
||||
|
||||
self.hass.block_till_done()
|
||||
self.hass.start()
|
||||
self.hass.block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_start()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
state = hass.states.get("light.test_template_light")
|
||||
assert state.state == expected_state
|
||||
|
||||
def test_template_syntax_error(self):
|
||||
|
||||
async def test_template_syntax_error(hass):
|
||||
"""Test templating syntax error."""
|
||||
with assert_setup_component(0, "light"):
|
||||
assert setup.setup_component(
|
||||
self.hass,
|
||||
"light",
|
||||
with assert_setup_component(0, light.DOMAIN):
|
||||
assert await setup.async_setup_component(
|
||||
hass,
|
||||
light.DOMAIN,
|
||||
{
|
||||
"light": {
|
||||
"platform": "template",
|
||||
@ -211,18 +203,19 @@ class TestTemplateLight:
|
||||
},
|
||||
)
|
||||
|
||||
self.hass.block_till_done()
|
||||
self.hass.start()
|
||||
self.hass.block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_start()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert self.hass.states.all() == []
|
||||
assert hass.states.async_all() == []
|
||||
|
||||
def test_invalid_name_does_not_create(self):
|
||||
|
||||
async def test_invalid_name_does_not_create(hass):
|
||||
"""Test invalid name."""
|
||||
with assert_setup_component(0, "light"):
|
||||
assert setup.setup_component(
|
||||
self.hass,
|
||||
"light",
|
||||
with assert_setup_component(0, light.DOMAIN):
|
||||
assert await setup.async_setup_component(
|
||||
hass,
|
||||
light.DOMAIN,
|
||||
{
|
||||
"light": {
|
||||
"platform": "template",
|
||||
@ -250,18 +243,19 @@ class TestTemplateLight:
|
||||
},
|
||||
)
|
||||
|
||||
self.hass.block_till_done()
|
||||
self.hass.start()
|
||||
self.hass.block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_start()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert self.hass.states.all() == []
|
||||
assert hass.states.async_all() == []
|
||||
|
||||
def test_invalid_light_does_not_create(self):
|
||||
|
||||
async def test_invalid_light_does_not_create(hass):
|
||||
"""Test invalid light."""
|
||||
with assert_setup_component(0, "light"):
|
||||
assert setup.setup_component(
|
||||
self.hass,
|
||||
"light",
|
||||
with assert_setup_component(0, light.DOMAIN):
|
||||
assert await setup.async_setup_component(
|
||||
hass,
|
||||
light.DOMAIN,
|
||||
{
|
||||
"light": {
|
||||
"platform": "template",
|
||||
@ -270,31 +264,33 @@ class TestTemplateLight:
|
||||
},
|
||||
)
|
||||
|
||||
self.hass.block_till_done()
|
||||
self.hass.start()
|
||||
self.hass.block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_start()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert self.hass.states.all() == []
|
||||
assert hass.states.async_all() == []
|
||||
|
||||
def test_no_lights_does_not_create(self):
|
||||
|
||||
async def test_no_lights_does_not_create(hass):
|
||||
"""Test if there are no lights no creation."""
|
||||
with assert_setup_component(0, "light"):
|
||||
assert setup.setup_component(
|
||||
self.hass, "light", {"light": {"platform": "template"}}
|
||||
with assert_setup_component(0, light.DOMAIN):
|
||||
assert await setup.async_setup_component(
|
||||
hass, "light", {"light": {"platform": "template"}}
|
||||
)
|
||||
|
||||
self.hass.block_till_done()
|
||||
self.hass.start()
|
||||
self.hass.block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_start()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
assert self.hass.states.all() == []
|
||||
assert hass.states.async_all() == []
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"missing_key, count", [("value_template", 1), ("turn_on", 0), ("turn_off", 0)]
|
||||
)
|
||||
def test_missing_key(self, missing_key, count):
|
||||
)
|
||||
async def test_missing_key(hass, missing_key, count):
|
||||
"""Test missing template."""
|
||||
light = {
|
||||
light_config = {
|
||||
"light": {
|
||||
"platform": "template",
|
||||
"lights": {
|
||||
@ -320,23 +316,24 @@ class TestTemplateLight:
|
||||
}
|
||||
}
|
||||
|
||||
del light["light"]["lights"]["light_one"][missing_key]
|
||||
with assert_setup_component(count, "light"):
|
||||
assert setup.setup_component(self.hass, "light", light)
|
||||
self.hass.block_till_done()
|
||||
self.hass.start()
|
||||
self.hass.block_till_done()
|
||||
del light_config["light"]["lights"]["light_one"][missing_key]
|
||||
with assert_setup_component(count, light.DOMAIN):
|
||||
assert await setup.async_setup_component(hass, "light", light_config)
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_start()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
if count:
|
||||
assert self.hass.states.all() != []
|
||||
assert hass.states.async_all() != []
|
||||
else:
|
||||
assert self.hass.states.all() == []
|
||||
assert hass.states.async_all() == []
|
||||
|
||||
def test_on_action(self):
|
||||
|
||||
async def test_on_action(hass, calls):
|
||||
"""Test on action."""
|
||||
assert setup.setup_component(
|
||||
self.hass,
|
||||
"light",
|
||||
assert await setup.async_setup_component(
|
||||
hass,
|
||||
light.DOMAIN,
|
||||
{
|
||||
"light": {
|
||||
"platform": "template",
|
||||
@ -361,26 +358,31 @@ class TestTemplateLight:
|
||||
},
|
||||
)
|
||||
|
||||
self.hass.block_till_done()
|
||||
self.hass.start()
|
||||
self.hass.block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_start()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
self.hass.states.set("light.test_state", STATE_OFF)
|
||||
self.hass.block_till_done()
|
||||
hass.states.async_set("light.test_state", STATE_OFF)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
state = hass.states.get("light.test_template_light")
|
||||
assert state.state == STATE_OFF
|
||||
|
||||
common.turn_on(self.hass, "light.test_template_light")
|
||||
self.hass.block_till_done()
|
||||
await hass.services.async_call(
|
||||
light.DOMAIN,
|
||||
SERVICE_TURN_ON,
|
||||
{ATTR_ENTITY_ID: "light.test_template_light"},
|
||||
blocking=True,
|
||||
)
|
||||
|
||||
assert len(self.calls) == 1
|
||||
assert len(calls) == 1
|
||||
|
||||
def test_on_action_optimistic(self):
|
||||
|
||||
async def test_on_action_optimistic(hass, calls):
|
||||
"""Test on action with optimistic state."""
|
||||
assert setup.setup_component(
|
||||
self.hass,
|
||||
"light",
|
||||
assert await setup.async_setup_component(
|
||||
hass,
|
||||
light.DOMAIN,
|
||||
{
|
||||
"light": {
|
||||
"platform": "template",
|
||||
@ -404,28 +406,33 @@ class TestTemplateLight:
|
||||
},
|
||||
)
|
||||
|
||||
self.hass.block_till_done()
|
||||
self.hass.start()
|
||||
self.hass.block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_start()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
self.hass.states.set("light.test_state", STATE_OFF)
|
||||
self.hass.block_till_done()
|
||||
hass.states.async_set("light.test_state", STATE_OFF)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
state = hass.states.get("light.test_template_light")
|
||||
assert state.state == STATE_OFF
|
||||
|
||||
common.turn_on(self.hass, "light.test_template_light")
|
||||
self.hass.block_till_done()
|
||||
await hass.services.async_call(
|
||||
light.DOMAIN,
|
||||
SERVICE_TURN_ON,
|
||||
{ATTR_ENTITY_ID: "light.test_template_light"},
|
||||
blocking=True,
|
||||
)
|
||||
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
assert len(self.calls) == 1
|
||||
state = hass.states.get("light.test_template_light")
|
||||
assert len(calls) == 1
|
||||
assert state.state == STATE_ON
|
||||
|
||||
def test_off_action(self):
|
||||
|
||||
async def test_off_action(hass, calls):
|
||||
"""Test off action."""
|
||||
assert setup.setup_component(
|
||||
self.hass,
|
||||
"light",
|
||||
assert await setup.async_setup_component(
|
||||
hass,
|
||||
light.DOMAIN,
|
||||
{
|
||||
"light": {
|
||||
"platform": "template",
|
||||
@ -450,26 +457,31 @@ class TestTemplateLight:
|
||||
},
|
||||
)
|
||||
|
||||
self.hass.block_till_done()
|
||||
self.hass.start()
|
||||
self.hass.block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_start()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
self.hass.states.set("light.test_state", STATE_ON)
|
||||
self.hass.block_till_done()
|
||||
hass.states.async_set("light.test_state", STATE_ON)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
state = hass.states.get("light.test_template_light")
|
||||
assert state.state == STATE_ON
|
||||
|
||||
common.turn_off(self.hass, "light.test_template_light")
|
||||
self.hass.block_till_done()
|
||||
await hass.services.async_call(
|
||||
light.DOMAIN,
|
||||
SERVICE_TURN_OFF,
|
||||
{ATTR_ENTITY_ID: "light.test_template_light"},
|
||||
blocking=True,
|
||||
)
|
||||
|
||||
assert len(self.calls) == 1
|
||||
assert len(calls) == 1
|
||||
|
||||
def test_off_action_optimistic(self):
|
||||
|
||||
async def test_off_action_optimistic(hass, calls):
|
||||
"""Test off action with optimistic state."""
|
||||
assert setup.setup_component(
|
||||
self.hass,
|
||||
"light",
|
||||
assert await setup.async_setup_component(
|
||||
hass,
|
||||
light.DOMAIN,
|
||||
{
|
||||
"light": {
|
||||
"platform": "template",
|
||||
@ -493,25 +505,30 @@ class TestTemplateLight:
|
||||
},
|
||||
)
|
||||
|
||||
self.hass.block_till_done()
|
||||
self.hass.start()
|
||||
self.hass.block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_start()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
state = hass.states.get("light.test_template_light")
|
||||
assert state.state == STATE_OFF
|
||||
|
||||
common.turn_off(self.hass, "light.test_template_light")
|
||||
self.hass.block_till_done()
|
||||
await hass.services.async_call(
|
||||
light.DOMAIN,
|
||||
SERVICE_TURN_OFF,
|
||||
{ATTR_ENTITY_ID: "light.test_template_light"},
|
||||
blocking=True,
|
||||
)
|
||||
|
||||
assert len(self.calls) == 1
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
assert len(calls) == 1
|
||||
state = hass.states.get("light.test_template_light")
|
||||
assert state.state == STATE_OFF
|
||||
|
||||
def test_white_value_action_no_template(self):
|
||||
|
||||
async def test_white_value_action_no_template(hass, calls):
|
||||
"""Test setting white value with optimistic template."""
|
||||
assert setup.setup_component(
|
||||
self.hass,
|
||||
"light",
|
||||
assert await setup.async_setup_component(
|
||||
hass,
|
||||
light.DOMAIN,
|
||||
{
|
||||
"light": {
|
||||
"platform": "template",
|
||||
@ -538,25 +555,29 @@ class TestTemplateLight:
|
||||
}
|
||||
},
|
||||
)
|
||||
self.hass.block_till_done()
|
||||
self.hass.start()
|
||||
self.hass.block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_start()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
state = hass.states.get("light.test_template_light")
|
||||
assert state.attributes.get("white_value") is None
|
||||
|
||||
common.turn_on(
|
||||
self.hass, "light.test_template_light", **{ATTR_WHITE_VALUE: 124}
|
||||
await hass.services.async_call(
|
||||
light.DOMAIN,
|
||||
SERVICE_TURN_ON,
|
||||
{ATTR_ENTITY_ID: "light.test_template_light", ATTR_WHITE_VALUE: 124},
|
||||
blocking=True,
|
||||
)
|
||||
self.hass.block_till_done()
|
||||
assert len(self.calls) == 1
|
||||
assert self.calls[0].data["white_value"] == 124
|
||||
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
assert len(calls) == 1
|
||||
assert calls[0].data["white_value"] == 124
|
||||
|
||||
state = hass.states.get("light.test_template_light")
|
||||
assert state is not None
|
||||
assert state.attributes.get("white_value") == 124
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"expected_white_value,template",
|
||||
[
|
||||
(255, "{{255}}"),
|
||||
@ -565,13 +586,13 @@ class TestTemplateLight:
|
||||
(None, "{{ none }}"),
|
||||
(None, ""),
|
||||
],
|
||||
)
|
||||
def test_white_value_template(self, expected_white_value, template):
|
||||
)
|
||||
async def test_white_value_template(hass, expected_white_value, template):
|
||||
"""Test the template for the white value."""
|
||||
with assert_setup_component(1, "light"):
|
||||
assert setup.setup_component(
|
||||
self.hass,
|
||||
"light",
|
||||
with assert_setup_component(1, light.DOMAIN):
|
||||
assert await setup.async_setup_component(
|
||||
hass,
|
||||
light.DOMAIN,
|
||||
{
|
||||
"light": {
|
||||
"platform": "template",
|
||||
@ -600,19 +621,20 @@ class TestTemplateLight:
|
||||
},
|
||||
)
|
||||
|
||||
self.hass.block_till_done()
|
||||
self.hass.start()
|
||||
self.hass.block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_start()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
state = hass.states.get("light.test_template_light")
|
||||
assert state is not None
|
||||
assert state.attributes.get("white_value") == expected_white_value
|
||||
|
||||
def test_level_action_no_template(self):
|
||||
|
||||
async def test_level_action_no_template(hass, calls):
|
||||
"""Test setting brightness with optimistic template."""
|
||||
assert setup.setup_component(
|
||||
self.hass,
|
||||
"light",
|
||||
assert await setup.async_setup_component(
|
||||
hass,
|
||||
light.DOMAIN,
|
||||
{
|
||||
"light": {
|
||||
"platform": "template",
|
||||
@ -639,24 +661,30 @@ class TestTemplateLight:
|
||||
}
|
||||
},
|
||||
)
|
||||
self.hass.block_till_done()
|
||||
self.hass.start()
|
||||
self.hass.block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_start()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
state = hass.states.get("light.test_template_light")
|
||||
assert state.attributes.get("brightness") is None
|
||||
|
||||
common.turn_on(self.hass, "light.test_template_light", **{ATTR_BRIGHTNESS: 124})
|
||||
self.hass.block_till_done()
|
||||
assert len(self.calls) == 1
|
||||
assert self.calls[0].data["brightness"] == 124
|
||||
await hass.services.async_call(
|
||||
light.DOMAIN,
|
||||
SERVICE_TURN_ON,
|
||||
{ATTR_ENTITY_ID: "light.test_template_light", ATTR_BRIGHTNESS: 124},
|
||||
blocking=True,
|
||||
)
|
||||
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
assert len(calls) == 1
|
||||
assert calls[0].data["brightness"] == 124
|
||||
|
||||
state = hass.states.get("light.test_template_light")
|
||||
_LOGGER.info(str(state.attributes))
|
||||
assert state is not None
|
||||
assert state.attributes.get("brightness") == 124
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"expected_level,template",
|
||||
[
|
||||
(255, "{{255}}"),
|
||||
@ -665,13 +693,13 @@ class TestTemplateLight:
|
||||
(None, "{{ none }}"),
|
||||
(None, ""),
|
||||
],
|
||||
)
|
||||
def test_level_template(self, expected_level, template):
|
||||
)
|
||||
async def test_level_template(hass, expected_level, template):
|
||||
"""Test the template for the level."""
|
||||
with assert_setup_component(1, "light"):
|
||||
assert setup.setup_component(
|
||||
self.hass,
|
||||
"light",
|
||||
with assert_setup_component(1, light.DOMAIN):
|
||||
assert await setup.async_setup_component(
|
||||
hass,
|
||||
light.DOMAIN,
|
||||
{
|
||||
"light": {
|
||||
"platform": "template",
|
||||
@ -700,15 +728,16 @@ class TestTemplateLight:
|
||||
},
|
||||
)
|
||||
|
||||
self.hass.block_till_done()
|
||||
self.hass.start()
|
||||
self.hass.block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_start()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
state = hass.states.get("light.test_template_light")
|
||||
assert state is not None
|
||||
assert state.attributes.get("brightness") == expected_level
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"expected_temp,template",
|
||||
[
|
||||
(500, "{{500}}"),
|
||||
@ -718,13 +747,13 @@ class TestTemplateLight:
|
||||
(None, "{{ none }}"),
|
||||
(None, ""),
|
||||
],
|
||||
)
|
||||
def test_temperature_template(self, expected_temp, template):
|
||||
)
|
||||
async def test_temperature_template(hass, expected_temp, template):
|
||||
"""Test the template for the temperature."""
|
||||
with assert_setup_component(1, "light"):
|
||||
assert setup.setup_component(
|
||||
self.hass,
|
||||
"light",
|
||||
with assert_setup_component(1, light.DOMAIN):
|
||||
assert await setup.async_setup_component(
|
||||
hass,
|
||||
light.DOMAIN,
|
||||
{
|
||||
"light": {
|
||||
"platform": "template",
|
||||
@ -753,19 +782,20 @@ class TestTemplateLight:
|
||||
},
|
||||
)
|
||||
|
||||
self.hass.block_till_done()
|
||||
self.hass.start()
|
||||
self.hass.block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_start()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
state = hass.states.get("light.test_template_light")
|
||||
assert state is not None
|
||||
assert state.attributes.get("color_temp") == expected_temp
|
||||
|
||||
def test_temperature_action_no_template(self):
|
||||
|
||||
async def test_temperature_action_no_template(hass, calls):
|
||||
"""Test setting temperature with optimistic template."""
|
||||
assert setup.setup_component(
|
||||
self.hass,
|
||||
"light",
|
||||
assert await setup.async_setup_component(
|
||||
hass,
|
||||
light.DOMAIN,
|
||||
{
|
||||
"light": {
|
||||
"platform": "template",
|
||||
@ -792,29 +822,35 @@ class TestTemplateLight:
|
||||
}
|
||||
},
|
||||
)
|
||||
self.hass.block_till_done()
|
||||
self.hass.start()
|
||||
self.hass.block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_start()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
state = hass.states.get("light.test_template_light")
|
||||
assert state.attributes.get("color_template") is None
|
||||
|
||||
common.turn_on(self.hass, "light.test_template_light", **{ATTR_COLOR_TEMP: 345})
|
||||
self.hass.block_till_done()
|
||||
assert len(self.calls) == 1
|
||||
assert self.calls[0].data["color_temp"] == 345
|
||||
await hass.services.async_call(
|
||||
light.DOMAIN,
|
||||
SERVICE_TURN_ON,
|
||||
{ATTR_ENTITY_ID: "light.test_template_light", ATTR_COLOR_TEMP: 345},
|
||||
blocking=True,
|
||||
)
|
||||
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
assert len(calls) == 1
|
||||
assert calls[0].data["color_temp"] == 345
|
||||
|
||||
state = hass.states.get("light.test_template_light")
|
||||
_LOGGER.info(str(state.attributes))
|
||||
assert state is not None
|
||||
assert state.attributes.get("color_temp") == 345
|
||||
|
||||
def test_friendly_name(self):
|
||||
|
||||
async def test_friendly_name(hass):
|
||||
"""Test the accessibility of the friendly_name attribute."""
|
||||
with assert_setup_component(1, "light"):
|
||||
assert setup.setup_component(
|
||||
self.hass,
|
||||
"light",
|
||||
with assert_setup_component(1, light.DOMAIN):
|
||||
assert await setup.async_setup_component(
|
||||
hass,
|
||||
light.DOMAIN,
|
||||
{
|
||||
"light": {
|
||||
"platform": "template",
|
||||
@ -843,21 +879,22 @@ class TestTemplateLight:
|
||||
},
|
||||
)
|
||||
|
||||
self.hass.block_till_done()
|
||||
self.hass.start()
|
||||
self.hass.block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_start()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
state = hass.states.get("light.test_template_light")
|
||||
assert state is not None
|
||||
|
||||
assert state.attributes.get("friendly_name") == "Template light"
|
||||
|
||||
def test_icon_template(self):
|
||||
|
||||
async def test_icon_template(hass):
|
||||
"""Test icon template."""
|
||||
with assert_setup_component(1, "light"):
|
||||
assert setup.setup_component(
|
||||
self.hass,
|
||||
"light",
|
||||
with assert_setup_component(1, light.DOMAIN):
|
||||
assert await setup.async_setup_component(
|
||||
hass,
|
||||
light.DOMAIN,
|
||||
{
|
||||
"light": {
|
||||
"platform": "template",
|
||||
@ -889,26 +926,27 @@ class TestTemplateLight:
|
||||
},
|
||||
)
|
||||
|
||||
self.hass.block_till_done()
|
||||
self.hass.start()
|
||||
self.hass.block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_start()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
state = hass.states.get("light.test_template_light")
|
||||
assert state.attributes.get("icon") == ""
|
||||
|
||||
state = self.hass.states.set("light.test_state", STATE_ON)
|
||||
self.hass.block_till_done()
|
||||
state = hass.states.async_set("light.test_state", STATE_ON)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
state = hass.states.get("light.test_template_light")
|
||||
|
||||
assert state.attributes["icon"] == "mdi:check"
|
||||
|
||||
def test_entity_picture_template(self):
|
||||
|
||||
async def test_entity_picture_template(hass):
|
||||
"""Test entity_picture template."""
|
||||
with assert_setup_component(1, "light"):
|
||||
assert setup.setup_component(
|
||||
self.hass,
|
||||
"light",
|
||||
with assert_setup_component(1, light.DOMAIN):
|
||||
assert await setup.async_setup_component(
|
||||
hass,
|
||||
light.DOMAIN,
|
||||
{
|
||||
"light": {
|
||||
"platform": "template",
|
||||
@ -940,25 +978,26 @@ class TestTemplateLight:
|
||||
},
|
||||
)
|
||||
|
||||
self.hass.block_till_done()
|
||||
self.hass.start()
|
||||
self.hass.block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_start()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
state = hass.states.get("light.test_template_light")
|
||||
assert state.attributes.get("entity_picture") == ""
|
||||
|
||||
state = self.hass.states.set("light.test_state", STATE_ON)
|
||||
self.hass.block_till_done()
|
||||
state = hass.states.async_set("light.test_state", STATE_ON)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
state = hass.states.get("light.test_template_light")
|
||||
|
||||
assert state.attributes["entity_picture"] == "/local/light.png"
|
||||
|
||||
def test_color_action_no_template(self):
|
||||
|
||||
async def test_color_action_no_template(hass, calls):
|
||||
"""Test setting color with optimistic template."""
|
||||
assert setup.setup_component(
|
||||
self.hass,
|
||||
"light",
|
||||
assert await setup.async_setup_component(
|
||||
hass,
|
||||
light.DOMAIN,
|
||||
{
|
||||
"light": {
|
||||
"platform": "template",
|
||||
@ -996,32 +1035,36 @@ class TestTemplateLight:
|
||||
}
|
||||
},
|
||||
)
|
||||
self.hass.block_till_done()
|
||||
self.hass.start()
|
||||
self.hass.block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_start()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
state = hass.states.get("light.test_template_light")
|
||||
assert state.attributes.get("hs_color") is None
|
||||
|
||||
common.turn_on(
|
||||
self.hass, "light.test_template_light", **{ATTR_HS_COLOR: (40, 50)}
|
||||
await hass.services.async_call(
|
||||
light.DOMAIN,
|
||||
SERVICE_TURN_ON,
|
||||
{ATTR_ENTITY_ID: "light.test_template_light", ATTR_HS_COLOR: (40, 50)},
|
||||
blocking=True,
|
||||
)
|
||||
self.hass.block_till_done()
|
||||
assert len(self.calls) == 2
|
||||
assert self.calls[0].data["h"] == 40
|
||||
assert self.calls[0].data["s"] == 50
|
||||
assert self.calls[1].data["h"] == 40
|
||||
assert self.calls[1].data["s"] == 50
|
||||
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
assert len(calls) == 2
|
||||
assert calls[0].data["h"] == 40
|
||||
assert calls[0].data["s"] == 50
|
||||
assert calls[1].data["h"] == 40
|
||||
assert calls[1].data["s"] == 50
|
||||
|
||||
state = hass.states.get("light.test_template_light")
|
||||
_LOGGER.info(str(state.attributes))
|
||||
assert state is not None
|
||||
assert self.calls[0].data["h"] == 40
|
||||
assert self.calls[0].data["s"] == 50
|
||||
assert self.calls[1].data["h"] == 40
|
||||
assert self.calls[1].data["s"] == 50
|
||||
assert calls[0].data["h"] == 40
|
||||
assert calls[0].data["s"] == 50
|
||||
assert calls[1].data["h"] == 40
|
||||
assert calls[1].data["s"] == 50
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"expected_hs,template",
|
||||
[
|
||||
((360, 100), "{{(360, 100)}}"),
|
||||
@ -1032,13 +1075,13 @@ class TestTemplateLight:
|
||||
(None, ""),
|
||||
(None, "{{ none }}"),
|
||||
],
|
||||
)
|
||||
def test_color_template(self, expected_hs, template):
|
||||
)
|
||||
async def test_color_template(hass, expected_hs, template):
|
||||
"""Test the template for the color."""
|
||||
with assert_setup_component(1, "light"):
|
||||
assert setup.setup_component(
|
||||
self.hass,
|
||||
"light",
|
||||
with assert_setup_component(1, light.DOMAIN):
|
||||
assert await setup.async_setup_component(
|
||||
hass,
|
||||
light.DOMAIN,
|
||||
{
|
||||
"light": {
|
||||
"platform": "template",
|
||||
@ -1068,10 +1111,10 @@ class TestTemplateLight:
|
||||
}
|
||||
},
|
||||
)
|
||||
self.hass.block_till_done()
|
||||
self.hass.start()
|
||||
self.hass.block_till_done()
|
||||
state = self.hass.states.get("light.test_template_light")
|
||||
await hass.async_block_till_done()
|
||||
await hass.async_start()
|
||||
await hass.async_block_till_done()
|
||||
state = hass.states.get("light.test_template_light")
|
||||
assert state is not None
|
||||
assert state.attributes.get("hs_color") == expected_hs
|
||||
|
||||
@ -1080,7 +1123,7 @@ async def test_available_template_with_entities(hass):
|
||||
"""Test availability templates with values from other entities."""
|
||||
await setup.async_setup_component(
|
||||
hass,
|
||||
"light",
|
||||
light.DOMAIN,
|
||||
{
|
||||
"light": {
|
||||
"platform": "template",
|
||||
@ -1130,7 +1173,7 @@ async def test_invalid_availability_template_keeps_component_available(hass, cap
|
||||
"""Test that an invalid availability keeps the device available."""
|
||||
await setup.async_setup_component(
|
||||
hass,
|
||||
"light",
|
||||
light.DOMAIN,
|
||||
{
|
||||
"light": {
|
||||
"platform": "template",
|
||||
@ -1170,7 +1213,7 @@ async def test_unique_id(hass):
|
||||
"""Test unique_id option only creates one light per id."""
|
||||
await setup.async_setup_component(
|
||||
hass,
|
||||
"light",
|
||||
light.DOMAIN,
|
||||
{
|
||||
"light": {
|
||||
"platform": "template",
|
||||
|
||||
Loading…
Reference in New Issue
Block a user