Move script component tests to script helper tests
This commit is contained in:
Paulus Schoutsen
parent
f76d545a08
commit
09a771a026
2 changed files with 177 additions and 197 deletions
|
|
@ -1,19 +1,17 @@
|
||||||
"""The tests for the Script component."""
|
"""The tests for the Script component."""
|
||||||
# pylint: disable=too-many-public-methods,protected-access
|
# pylint: disable=too-many-public-methods,protected-access
|
||||||
from datetime import timedelta
|
|
||||||
import unittest
|
import unittest
|
||||||
|
|
||||||
from homeassistant.bootstrap import _setup_component
|
from homeassistant.bootstrap import _setup_component
|
||||||
from homeassistant.components import script
|
from homeassistant.components import script
|
||||||
import homeassistant.util.dt as dt_util
|
|
||||||
|
|
||||||
from tests.common import fire_time_changed, get_test_home_assistant
|
from tests.common import get_test_home_assistant
|
||||||
|
|
||||||
|
|
||||||
ENTITY_ID = 'script.test'
|
ENTITY_ID = 'script.test'
|
||||||
|
|
||||||
|
|
||||||
class TestScript(unittest.TestCase):
|
class TestScriptComponent(unittest.TestCase):
|
||||||
"""Test the Script component."""
|
"""Test the Script component."""
|
||||||
|
|
||||||
def setUp(self): # pylint: disable=invalid-name
|
def setUp(self): # pylint: disable=invalid-name
|
||||||
|
|
@ -49,199 +47,6 @@ class TestScript(unittest.TestCase):
|
||||||
|
|
||||||
self.assertEqual(0, len(self.hass.states.entity_ids('script')))
|
self.assertEqual(0, len(self.hass.states.entity_ids('script')))
|
||||||
|
|
||||||
def test_firing_event(self):
|
|
||||||
"""Test the firing of events."""
|
|
||||||
event = 'test_event'
|
|
||||||
calls = []
|
|
||||||
|
|
||||||
def record_event(event):
|
|
||||||
"""Add recorded event to set."""
|
|
||||||
calls.append(event)
|
|
||||||
|
|
||||||
self.hass.bus.listen(event, record_event)
|
|
||||||
|
|
||||||
assert _setup_component(self.hass, 'script', {
|
|
||||||
'script': {
|
|
||||||
'test': {
|
|
||||||
'alias': 'Test Script',
|
|
||||||
'sequence': [{
|
|
||||||
'event': event,
|
|
||||||
'event_data': {
|
|
||||||
'hello': 'world'
|
|
||||||
}
|
|
||||||
}]
|
|
||||||
}
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
script.turn_on(self.hass, ENTITY_ID)
|
|
||||||
self.hass.pool.block_till_done()
|
|
||||||
|
|
||||||
self.assertEqual(1, len(calls))
|
|
||||||
self.assertEqual('world', calls[0].data.get('hello'))
|
|
||||||
self.assertIsNone(
|
|
||||||
self.hass.states.get(ENTITY_ID).attributes.get('can_cancel'))
|
|
||||||
|
|
||||||
def test_calling_service(self):
|
|
||||||
"""Test the calling of a service."""
|
|
||||||
calls = []
|
|
||||||
|
|
||||||
def record_call(service):
|
|
||||||
"""Add recorded event to set."""
|
|
||||||
calls.append(service)
|
|
||||||
|
|
||||||
self.hass.services.register('test', 'script', record_call)
|
|
||||||
|
|
||||||
assert _setup_component(self.hass, 'script', {
|
|
||||||
'script': {
|
|
||||||
'test': {
|
|
||||||
'sequence': [{
|
|
||||||
'service': 'test.script',
|
|
||||||
'data': {
|
|
||||||
'hello': 'world'
|
|
||||||
}
|
|
||||||
}]
|
|
||||||
}
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
script.turn_on(self.hass, ENTITY_ID)
|
|
||||||
self.hass.pool.block_till_done()
|
|
||||||
|
|
||||||
self.assertEqual(1, len(calls))
|
|
||||||
self.assertEqual('world', calls[0].data.get('hello'))
|
|
||||||
|
|
||||||
def test_calling_service_template(self):
|
|
||||||
"""Test the calling of a service."""
|
|
||||||
calls = []
|
|
||||||
|
|
||||||
def record_call(service):
|
|
||||||
"""Add recorded event to set."""
|
|
||||||
calls.append(service)
|
|
||||||
|
|
||||||
self.hass.services.register('test', 'script', record_call)
|
|
||||||
|
|
||||||
assert _setup_component(self.hass, 'script', {
|
|
||||||
'script': {
|
|
||||||
'test': {
|
|
||||||
'sequence': [{
|
|
||||||
'service_template': """
|
|
||||||
{% if True %}
|
|
||||||
test.script
|
|
||||||
{% else %}
|
|
||||||
test.not_script
|
|
||||||
{% endif %}""",
|
|
||||||
'data_template': {
|
|
||||||
'hello': """
|
|
||||||
{% if True %}
|
|
||||||
world
|
|
||||||
{% else %}
|
|
||||||
Not world
|
|
||||||
{% endif %}
|
|
||||||
"""
|
|
||||||
}
|
|
||||||
}]
|
|
||||||
}
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
script.turn_on(self.hass, ENTITY_ID)
|
|
||||||
self.hass.pool.block_till_done()
|
|
||||||
|
|
||||||
self.assertEqual(1, len(calls))
|
|
||||||
self.assertEqual('world', calls[0].data.get('hello'))
|
|
||||||
|
|
||||||
def test_delay(self):
|
|
||||||
"""Test the delay."""
|
|
||||||
event = 'test_event'
|
|
||||||
calls = []
|
|
||||||
|
|
||||||
def record_event(event):
|
|
||||||
"""Add recorded event to set."""
|
|
||||||
calls.append(event)
|
|
||||||
|
|
||||||
self.hass.bus.listen(event, record_event)
|
|
||||||
|
|
||||||
assert _setup_component(self.hass, 'script', {
|
|
||||||
'script': {
|
|
||||||
'test': {
|
|
||||||
'sequence': [{
|
|
||||||
'event': event
|
|
||||||
}, {
|
|
||||||
'delay': {
|
|
||||||
'seconds': 5
|
|
||||||
}
|
|
||||||
}, {
|
|
||||||
'event': event,
|
|
||||||
}]
|
|
||||||
}
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
script.turn_on(self.hass, ENTITY_ID)
|
|
||||||
self.hass.pool.block_till_done()
|
|
||||||
|
|
||||||
self.assertTrue(script.is_on(self.hass, ENTITY_ID))
|
|
||||||
self.assertTrue(
|
|
||||||
self.hass.states.get(ENTITY_ID).attributes.get('can_cancel'))
|
|
||||||
|
|
||||||
self.assertEqual(
|
|
||||||
event,
|
|
||||||
self.hass.states.get(ENTITY_ID).attributes.get('last_action'))
|
|
||||||
self.assertEqual(1, len(calls))
|
|
||||||
|
|
||||||
future = dt_util.utcnow() + timedelta(seconds=5)
|
|
||||||
fire_time_changed(self.hass, future)
|
|
||||||
self.hass.pool.block_till_done()
|
|
||||||
|
|
||||||
self.assertFalse(script.is_on(self.hass, ENTITY_ID))
|
|
||||||
|
|
||||||
self.assertEqual(2, len(calls))
|
|
||||||
|
|
||||||
def test_cancel_while_delay(self):
|
|
||||||
"""Test the cancelling while the delay is present."""
|
|
||||||
event = 'test_event'
|
|
||||||
calls = []
|
|
||||||
|
|
||||||
def record_event(event):
|
|
||||||
"""Add recorded event to set."""
|
|
||||||
calls.append(event)
|
|
||||||
|
|
||||||
self.hass.bus.listen(event, record_event)
|
|
||||||
|
|
||||||
assert _setup_component(self.hass, 'script', {
|
|
||||||
'script': {
|
|
||||||
'test': {
|
|
||||||
'sequence': [{
|
|
||||||
'delay': {
|
|
||||||
'seconds': 5
|
|
||||||
}
|
|
||||||
}, {
|
|
||||||
'event': event,
|
|
||||||
}]
|
|
||||||
}
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
script.turn_on(self.hass, ENTITY_ID)
|
|
||||||
self.hass.pool.block_till_done()
|
|
||||||
|
|
||||||
self.assertTrue(script.is_on(self.hass, ENTITY_ID))
|
|
||||||
|
|
||||||
self.assertEqual(0, len(calls))
|
|
||||||
|
|
||||||
script.turn_off(self.hass, ENTITY_ID)
|
|
||||||
self.hass.pool.block_till_done()
|
|
||||||
self.assertFalse(script.is_on(self.hass, ENTITY_ID))
|
|
||||||
|
|
||||||
future = dt_util.utcnow() + timedelta(seconds=5)
|
|
||||||
fire_time_changed(self.hass, future)
|
|
||||||
self.hass.pool.block_till_done()
|
|
||||||
|
|
||||||
self.assertFalse(script.is_on(self.hass, ENTITY_ID))
|
|
||||||
|
|
||||||
self.assertEqual(0, len(calls))
|
|
||||||
|
|
||||||
def test_turn_on_service(self):
|
def test_turn_on_service(self):
|
||||||
"""Verify that the turn_on service."""
|
"""Verify that the turn_on service."""
|
||||||
event = 'test_event'
|
event = 'test_event'
|
||||||
|
|
|
||||||
175
tests/helpers/test_script.py
Normal file
175
tests/helpers/test_script.py
Normal file
|
|
@ -0,0 +1,175 @@
|
||||||
|
"""The tests for the Script component."""
|
||||||
|
# pylint: disable=too-many-public-methods,protected-access
|
||||||
|
from datetime import timedelta
|
||||||
|
import unittest
|
||||||
|
|
||||||
|
from homeassistant.bootstrap import _setup_component
|
||||||
|
import homeassistant.util.dt as dt_util
|
||||||
|
from homeassistant.helpers import script
|
||||||
|
|
||||||
|
from tests.common import fire_time_changed, get_test_home_assistant
|
||||||
|
|
||||||
|
|
||||||
|
ENTITY_ID = 'script.test'
|
||||||
|
|
||||||
|
|
||||||
|
class TestScriptHelper(unittest.TestCase):
|
||||||
|
"""Test the Script component."""
|
||||||
|
|
||||||
|
def setUp(self): # pylint: disable=invalid-name
|
||||||
|
"""Setup things to be run when tests are started."""
|
||||||
|
self.hass = get_test_home_assistant()
|
||||||
|
|
||||||
|
def tearDown(self): # pylint: disable=invalid-name
|
||||||
|
"""Stop down everything that was started."""
|
||||||
|
self.hass.stop()
|
||||||
|
|
||||||
|
def test_firing_event(self):
|
||||||
|
"""Test the firing of events."""
|
||||||
|
event = 'test_event'
|
||||||
|
calls = []
|
||||||
|
|
||||||
|
def record_event(event):
|
||||||
|
"""Add recorded event to set."""
|
||||||
|
calls.append(event)
|
||||||
|
|
||||||
|
self.hass.bus.listen(event, record_event)
|
||||||
|
|
||||||
|
script_obj = script.Script(self.hass, {
|
||||||
|
'event': event,
|
||||||
|
'event_data': {
|
||||||
|
'hello': 'world'
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
script_obj.run()
|
||||||
|
|
||||||
|
self.hass.pool.block_till_done()
|
||||||
|
|
||||||
|
assert len(calls) == 1
|
||||||
|
assert calls[0].data.get('hello') == 'world'
|
||||||
|
assert not script_obj.can_cancel
|
||||||
|
|
||||||
|
def test_calling_service(self):
|
||||||
|
"""Test the calling of a service."""
|
||||||
|
calls = []
|
||||||
|
|
||||||
|
def record_call(service):
|
||||||
|
"""Add recorded event to set."""
|
||||||
|
calls.append(service)
|
||||||
|
|
||||||
|
self.hass.services.register('test', 'script', record_call)
|
||||||
|
|
||||||
|
script_obj = script.Script(self.hass, {
|
||||||
|
'service': 'test.script',
|
||||||
|
'data': {
|
||||||
|
'hello': 'world'
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
script_obj.run()
|
||||||
|
self.hass.pool.block_till_done()
|
||||||
|
|
||||||
|
assert len(calls) == 1
|
||||||
|
assert calls[0].data.get('hello') == 'world'
|
||||||
|
|
||||||
|
def test_calling_service_template(self):
|
||||||
|
"""Test the calling of a service."""
|
||||||
|
calls = []
|
||||||
|
|
||||||
|
def record_call(service):
|
||||||
|
"""Add recorded event to set."""
|
||||||
|
calls.append(service)
|
||||||
|
|
||||||
|
self.hass.services.register('test', 'script', record_call)
|
||||||
|
|
||||||
|
script_obj = script.Script(self.hass, {
|
||||||
|
'service_template': """
|
||||||
|
{% if True %}
|
||||||
|
test.script
|
||||||
|
{% else %}
|
||||||
|
test.not_script
|
||||||
|
{% endif %}""",
|
||||||
|
'data_template': {
|
||||||
|
'hello': """
|
||||||
|
{% if True %}
|
||||||
|
world
|
||||||
|
{% else %}
|
||||||
|
Not world
|
||||||
|
{% endif %}
|
||||||
|
"""
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
script_obj.run()
|
||||||
|
|
||||||
|
self.hass.pool.block_till_done()
|
||||||
|
|
||||||
|
assert len(calls) == 1
|
||||||
|
assert calls[0].data.get('hello') == 'world'
|
||||||
|
|
||||||
|
def test_delay(self):
|
||||||
|
"""Test the delay."""
|
||||||
|
event = 'test_event'
|
||||||
|
events = []
|
||||||
|
|
||||||
|
def record_event(event):
|
||||||
|
"""Add recorded event to set."""
|
||||||
|
events.append(event)
|
||||||
|
|
||||||
|
self.hass.bus.listen(event, record_event)
|
||||||
|
|
||||||
|
script_obj = script.Script(self.hass, [
|
||||||
|
{'event': event},
|
||||||
|
{'delay': {'seconds': 5}},
|
||||||
|
{'event': event}])
|
||||||
|
|
||||||
|
script_obj.run()
|
||||||
|
|
||||||
|
self.hass.pool.block_till_done()
|
||||||
|
|
||||||
|
assert script_obj.is_running
|
||||||
|
assert script_obj.can_cancel
|
||||||
|
assert script_obj.last_action == event
|
||||||
|
assert len(events) == 1
|
||||||
|
|
||||||
|
future = dt_util.utcnow() + timedelta(seconds=5)
|
||||||
|
fire_time_changed(self.hass, future)
|
||||||
|
self.hass.pool.block_till_done()
|
||||||
|
|
||||||
|
assert not script_obj.is_running
|
||||||
|
assert len(events) == 2
|
||||||
|
|
||||||
|
def test_cancel_while_delay(self):
|
||||||
|
"""Test the cancelling while the delay is present."""
|
||||||
|
event = 'test_event'
|
||||||
|
events = []
|
||||||
|
|
||||||
|
def record_event(event):
|
||||||
|
"""Add recorded event to set."""
|
||||||
|
events.append(event)
|
||||||
|
|
||||||
|
self.hass.bus.listen(event, record_event)
|
||||||
|
|
||||||
|
script_obj = script.Script(self.hass, [
|
||||||
|
{'delay': {'seconds': 5}},
|
||||||
|
{'event': event}])
|
||||||
|
|
||||||
|
script_obj.run()
|
||||||
|
|
||||||
|
self.hass.pool.block_till_done()
|
||||||
|
|
||||||
|
assert script_obj.is_running
|
||||||
|
assert len(events) == 0
|
||||||
|
|
||||||
|
script_obj.stop()
|
||||||
|
|
||||||
|
assert not script_obj.is_running
|
||||||
|
|
||||||
|
# Make sure the script is really stopped.
|
||||||
|
future = dt_util.utcnow() + timedelta(seconds=5)
|
||||||
|
fire_time_changed(self.hass, future)
|
||||||
|
self.hass.pool.block_till_done()
|
||||||
|
|
||||||
|
assert not script_obj.is_running
|
||||||
|
assert len(events) == 0
|
||||||
Loading…
Add table
Add a link
Reference in a new issue