epenet
79 lines
2.9 KiB
Python
79 lines
2.9 KiB
Python
"""Test state wrapper classes and helpers for Home Assistant templates."""
|
|
|
|
from datetime import timedelta
|
|
|
|
import pytest
|
|
|
|
from homeassistant.const import STATE_ON
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import template
|
|
from homeassistant.helpers.json import json_dumps
|
|
from homeassistant.helpers.template import states as template_states
|
|
from homeassistant.util import dt as dt_util
|
|
|
|
from tests.common import async_fire_time_changed
|
|
|
|
|
|
async def test_template_states_blocks_setitem(hass: HomeAssistant) -> None:
|
|
"""Test we cannot setitem on TemplateStates."""
|
|
hass.states.async_set("light.new", STATE_ON)
|
|
state = hass.states.get("light.new")
|
|
template_state = template.TemplateState(hass, state, True)
|
|
with pytest.raises(RuntimeError):
|
|
template_state["any"] = "any"
|
|
|
|
|
|
async def test_template_states_can_serialize(hass: HomeAssistant) -> None:
|
|
"""Test TemplateState is serializable."""
|
|
hass.states.async_set("light.new", STATE_ON)
|
|
state = hass.states.get("light.new")
|
|
template_state = template.TemplateState(hass, state, True)
|
|
assert template_state.as_dict() is template_state.as_dict()
|
|
assert json_dumps(template_state) == json_dumps(template_state)
|
|
|
|
|
|
async def test_lru_increases_with_many_entities(hass: HomeAssistant) -> None:
|
|
"""Test that the template internal LRU cache increases with many entities."""
|
|
# We do not actually want to record 4096 entities so we mock the entity count
|
|
mock_entity_count = 16
|
|
|
|
assert (
|
|
template_states.CACHED_TEMPLATE_LRU.get_size()
|
|
== template_states.CACHED_TEMPLATE_STATES
|
|
)
|
|
assert (
|
|
template_states.CACHED_TEMPLATE_NO_COLLECT_LRU.get_size()
|
|
== template_states.CACHED_TEMPLATE_STATES
|
|
)
|
|
template_states.CACHED_TEMPLATE_LRU.set_size(8)
|
|
template_states.CACHED_TEMPLATE_NO_COLLECT_LRU.set_size(8)
|
|
|
|
template.async_setup(hass)
|
|
for i in range(mock_entity_count):
|
|
hass.states.async_set(f"sensor.sensor{i}", "on")
|
|
|
|
async_fire_time_changed(hass, dt_util.utcnow() + timedelta(minutes=10))
|
|
await hass.async_block_till_done()
|
|
|
|
assert template_states.CACHED_TEMPLATE_LRU.get_size() == int(
|
|
round(mock_entity_count * template_states.ENTITY_COUNT_GROWTH_FACTOR)
|
|
)
|
|
assert template_states.CACHED_TEMPLATE_NO_COLLECT_LRU.get_size() == int(
|
|
round(mock_entity_count * template_states.ENTITY_COUNT_GROWTH_FACTOR)
|
|
)
|
|
|
|
await hass.async_stop()
|
|
|
|
for i in range(mock_entity_count):
|
|
hass.states.async_set(f"sensor.sensor_add_{i}", "on")
|
|
|
|
async_fire_time_changed(hass, dt_util.utcnow() + timedelta(minutes=20))
|
|
await hass.async_block_till_done()
|
|
|
|
assert template_states.CACHED_TEMPLATE_LRU.get_size() == int(
|
|
round(mock_entity_count * template_states.ENTITY_COUNT_GROWTH_FACTOR)
|
|
)
|
|
assert template_states.CACHED_TEMPLATE_NO_COLLECT_LRU.get_size() == int(
|
|
round(mock_entity_count * template_states.ENTITY_COUNT_GROWTH_FACTOR)
|
|
)
|