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

Adds states and state_attr as a filter, adds is_state and is_state_attr as a test. (#79473)

Browse Source
This commit is contained in:
Petro31
2022-10-25 13:49:51 -04:00
committed by GitHub
parent d3ada34498
commit f73fc9e355
2 changed files with 72 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

68
tests/helpers/test_template.py
View File

@@ -1348,6 +1348,22 @@ def test_is_state(hass):
)
assert tpl.async_render() is False
tpl = template.Template(
"""
{% if "test.object" is is_state("available") %}yes{% else %}no{% endif %}
""",
hass,
)
assert tpl.async_render() == "yes"
tpl = template.Template(
"""
{{ ['test.object'] | select("is_state", "available") | first | default }}
""",
hass,
)
assert tpl.async_render() == "test.object"
def test_is_state_attr(hass):
"""Test is_state_attr method."""
@@ -1368,10 +1384,28 @@ def test_is_state_attr(hass):
)
assert tpl.async_render() is False
tpl = template.Template(
"""
{% if "test.object" is is_state_attr("mode", "on") %}yes{% else %}no{% endif %}
""",
hass,
)
assert tpl.async_render() == "yes"
tpl = template.Template(
"""
{{ ['test.object'] | select("is_state_attr", "mode", "on") | first | default }}
""",
hass,
)
assert tpl.async_render() == "test.object"
def test_state_attr(hass):
"""Test state_attr method."""
hass.states.async_set("test.object", "available", {"mode": "on"})
hass.states.async_set(
"test.object", "available", {"effect": "action", "mode": "on"}
)
tpl = template.Template(
"""
{% if state_attr("test.object", "mode") == "on" %}yes{% else %}no{% endif %}
@@ -1388,6 +1422,22 @@ def test_state_attr(hass):
)
assert tpl.async_render() is True
tpl = template.Template(
"""
{% if "test.object" | state_attr("mode") == "on" %}yes{% else %}no{% endif %}
""",
hass,
)
assert tpl.async_render() == "yes"
tpl = template.Template(
"""
{{ ['test.object'] | map("state_attr", "effect") | first | default }}
""",
hass,
)
assert tpl.async_render() == "action"
def test_states_function(hass):
"""Test using states as a function."""
@@ -1398,6 +1448,22 @@ def test_states_function(hass):
tpl2 = template.Template('{{ states("test.object2") }}', hass)
assert tpl2.async_render() == "unknown"
tpl = template.Template(
"""
{% if "test.object" | states == "available" %}yes{% else %}no{% endif %}
""",
hass,
)
assert tpl.async_render() == "yes"
tpl = template.Template(
"""
{{ ['test.object'] | map("states") | first | default }}
""",
hass,
)
assert tpl.async_render() == "available"
@patch(
"homeassistant.helpers.template.TemplateEnvironment.is_safe_callable",