jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-11-12 12:30:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Streamline template tests (#154586)

Browse Source
This commit is contained in:
Aarni Koskela
2025-10-19 02:24:58 +03:00
committed by GitHub
parent ef69e6d54b
commit 24b7cf261c
8 changed files with 851 additions and 1576 deletions
Download Patch File Download Diff File Expand all files Collapse all files

249
tests/helpers/template/extensions/test_regex.py
View File

@@ -6,260 +6,169 @@ import pytest
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import TemplateError
from homeassistant.helpers import template
from tests.helpers.template.helpers import render
def test_regex_match(hass: HomeAssistant) -> None:
"""Test regex_match method."""
tpl = template.Template(
r"""
{{ '123-456-7890' | regex_match('(\\d{3})-(\\d{3})-(\\d{4})') }}
""",
hass,
)
assert tpl.async_render() is True
tpl = template.Template(
"""
{{ 'Home Assistant test' | regex_match('home', True) }}
""",
hass,
result = render(
hass, r"""{{ '123-456-7890' | regex_match('(\\d{3})-(\\d{3})-(\\d{4})') }}"""
)
assert tpl.async_render() is True
assert result is True
tpl = template.Template(
"""
{{ 'Another Home Assistant test' | regex_match('Home') }}
""",
hass,
)
assert tpl.async_render() is False
result = render(hass, """{{ 'Home Assistant test' | regex_match('home', True) }}""")
assert result is True
tpl = template.Template(
"""
{{ ['Home Assistant test'] | regex_match('.*Assist') }}
""",
hass,
)
assert tpl.async_render() is True
result = render(hass, """{{ 'Another Home Assistant test'|regex_match('Home') }}""")
assert result is False
result = render(hass, """{{ ['Home Assistant test'] | regex_match('.*Assist') }}""")
assert result is True
def test_match_test(hass: HomeAssistant) -> None:
"""Test match test."""
tpl = template.Template(
r"""
{{ '123-456-7890' is match('(\\d{3})-(\\d{3})-(\\d{4})') }}
""",
hass,
result = render(
hass, r"""{{ '123-456-7890' is match('(\\d{3})-(\\d{3})-(\\d{4})') }}"""
)
assert tpl.async_render() is True
assert result is True
def test_regex_search(hass: HomeAssistant) -> None:
"""Test regex_search method."""
tpl = template.Template(
r"""
{{ '123-456-7890' | regex_search('(\\d{3})-(\\d{3})-(\\d{4})') }}
""",
hass,
)
assert tpl.async_render() is True
tpl = template.Template(
"""
{{ 'Home Assistant test' | regex_search('home', True) }}
""",
hass,
result = render(
hass, r"""{{ '123-456-7890' | regex_search('(\\d{3})-(\\d{3})-(\\d{4})') }}"""
)
assert tpl.async_render() is True
assert result is True
tpl = template.Template(
"""
{{ 'Another Home Assistant test' | regex_search('Home') }}
""",
hass,
result = render(
hass, """{{ 'Home Assistant test' | regex_search('home', True) }}"""
)
assert tpl.async_render() is True
assert result is True
tpl = template.Template(
"""
{{ ['Home Assistant test'] | regex_search('Assist') }}
""",
hass,
result = render(
hass, """ {{ 'Another Home Assistant test' | regex_search('Home') }}"""
)
assert tpl.async_render() is True
assert result is True
result = render(hass, """{{ ['Home Assistant test'] | regex_search('Assist') }}""")
assert result is True
def test_search_test(hass: HomeAssistant) -> None:
"""Test search test."""
tpl = template.Template(
r"""
{{ '123-456-7890' is search('(\\d{3})-(\\d{3})-(\\d{4})') }}
""",
hass,
result = render(
hass, r"""{{ '123-456-7890' is search('(\\d{3})-(\\d{3})-(\\d{4})') }}"""
)
assert tpl.async_render() is True
assert result is True
def test_regex_replace(hass: HomeAssistant) -> None:
"""Test regex_replace method."""
tpl = template.Template(
r"""
{{ 'Hello World' | regex_replace('(Hello\\s)',) }}
""",
hass,
)
assert tpl.async_render() == "World"
tpl = template.Template(
"""
{{ ['Home hinderant test'] | regex_replace('hinder', 'Assist') }}
""",
hass,
result = render(hass, r"""{{ 'Hello World' | regex_replace('(Hello\\s)',) }}""")
assert result == "World"
result = render(
hass, """{{ ['Home hinderant test'] | regex_replace('hinder', 'Assist') }}"""
)
assert tpl.async_render() == ["Home Assistant test"]
assert result == ["Home Assistant test"]
def test_regex_findall(hass: HomeAssistant) -> None:
"""Test regex_findall method."""
tpl = template.Template(
"""
{{ 'Flight from JFK to LHR' | regex_findall('([A-Z]{3})') }}
""",
hass,
result = render(
hass, """{{ 'Flight from JFK to LHR' | regex_findall('([A-Z]{3})') }}"""
)
assert tpl.async_render() == ["JFK", "LHR"]
assert result == ["JFK", "LHR"]
def test_regex_findall_index(hass: HomeAssistant) -> None:
"""Test regex_findall_index method."""
tpl = template.Template(
"""
{{ 'Flight from JFK to LHR' | regex_findall_index('([A-Z]{3})', 0) }}
""",
hass,
)
assert tpl.async_render() == "JFK"
tpl = template.Template(
"""
{{ 'Flight from JFK to LHR' | regex_findall_index('([A-Z]{3})', 1) }}
""",
result = render(
hass,
"""{{ 'Flight from JFK to LHR' | regex_findall_index('([A-Z]{3})', 0) }}""",
)
assert tpl.async_render() == "LHR"
assert result == "JFK"
result = render(
hass,
"""{{ 'Flight from JFK to LHR' | regex_findall_index('([A-Z]{3})', 1) }}""",
)
assert result == "LHR"
def test_regex_ignorecase_parameter(hass: HomeAssistant) -> None:
"""Test ignorecase parameter across all regex functions."""
# Test regex_match with ignorecase
tpl = template.Template(
"""
{{ 'TEST' | regex_match('test', True) }}
""",
hass,
)
assert tpl.async_render() is True
result = render(hass, """{{ 'TEST' | regex_match('test', True) }}""")
assert result is True
# Test regex_search with ignorecase
tpl = template.Template(
"""
{{ 'TEST STRING' | regex_search('test', True) }}
""",
hass,
)
assert tpl.async_render() is True
result = render(hass, """{{ 'TEST STRING' | regex_search('test', True) }}""")
assert result is True
# Test regex_replace with ignorecase
tpl = template.Template(
"""
{{ 'TEST' | regex_replace('test', 'replaced', True) }}
""",
hass,
)
assert tpl.async_render() == "replaced"
result = render(hass, """{{ 'TEST' | regex_replace('test', 'replaced', True) }}""")
assert result == "replaced"
# Test regex_findall with ignorecase
tpl = template.Template(
"""
{{ 'TEST test Test' | regex_findall('test', True) }}
""",
hass,
)
assert tpl.async_render() == ["TEST", "test", "Test"]
result = render(hass, """{{ 'TEST test Test' | regex_findall('test', True) }}""")
assert result == ["TEST", "test", "Test"]
def test_regex_with_non_string_input(hass: HomeAssistant) -> None:
"""Test regex functions with non-string input (automatic conversion)."""
# Test with integer
tpl = template.Template(
r"""
{{ 12345 | regex_match('\\d+') }}
""",
hass,
)
assert tpl.async_render() is True
result = render(hass, r"""{{ 12345 | regex_match('\\d+') }}""")
assert result is True
# Test with list (string conversion)
tpl = template.Template(
r"""
{{ [1, 2, 3] | regex_search('\\d') }}
""",
hass,
)
assert tpl.async_render() is True
result = render(hass, r"""{{ [1, 2, 3] | regex_search('\\d') }}""")
assert result is True
def test_regex_edge_cases(hass: HomeAssistant) -> None:
"""Test regex functions with edge cases."""
# Test with empty string
tpl = template.Template(
"""
{{ '' | regex_match('.*') }}
""",
hass,
)
assert tpl.async_render() is True
assert render(hass, """{{ '' | regex_match('.*') }}""") is True
# Test regex_findall_index with out of bounds index
tpl = template.Template(
"""
{{ 'test' | regex_findall_index('t', 5) }}
""",
hass,
)
with pytest.raises(TemplateError):
tpl.async_render()
render(hass, """{{ 'test' | regex_findall_index('t', 5) }}""")
# Test with invalid regex pattern
tpl = template.Template(
"""
{{ 'test' | regex_match('[') }}
""",
hass,
)
with pytest.raises(TemplateError): # re.error wrapped in TemplateError
tpl.async_render()
render(hass, """{{ 'test' | regex_match('[') }}""")
def test_regex_groups_and_replacement_patterns(hass: HomeAssistant) -> None:
"""Test regex with groups and replacement patterns."""
# Test replacement with groups
tpl = template.Template(
r"""
{{ 'John Doe' | regex_replace('(\\w+) (\\w+)', '\\2, \\1') }}
""",
hass,
result = render(
hass, r"""{{ 'John Doe' | regex_replace('(\\w+) (\\w+)', '\\2, \\1') }}"""
)
assert tpl.async_render() == "Doe, John"
assert result == "Doe, John"
# Test findall with groups
tpl = template.Template(
r"""
{{ 'Email: test@example.com, Phone: 123-456-7890' | regex_findall('(\\w+@\\w+\\.\\w+)|(\\d{3}-\\d{3}-\\d{4})') }}
""",
result = render(
hass,
r"""{{ 'Email: test@example.com, Phone: 123-456-7890' | regex_findall('(\\w+@\\w+\\.\\w+)|(\\d{3}-\\d{3}-\\d{4})') }}""",
)
result = tpl.async_render()
# The result will contain tuples with empty strings for non-matching groups
assert len(result) == 2