home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-08-04 21:25:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

working on unit testing

Browse Source
This commit is contained in:
Brian Cribbs
2017-05-17 16:20:25 -04:00
parent 7693083be7
commit c2fa266aa9
2 changed files with 213 additions and 129 deletions
Download Patch File Download Diff File Expand all files Collapse all files

67
homeassistant/components/light/template.py
View File

@@ -62,6 +62,7 @@ def async_setup_platform(hass, config, async_add_devices, discovery_info=None):
level_action = device_config[LEVEL_ACTION]
level_template = device_config[LEVEL_TEMPLATE]
entity_ids = []
if state_template is not None:
entity_ids = (device_config.get(ATTR_ENTITY_ID) or
state_template.extract_entities())
@@ -98,6 +99,7 @@ class LightTemplate(Light):
self._off_script = Script(hass, off_action)
self._level_script = Script(hass, level_action)
self._level_template = level_template
self._state = False
self._brightness = 0
self._entities = entity_ids
@@ -107,29 +109,6 @@ class LightTemplate(Light):
if self._level_template is not None:
self._level_template.hass = self.hass
@asyncio.coroutine
def async_added_to_hass(self):
"""Register callbacks."""
state = yield from async_get_last_state(self.hass, self.entity_id)
if state:
self._state = state.state == STATE_ON
@callback
def template_light_state_listener(entity, old_state, new_state):
"""Handle target device state changes."""
self.hass.async_add_job(self.async_update_ha_state(True))
@callback
def template_light_startup(event):
"""Update template on startup."""
async_track_state_change(
self.hass, self._entities, template_light_state_listener)
self.hass.async_add_job(self.async_update_ha_state(True))
self.hass.bus.async_listen_once(
EVENT_HOMEASSISTANT_START, template_light_startup)
@property
def brightness(self):
"""Return the brightness of the light."""
@@ -145,29 +124,56 @@ class LightTemplate(Light):
return 0
@property
def is_on(self):
"""Return true if device is on."""
return self._state
@asyncio.coroutine
def async_added_to_hass(self):
"""Register callbacks."""
state = yield from async_get_last_state(self.hass, self.entity_id)
_LOGGER.error("Previous state thawed " + str(state))
if state:
self._state = state.state == STATE_ON
@callback
def template_light_state_listener(entity, old_state, new_state):
"""Handle target device state changes."""
_LOGGER.error("state listener callback")
self.hass.async_add_job(self.async_update_ha_state(True))
@callback
def template_light_startup(event):
"""Update template on startup."""
_LOGGER.error("startup callback")
async_track_state_change(
self.hass, self._entities, template_light_state_listener)
self.hass.async_add_job(self.async_update_ha_state(True))
self.hass.bus.async_listen_once(
EVENT_HOMEASSISTANT_START, template_light_startup)
@asyncio.coroutine
def async_turn_on(self, **kwargs):
"""Turn the light on."""
_LOGGER.error(str(kwargs))
_LOGGER.error(str(self._level_script))
if ATTR_BRIGHTNESS in kwargs and self._level_script:
self.hass.async_add_job(self._level_script.async_run(
{"brightness": kwargs[ATTR_BRIGHTNESS]}))
self._brightness = kwargs[ATTR_BRIGHTNESS]
else:
self.hass.async_add_job(self._on_script.async_run())
self._state = True
self.schedule_update_ha_state(True)
@asyncio.coroutine
def async_turn_off(self, **kwargs):
"""Turn the light off."""
self.hass.async_add_job(self._off_script.async_run())
self._state = False
self.schedule_update_ha_state()
@property
def is_on(self):
"""Return true if device is on."""
return self._state
@asyncio.coroutine
def async_update(self):
@@ -191,3 +197,4 @@ class LightTemplate(Light):
except TemplateError as ex:
_LOGGER.error(ex)
self._state = None

273
tests/components/light/test_template.py
View File

@@ -1,15 +1,18 @@
"""The tests for the Template light platform."""
import logging
import asyncio
from homeassistant.core import callback, State, CoreState
from homeassistant import setup
import homeassistant.components as core
from homeassistant.components.light import (
ATTR_BRIGHTNESS, ATTR_TRANSITION)
from homeassistant.const import STATE_ON, STATE_OFF
from homeassistant.helpers.restore_state import DATA_RESTORE_CACHE
from tests.common import (
get_test_home_assistant, assert_setup_component, mock_component)
_LOGGER = logging.getLogger(__name__)
class TestTemplateLight:
"""Test the Template light."""
@@ -151,7 +154,7 @@ class TestTemplateLight:
def test_template_syntax_error(self):
"""Test templating syntax error."""
with assert_setup_component(1):
with assert_setup_component(0):
assert setup.setup_component(self.hass, 'light', {
'light': {
'platform': 'template',
@@ -186,21 +189,27 @@ class TestTemplateLight:
def test_invalid_name_does_not_create(self):
"""Test invalid name."""
with assert_setup_component(0):
assert setup.setup_component(self.hass, 'switch', {
'switch': {
assert setup.setup_component(self.hass, 'light', {
'light': {
'platform': 'template',
'switches': {
'test INVALID switch': {
'value_template':
"{{ rubbish }",
'lights': {
'bad name here': {
'value_template': "{{ 1== 1}}",
'turn_on': {
'service': 'switch.turn_on',
'entity_id': 'switch.test_state'
'service': 'light.turn_on',
'entity_id': 'light.test_state'
},
'turn_off': {
'service': 'switch.turn_off',
'entity_id': 'switch.test_state'
'service': 'light.turn_off',
'entity_id': 'light.test_state'
},
'set_level': {
'service': 'light.turn_on',
'data_template': {
'entity_id': 'light.test_state',
'brightness': '{{brightness}}'
}
}
}
}
}
@@ -211,14 +220,14 @@ class TestTemplateLight:
assert self.hass.states.all() == []
def test_invalid_switch_does_not_create(self):
"""Test invalid switch."""
def test_invalid_light_does_not_create(self):
"""Test invalid light."""
with assert_setup_component(0):
assert setup.setup_component(self.hass, 'switch', {
'switch': {
assert setup.setup_component(self.hass, 'light', {
'light': {
'platform': 'template',
'switches': {
'test_template_switch': 'Invalid'
'test_template_light': 'Invalid'
}
}
})
@@ -228,11 +237,11 @@ class TestTemplateLight:
assert self.hass.states.all() == []
def test_no_switches_does_not_create(self):
"""Test if there are no switches no creation."""
def test_no_lights_does_not_create(self):
"""Test if there are no lights no creation."""
with assert_setup_component(0):
assert setup.setup_component(self.hass, 'switch', {
'switch': {
assert setup.setup_component(self.hass, 'light', {
'light': {
'platform': 'template'
}
})
@@ -242,24 +251,29 @@ class TestTemplateLight:
assert self.hass.states.all() == []
def test_missing_template_does_not_create(self):
def test_missing_template_does_create(self):
"""Test missing template."""
with assert_setup_component(0):
assert setup.setup_component(self.hass, 'switch', {
'switch': {
with assert_setup_component(1):
assert setup.setup_component(self.hass, 'light', {
'light': {
'platform': 'template',
'switches': {
'test_template_switch': {
'not_value_template':
"{{ states.switch.test_state.state }}",
'lights': {
'light_one': {
'turn_on': {
'service': 'switch.turn_on',
'entity_id': 'switch.test_state'
'service': 'light.turn_on',
'entity_id': 'light.test_state'
},
'turn_off': {
'service': 'switch.turn_off',
'entity_id': 'switch.test_state'
'service': 'light.turn_off',
'entity_id': 'light.test_state'
},
'set_level': {
'service': 'light.turn_on',
'data_template': {
'entity_id': 'light.test_state',
'brightness': '{{brightness}}'
}
}
}
}
}
@@ -268,26 +282,28 @@ class TestTemplateLight:
self.hass.start()
self.hass.block_till_done()
assert self.hass.states.all() == []
assert self.hass.states.all() != []
def test_missing_on_does_not_create(self):
"""Test missing on."""
with assert_setup_component(0):
assert setup.setup_component(self.hass, 'switch', {
'switch': {
assert setup.setup_component(self.hass, 'light', {
'light': {
'platform': 'template',
'switches': {
'test_template_switch': {
'value_template':
"{{ states.switch.test_state.state }}",
'not_on': {
'service': 'switch.turn_on',
'entity_id': 'switch.test_state'
},
'lights': {
'bad name here': {
'value_template': "{{ 1== 1}}",
'turn_off': {
'service': 'switch.turn_off',
'entity_id': 'switch.test_state'
'service': 'light.turn_off',
'entity_id': 'light.test_state'
},
'set_level': {
'service': 'light.turn_on',
'data_template': {
'entity_id': 'light.test_state',
'brightness': '{{brightness}}'
}
}
}
}
}
@@ -301,21 +317,23 @@ class TestTemplateLight:
def test_missing_off_does_not_create(self):
"""Test missing off."""
with assert_setup_component(0):
assert setup.setup_component(self.hass, 'switch', {
'switch': {
assert setup.setup_component(self.hass, 'light', {
'light': {
'platform': 'template',
'switches': {
'test_template_switch': {
'value_template':
"{{ states.switch.test_state.state }}",
'lights': {
'bad name here': {
'value_template': "{{ 1== 1}}",
'turn_on': {
'service': 'switch.turn_on',
'entity_id': 'switch.test_state'
},
'not_off': {
'service': 'switch.turn_off',
'entity_id': 'switch.test_state'
'service': 'light.turn_on',
'entity_id': 'light.test_state'
},
'set_level': {
'service': 'light.turn_on',
'data_template': {
'entity_id': 'light.test_state',
'brightness': '{{brightness}}'
}
}
}
}
}
@@ -328,20 +346,26 @@ class TestTemplateLight:
def test_on_action(self):
"""Test on action."""
assert setup.setup_component(self.hass, 'switch', {
'switch': {
assert setup.setup_component(self.hass, 'light', {
'light': {
'platform': 'template',
'switches': {
'test_template_switch': {
'value_template':
"{{ states.switch.test_state.state }}",
'lights': {
'test_template_light': {
'value_template': "{{states.light.test_state.state}}",
'turn_on': {
'service': 'test.automation'
'service': 'test.automation',
},
'turn_off': {
'service': 'switch.turn_off',
'entity_id': 'switch.test_state'
'service': 'light.turn_off',
'entity_id': 'light.test_state'
},
'set_level': {
'service': 'light.turn_on',
'data_template': {
'entity_id': 'light.test_state',
'brightness': '{{brightness}}'
}
}
}
}
}
@@ -350,34 +374,39 @@ class TestTemplateLight:
self.hass.start()
self.hass.block_till_done()
self.hass.states.set('switch.test_state', STATE_OFF)
self.hass.states.set('light.test_state', STATE_OFF)
self.hass.block_till_done()
state = self.hass.states.get('switch.test_template_switch')
state = self.hass.states.get('light.test_template_light')
assert state.state == STATE_OFF
core.switch.turn_on(self.hass, 'switch.test_template_switch')
core.light.turn_on(self.hass, 'light.test_template_light')
self.hass.block_till_done()
assert len(self.calls) == 1
def test_off_action(self):
"""Test off action."""
assert setup.setup_component(self.hass, 'switch', {
'switch': {
assert setup.setup_component(self.hass, 'light', {
'light': {
'platform': 'template',
'switches': {
'test_template_switch': {
'value_template':
"{{ states.switch.test_state.state }}",
'lights': {
'test_template_light': {
'value_template': "{{states.light.test_state.state}}",
'turn_on': {
'service': 'switch.turn_on',
'entity_id': 'switch.test_state'
'service': 'light.turn_on',
'entity_id': 'light.test_state'
},
'turn_off': {
'service': 'test.automation'
'service': 'test.automation',
},
'set_level': {
'service': 'light.turn_on',
'data_template': {
'entity_id': 'light.test_state',
'brightness': '{{brightness}}'
}
}
}
}
}
@@ -386,54 +415,102 @@ class TestTemplateLight:
self.hass.start()
self.hass.block_till_done()
self.hass.states.set('switch.test_state', STATE_ON)
self.hass.states.set('light.test_state', STATE_ON)
self.hass.block_till_done()
state = self.hass.states.get('switch.test_template_switch')
state = self.hass.states.get('light.test_template_light')
assert state.state == STATE_ON
core.switch.turn_off(self.hass, 'switch.test_template_switch')
core.light.turn_off(self.hass, 'light.test_template_light')
self.hass.block_till_done()
assert len(self.calls) == 1
def test_level_action_no_template(self):
"""Test setting brightness"""
assert setup.setup_component(self.hass, 'light', {
'light': {
'platform': 'template',
'lights': {
'test_template_light': {
'turn_on': {
'service': 'light.turn_on',
'entity_id': 'light.test_state'
},
'turn_off': {
'service': 'light.turn_off',
'entity_id': 'light.test_state'
},
'set_level': {
'service': 'test.automation',
'data_template': {
'entity_id': 'light.test_state',
'brightness': '{{brightness}}'
}
},
'level_template': '{{states.light.test_state.attributes.brightness}}'
}
}
}
})
self.hass.start()
self.hass.block_till_done()
state = self.hass.states.get('light.test_template_light')
assert state.attributes.get('brightness') == None
core.light.turn_on(
self.hass, 'light.test_template_light', **{ATTR_BRIGHTNESS: 124})
self.hass.block_till_done()
state = self.hass.states.get('light.test_template_light')
assert len(self.calls) == 1
assert self.calls[0].data['brightness'] == '124'
# need to add assertions about optimistic state
# assert state.attributes.get('brightness') == 124
@asyncio.coroutine
def test_restore_state(hass):
"""Ensure states are restored on startup."""
hass.data[DATA_RESTORE_CACHE] = {
'switch.test_template_switch':
State('switch.test_template_switch', 'on'),
'light.test_template_light':
State('light.test_template_light', 'on'),
}
hass.state = CoreState.starting
mock_component(hass, 'recorder')
yield from setup.async_setup_component(hass, 'switch', {
'switch': {
yield from setup.async_setup_component(hass, 'light', {
'light': {
'platform': 'template',
'switches': {
'test_template_switch': {
'lights': {
'test_template_light': {
'value_template':
"{{ states.switch.test_state.state }}",
"{{states.light.test_state.state}}",
'turn_on': {
'service': 'switch.turn_on',
'entity_id': 'switch.test_state'
'service': 'test.automation',
},
'turn_off': {
'service': 'switch.turn_off',
'entity_id': 'switch.test_state'
'service': 'light.turn_off',
'entity_id': 'light.test_state'
},
'set_level': {
'service': 'test.automation',
'data_template': {
'entity_id': 'light.test_state',
'brightness': '{{brightness}}'
}
}
}
}
}
})
state = hass.states.get('switch.test_template_switch')
state = hass.states.get('light.test_template_light')
assert state.state == 'on'
yield from hass.async_start()
yield from hass.async_block_till_done()
state = hass.states.get('switch.test_template_switch')
assert state.state == 'unavailable'
state = hass.states.get('light.test_template_light')
assert state.state == 'off'