f195ecca4b
* Moved climate components with tests into platform dirs. * Updated tests from climate component. * Moved binary_sensor components with tests into platform dirs. * Updated tests from binary_sensor component. * Moved calendar components with tests into platform dirs. * Updated tests from calendar component. * Moved camera components with tests into platform dirs. * Updated tests from camera component. * Moved cover components with tests into platform dirs. * Updated tests from cover component. * Moved device_tracker components with tests into platform dirs. * Updated tests from device_tracker component. * Moved fan components with tests into platform dirs. * Updated tests from fan component. * Moved geo_location components with tests into platform dirs. * Updated tests from geo_location component. * Moved image_processing components with tests into platform dirs. * Updated tests from image_processing component. * Moved light components with tests into platform dirs. * Updated tests from light component. * Moved lock components with tests into platform dirs. * Moved media_player components with tests into platform dirs. * Updated tests from media_player component. * Moved scene components with tests into platform dirs. * Moved sensor components with tests into platform dirs. * Updated tests from sensor component. * Moved switch components with tests into platform dirs. * Updated tests from sensor component. * Moved vacuum components with tests into platform dirs. * Updated tests from vacuum component. * Moved weather components with tests into platform dirs. * Fixed __init__.py files * Fixes for stuff moved as part of this branch. * Fix stuff needed to merge with balloob's branch. * Formatting issues. * Missing __init__.py files. * Fix-ups * Fixup * Regenerated requirements. * Linting errors fixed. * Fixed more broken tests. * Missing init files. * Fix broken tests. * More broken tests * There seems to be a thread race condition. I suspect the logger stuff is running in another thread, which means waiting until the aio loop is done is missing the log messages. Used sleep instead because that allows the logger thread to run. I think the api_streams sensor might not be thread safe. * Disabled tests, will remove sensor in #22147 * Updated coverage and codeowners.
395 lines
15 KiB
Python
395 lines
15 KiB
Python
"""The test for the Template sensor platform."""
|
|
from homeassistant.const import EVENT_HOMEASSISTANT_START
|
|
from homeassistant.setup import setup_component, async_setup_component
|
|
|
|
from tests.common import get_test_home_assistant, assert_setup_component
|
|
|
|
|
|
class TestTemplateSensor:
|
|
"""Test the Template sensor."""
|
|
|
|
hass = 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()
|
|
|
|
def teardown_method(self, method):
|
|
"""Stop everything that was started."""
|
|
self.hass.stop()
|
|
|
|
def test_template(self):
|
|
"""Test template."""
|
|
with assert_setup_component(1):
|
|
assert setup_component(self.hass, 'sensor', {
|
|
'sensor': {
|
|
'platform': 'template',
|
|
'sensors': {
|
|
'test_template_sensor': {
|
|
'value_template':
|
|
"It {{ states.sensor.test_state.state }}."
|
|
}
|
|
}
|
|
}
|
|
})
|
|
|
|
self.hass.start()
|
|
self.hass.block_till_done()
|
|
|
|
state = self.hass.states.get('sensor.test_template_sensor')
|
|
assert state.state == 'It .'
|
|
|
|
self.hass.states.set('sensor.test_state', 'Works')
|
|
self.hass.block_till_done()
|
|
state = self.hass.states.get('sensor.test_template_sensor')
|
|
assert state.state == 'It Works.'
|
|
|
|
def test_icon_template(self):
|
|
"""Test icon template."""
|
|
with assert_setup_component(1):
|
|
assert setup_component(self.hass, 'sensor', {
|
|
'sensor': {
|
|
'platform': 'template',
|
|
'sensors': {
|
|
'test_template_sensor': {
|
|
'value_template':
|
|
"{{ states.sensor.test_state.state }}",
|
|
'icon_template':
|
|
"{% if states.sensor.test_state.state == "
|
|
"'Works' %}"
|
|
"mdi:check"
|
|
"{% endif %}"
|
|
}
|
|
}
|
|
}
|
|
})
|
|
|
|
self.hass.start()
|
|
self.hass.block_till_done()
|
|
|
|
state = self.hass.states.get('sensor.test_template_sensor')
|
|
assert state.attributes.get('icon') == ''
|
|
|
|
self.hass.states.set('sensor.test_state', 'Works')
|
|
self.hass.block_till_done()
|
|
state = self.hass.states.get('sensor.test_template_sensor')
|
|
assert state.attributes['icon'] == 'mdi:check'
|
|
|
|
def test_entity_picture_template(self):
|
|
"""Test entity_picture template."""
|
|
with assert_setup_component(1):
|
|
assert setup_component(self.hass, 'sensor', {
|
|
'sensor': {
|
|
'platform': 'template',
|
|
'sensors': {
|
|
'test_template_sensor': {
|
|
'value_template':
|
|
"{{ states.sensor.test_state.state }}",
|
|
'entity_picture_template':
|
|
"{% if states.sensor.test_state.state == "
|
|
"'Works' %}"
|
|
"/local/sensor.png"
|
|
"{% endif %}"
|
|
}
|
|
}
|
|
}
|
|
})
|
|
|
|
self.hass.start()
|
|
self.hass.block_till_done()
|
|
|
|
state = self.hass.states.get('sensor.test_template_sensor')
|
|
assert state.attributes.get('entity_picture') == ''
|
|
|
|
self.hass.states.set('sensor.test_state', 'Works')
|
|
self.hass.block_till_done()
|
|
state = self.hass.states.get('sensor.test_template_sensor')
|
|
assert state.attributes['entity_picture'] == '/local/sensor.png'
|
|
|
|
def test_friendly_name_template(self):
|
|
"""Test friendly_name template."""
|
|
with assert_setup_component(1):
|
|
assert setup_component(self.hass, 'sensor', {
|
|
'sensor': {
|
|
'platform': 'template',
|
|
'sensors': {
|
|
'test_template_sensor': {
|
|
'value_template':
|
|
"{{ states.sensor.test_state.state }}",
|
|
'friendly_name_template':
|
|
"It {{ states.sensor.test_state.state }}."
|
|
}
|
|
}
|
|
}
|
|
})
|
|
|
|
self.hass.start()
|
|
self.hass.block_till_done()
|
|
|
|
state = self.hass.states.get('sensor.test_template_sensor')
|
|
assert state.attributes.get('friendly_name') == 'It .'
|
|
|
|
self.hass.states.set('sensor.test_state', 'Works')
|
|
self.hass.block_till_done()
|
|
state = self.hass.states.get('sensor.test_template_sensor')
|
|
assert state.attributes['friendly_name'] == 'It Works.'
|
|
|
|
def test_friendly_name_template_with_unknown_state(self):
|
|
"""Test friendly_name template with an unknown value_template."""
|
|
with assert_setup_component(1):
|
|
assert setup_component(self.hass, 'sensor', {
|
|
'sensor': {
|
|
'platform': 'template',
|
|
'sensors': {
|
|
'test_template_sensor': {
|
|
'value_template': "{{ states.fourohfour.state }}",
|
|
'friendly_name_template':
|
|
"It {{ states.sensor.test_state.state }}."
|
|
}
|
|
}
|
|
}
|
|
})
|
|
|
|
self.hass.start()
|
|
self.hass.block_till_done()
|
|
|
|
state = self.hass.states.get('sensor.test_template_sensor')
|
|
assert state.attributes['friendly_name'] == 'It .'
|
|
|
|
self.hass.states.set('sensor.test_state', 'Works')
|
|
self.hass.block_till_done()
|
|
state = self.hass.states.get('sensor.test_template_sensor')
|
|
assert state.attributes['friendly_name'] == 'It Works.'
|
|
|
|
def test_template_syntax_error(self):
|
|
"""Test templating syntax error."""
|
|
with assert_setup_component(0):
|
|
assert setup_component(self.hass, 'sensor', {
|
|
'sensor': {
|
|
'platform': 'template',
|
|
'sensors': {
|
|
'test_template_sensor': {
|
|
'value_template':
|
|
"{% if rubbish %}"
|
|
}
|
|
}
|
|
}
|
|
})
|
|
|
|
self.hass.start()
|
|
self.hass.block_till_done()
|
|
assert self.hass.states.all() == []
|
|
|
|
def test_template_attribute_missing(self):
|
|
"""Test missing attribute template."""
|
|
with assert_setup_component(1):
|
|
assert setup_component(self.hass, 'sensor', {
|
|
'sensor': {
|
|
'platform': 'template',
|
|
'sensors': {
|
|
'test_template_sensor': {
|
|
'value_template': 'It {{ states.sensor.test_state'
|
|
'.attributes.missing }}.'
|
|
}
|
|
}
|
|
}
|
|
})
|
|
|
|
self.hass.start()
|
|
self.hass.block_till_done()
|
|
|
|
state = self.hass.states.get('sensor.test_template_sensor')
|
|
assert state.state == 'unknown'
|
|
|
|
def test_invalid_name_does_not_create(self):
|
|
"""Test invalid name."""
|
|
with assert_setup_component(0):
|
|
assert setup_component(self.hass, 'sensor', {
|
|
'sensor': {
|
|
'platform': 'template',
|
|
'sensors': {
|
|
'test INVALID sensor': {
|
|
'value_template':
|
|
"{{ states.sensor.test_state.state }}"
|
|
}
|
|
}
|
|
}
|
|
})
|
|
|
|
self.hass.start()
|
|
self.hass.block_till_done()
|
|
|
|
assert self.hass.states.all() == []
|
|
|
|
def test_invalid_sensor_does_not_create(self):
|
|
"""Test invalid sensor."""
|
|
with assert_setup_component(0):
|
|
assert setup_component(self.hass, 'sensor', {
|
|
'sensor': {
|
|
'platform': 'template',
|
|
'sensors': {
|
|
'test_template_sensor': 'invalid'
|
|
}
|
|
}
|
|
})
|
|
|
|
self.hass.start()
|
|
|
|
assert self.hass.states.all() == []
|
|
|
|
def test_no_sensors_does_not_create(self):
|
|
"""Test no sensors."""
|
|
with assert_setup_component(0):
|
|
assert setup_component(self.hass, 'sensor', {
|
|
'sensor': {
|
|
'platform': 'template'
|
|
}
|
|
})
|
|
|
|
self.hass.start()
|
|
self.hass.block_till_done()
|
|
|
|
assert self.hass.states.all() == []
|
|
|
|
def test_missing_template_does_not_create(self):
|
|
"""Test missing template."""
|
|
with assert_setup_component(0):
|
|
assert setup_component(self.hass, 'sensor', {
|
|
'sensor': {
|
|
'platform': 'template',
|
|
'sensors': {
|
|
'test_template_sensor': {
|
|
'not_value_template':
|
|
"{{ states.sensor.test_state.state }}"
|
|
}
|
|
}
|
|
}
|
|
})
|
|
|
|
self.hass.start()
|
|
self.hass.block_till_done()
|
|
|
|
assert self.hass.states.all() == []
|
|
|
|
def test_setup_invalid_device_class(self):
|
|
"""Test setup with invalid device_class."""
|
|
with assert_setup_component(0):
|
|
assert setup_component(self.hass, 'sensor', {
|
|
'sensor': {
|
|
'platform': 'template',
|
|
'sensors': {
|
|
'test': {
|
|
'value_template':
|
|
'{{ states.sensor.test_sensor.state }}',
|
|
'device_class': 'foobarnotreal',
|
|
},
|
|
},
|
|
}
|
|
})
|
|
|
|
def test_setup_valid_device_class(self):
|
|
"""Test setup with valid device_class."""
|
|
with assert_setup_component(1):
|
|
assert setup_component(self.hass, 'sensor', {
|
|
'sensor': {
|
|
'platform': 'template',
|
|
'sensors': {
|
|
'test1': {
|
|
'value_template':
|
|
'{{ states.sensor.test_sensor.state }}',
|
|
'device_class': 'temperature',
|
|
},
|
|
'test2': {
|
|
'value_template':
|
|
'{{ states.sensor.test_sensor.state }}'
|
|
},
|
|
}
|
|
}
|
|
})
|
|
self.hass.block_till_done()
|
|
|
|
state = self.hass.states.get('sensor.test1')
|
|
assert state.attributes['device_class'] == 'temperature'
|
|
state = self.hass.states.get('sensor.test2')
|
|
assert 'device_class' not in state.attributes
|
|
|
|
|
|
async def test_no_template_match_all(hass, caplog):
|
|
"""Test that we do not allow sensors that match on all."""
|
|
hass.states.async_set('sensor.test_sensor', 'startup')
|
|
|
|
await async_setup_component(hass, 'sensor', {
|
|
'sensor': {
|
|
'platform': 'template',
|
|
'sensors': {
|
|
'invalid_state': {
|
|
'value_template': '{{ 1 + 1 }}',
|
|
},
|
|
'invalid_icon': {
|
|
'value_template':
|
|
'{{ states.sensor.test_sensor.state }}',
|
|
'icon_template': '{{ 1 + 1 }}',
|
|
},
|
|
'invalid_entity_picture': {
|
|
'value_template':
|
|
'{{ states.sensor.test_sensor.state }}',
|
|
'entity_picture_template': '{{ 1 + 1 }}',
|
|
},
|
|
'invalid_friendly_name': {
|
|
'value_template':
|
|
'{{ states.sensor.test_sensor.state }}',
|
|
'friendly_name_template': '{{ 1 + 1 }}',
|
|
},
|
|
}
|
|
}
|
|
})
|
|
await hass.async_block_till_done()
|
|
assert len(hass.states.async_all()) == 5
|
|
assert ('Template sensor invalid_state has no entity ids '
|
|
'configured to track nor were we able to extract the entities to '
|
|
'track from the value template') in caplog.text
|
|
assert ('Template sensor invalid_icon has no entity ids '
|
|
'configured to track nor were we able to extract the entities to '
|
|
'track from the icon template') in caplog.text
|
|
assert ('Template sensor invalid_entity_picture has no entity ids '
|
|
'configured to track nor were we able to extract the entities to '
|
|
'track from the entity_picture template') in caplog.text
|
|
assert ('Template sensor invalid_friendly_name has no entity ids '
|
|
'configured to track nor were we able to extract the entities to '
|
|
'track from the friendly_name template') in caplog.text
|
|
|
|
assert hass.states.get('sensor.invalid_state').state == 'unknown'
|
|
assert hass.states.get('sensor.invalid_icon').state == 'unknown'
|
|
assert hass.states.get('sensor.invalid_entity_picture').state == 'unknown'
|
|
assert hass.states.get('sensor.invalid_friendly_name').state == 'unknown'
|
|
|
|
hass.bus.async_fire(EVENT_HOMEASSISTANT_START)
|
|
await hass.async_block_till_done()
|
|
|
|
assert hass.states.get('sensor.invalid_state').state == '2'
|
|
assert hass.states.get('sensor.invalid_icon').state == 'startup'
|
|
assert hass.states.get('sensor.invalid_entity_picture').state == 'startup'
|
|
assert hass.states.get('sensor.invalid_friendly_name').state == 'startup'
|
|
|
|
hass.states.async_set('sensor.test_sensor', 'hello')
|
|
await hass.async_block_till_done()
|
|
|
|
assert hass.states.get('sensor.invalid_state').state == '2'
|
|
assert hass.states.get('sensor.invalid_icon').state == 'startup'
|
|
assert hass.states.get('sensor.invalid_entity_picture').state == 'startup'
|
|
assert hass.states.get('sensor.invalid_friendly_name').state == 'startup'
|
|
|
|
await hass.helpers.entity_component.async_update_entity(
|
|
'sensor.invalid_state')
|
|
await hass.helpers.entity_component.async_update_entity(
|
|
'sensor.invalid_icon')
|
|
await hass.helpers.entity_component.async_update_entity(
|
|
'sensor.invalid_entity_picture')
|
|
await hass.helpers.entity_component.async_update_entity(
|
|
'sensor.invalid_friendly_name')
|
|
|
|
assert hass.states.get('sensor.invalid_state').state == '2'
|
|
assert hass.states.get('sensor.invalid_icon').state == 'hello'
|
|
assert hass.states.get('sensor.invalid_entity_picture').state == 'hello'
|
|
assert hass.states.get('sensor.invalid_friendly_name').state == 'hello'
|