5830da63b1
Moved the service decorator to the service helpers module and moved the associated tests.
80 lines
2.4 KiB
Python
80 lines
2.4 KiB
Python
"""
|
|
tests.helpers.test_service
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
Test service helpers.
|
|
"""
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
from homeassistant.helpers import service
|
|
|
|
from tests.common import get_test_home_assistant, mock_service
|
|
|
|
|
|
class TestServiceHelpers(unittest.TestCase):
|
|
"""
|
|
Tests the Home Assistant service helpers.
|
|
"""
|
|
|
|
def setUp(self): # pylint: disable=invalid-name
|
|
""" things to be run when tests are started. """
|
|
self.hass = get_test_home_assistant()
|
|
self.calls = mock_service(self.hass, 'test_domain', 'test_service')
|
|
|
|
service.HASS = self.hass
|
|
|
|
def tearDown(self): # pylint: disable=invalid-name
|
|
""" Stop down stuff we started. """
|
|
self.hass.stop()
|
|
|
|
def test_service(self):
|
|
""" Test service registration decorator. """
|
|
runs = []
|
|
|
|
decor = service.service('test', 'test')
|
|
decor(lambda x, y: runs.append(1))
|
|
|
|
self.hass.services.call('test', 'test')
|
|
self.hass.pool.block_till_done()
|
|
self.assertEqual(1, len(runs))
|
|
|
|
def test_split_entity_string(self):
|
|
service.call_from_config(self.hass, {
|
|
'service': 'test_domain.test_service',
|
|
'entity_id': 'hello.world, sensor.beer'
|
|
})
|
|
self.hass.pool.block_till_done()
|
|
self.assertEqual(['hello.world', 'sensor.beer'],
|
|
self.calls[-1].data.get('entity_id'))
|
|
|
|
def test_not_mutate_input(self):
|
|
orig = {
|
|
'service': 'test_domain.test_service',
|
|
'entity_id': 'hello.world, sensor.beer',
|
|
'data': {
|
|
'hello': 1,
|
|
},
|
|
}
|
|
service.call_from_config(self.hass, orig)
|
|
self.hass.pool.block_till_done()
|
|
self.assertEqual({
|
|
'service': 'test_domain.test_service',
|
|
'entity_id': 'hello.world, sensor.beer',
|
|
'data': {
|
|
'hello': 1,
|
|
},
|
|
}, orig)
|
|
|
|
@patch('homeassistant.helpers.service._LOGGER.error')
|
|
def test_fail_silently_if_no_service(self, mock_log):
|
|
service.call_from_config(self.hass, None)
|
|
self.assertEqual(1, mock_log.call_count)
|
|
|
|
service.call_from_config(self.hass, {})
|
|
self.assertEqual(2, mock_log.call_count)
|
|
|
|
service.call_from_config(self.hass, {
|
|
'service': 'invalid'
|
|
})
|
|
self.assertEqual(3, mock_log.call_count)
|