88 lines
3.2 KiB
Python
88 lines
3.2 KiB
Python
"""The tests for the command line notification platform."""
|
|
import os
|
|
import tempfile
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
from homeassistant.setup import setup_component
|
|
import homeassistant.components.notify as notify
|
|
from tests.common import assert_setup_component, get_test_home_assistant
|
|
|
|
|
|
class TestCommandLine(unittest.TestCase):
|
|
"""Test the command line notifications."""
|
|
|
|
def setUp(self): # pylint: disable=invalid-name
|
|
"""Set up 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_setup(self):
|
|
"""Test setup."""
|
|
with assert_setup_component(1) as handle_config:
|
|
assert setup_component(self.hass, 'notify', {
|
|
'notify': {
|
|
'name': 'test',
|
|
'platform': 'command_line',
|
|
'command': 'echo $(cat); exit 1', }
|
|
})
|
|
assert handle_config[notify.DOMAIN]
|
|
|
|
def test_bad_config(self):
|
|
"""Test set up the platform with bad/missing configuration."""
|
|
config = {
|
|
notify.DOMAIN: {
|
|
'name': 'test',
|
|
'platform': 'command_line',
|
|
}
|
|
}
|
|
with assert_setup_component(0) as handle_config:
|
|
assert setup_component(self.hass, notify.DOMAIN, config)
|
|
assert not handle_config[notify.DOMAIN]
|
|
|
|
def test_command_line_output(self):
|
|
"""Test the command line output."""
|
|
with tempfile.TemporaryDirectory() as tempdirname:
|
|
filename = os.path.join(tempdirname, 'message.txt')
|
|
message = 'one, two, testing, testing'
|
|
with assert_setup_component(1) as handle_config:
|
|
self.assertTrue(setup_component(self.hass, notify.DOMAIN, {
|
|
'notify': {
|
|
'name': 'test',
|
|
'platform': 'command_line',
|
|
'command': 'echo $(cat) > {}'.format(filename)
|
|
}
|
|
}))
|
|
assert handle_config[notify.DOMAIN]
|
|
|
|
self.assertTrue(
|
|
self.hass.services.call('notify', 'test', {'message': message},
|
|
blocking=True)
|
|
)
|
|
|
|
with open(filename) as fil:
|
|
# the echo command adds a line break
|
|
self.assertEqual(fil.read(), "{}\n".format(message))
|
|
|
|
@patch('homeassistant.components.notify.command_line._LOGGER.error')
|
|
def test_error_for_none_zero_exit_code(self, mock_error):
|
|
"""Test if an error is logged for non zero exit codes."""
|
|
with assert_setup_component(1) as handle_config:
|
|
self.assertTrue(setup_component(self.hass, notify.DOMAIN, {
|
|
'notify': {
|
|
'name': 'test',
|
|
'platform': 'command_line',
|
|
'command': 'echo $(cat); exit 1'
|
|
}
|
|
}))
|
|
assert handle_config[notify.DOMAIN]
|
|
|
|
self.assertTrue(
|
|
self.hass.services.call('notify', 'test', {'message': 'error'},
|
|
blocking=True)
|
|
)
|
|
self.assertEqual(1, mock_error.call_count)
|