72 lines
2.4 KiB
Python
72 lines
2.4 KiB
Python
"""
|
|
tests.test_shell_command
|
|
~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
Tests demo component.
|
|
"""
|
|
import os
|
|
import tempfile
|
|
import unittest
|
|
from unittest.mock import patch
|
|
from subprocess import SubprocessError
|
|
|
|
from homeassistant import core
|
|
from homeassistant.components import shell_command
|
|
|
|
|
|
class TestShellCommand(unittest.TestCase):
|
|
""" Test the demo module. """
|
|
|
|
def setUp(self): # pylint: disable=invalid-name
|
|
self.hass = core.HomeAssistant()
|
|
|
|
def tearDown(self): # pylint: disable=invalid-name
|
|
""" Stop down stuff we started. """
|
|
self.hass.stop()
|
|
|
|
def test_executing_service(self):
|
|
""" Test if able to call a configured service. """
|
|
with tempfile.TemporaryDirectory() as tempdirname:
|
|
path = os.path.join(tempdirname, 'called.txt')
|
|
self.assertTrue(shell_command.setup(self.hass, {
|
|
'shell_command': {
|
|
'test_service': "date > {}".format(path)
|
|
}
|
|
}))
|
|
|
|
self.hass.services.call('shell_command', 'test_service',
|
|
blocking=True)
|
|
|
|
self.assertTrue(os.path.isfile(path))
|
|
|
|
def test_config_not_dict(self):
|
|
""" Test if config is not a dict. """
|
|
self.assertFalse(shell_command.setup(self.hass, {
|
|
'shell_command': ['some', 'weird', 'list']
|
|
}))
|
|
|
|
def test_config_not_valid_service_names(self):
|
|
""" Test if config contains invalid service names. """
|
|
self.assertFalse(shell_command.setup(self.hass, {
|
|
'shell_command': {
|
|
'this is invalid because space': 'touch bla.txt'
|
|
}}))
|
|
|
|
@patch('homeassistant.components.shell_command.subprocess.call',
|
|
side_effect=SubprocessError)
|
|
@patch('homeassistant.components.shell_command._LOGGER.error')
|
|
def test_subprocess_raising_error(self, mock_call, mock_error):
|
|
with tempfile.TemporaryDirectory() as tempdirname:
|
|
path = os.path.join(tempdirname, 'called.txt')
|
|
self.assertTrue(shell_command.setup(self.hass, {
|
|
'shell_command': {
|
|
'test_service': "touch {}".format(path)
|
|
}
|
|
}))
|
|
|
|
self.hass.services.call('shell_command', 'test_service',
|
|
blocking=True)
|
|
|
|
self.assertFalse(os.path.isfile(path))
|
|
self.assertEqual(1, mock_error.call_count)
|