59 lines
1.9 KiB
Python
59 lines
1.9 KiB
Python
"""
|
|
tests.components.notify.test_command_line
|
|
~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
Tests command line notification.
|
|
"""
|
|
import os
|
|
import tempfile
|
|
import unittest
|
|
|
|
from homeassistant import core
|
|
import homeassistant.components.notify as notify
|
|
from unittest.mock import patch
|
|
|
|
|
|
class TestCommandLine(unittest.TestCase):
|
|
""" Test the command line. """
|
|
|
|
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_command_line_output(self):
|
|
with tempfile.TemporaryDirectory() as tempdirname:
|
|
filename = os.path.join(tempdirname, 'message.txt')
|
|
message = 'one, two, testing, testing'
|
|
self.assertTrue(notify.setup(self.hass, {
|
|
'notify': {
|
|
'name': 'test',
|
|
'platform': 'command_line',
|
|
'command': 'echo $(cat) > {}'.format(filename)
|
|
}
|
|
}))
|
|
|
|
self.hass.services.call('notify', 'test', {'message': message},
|
|
blocking=True)
|
|
|
|
result = open(filename).read()
|
|
# the echo command adds a line break
|
|
self.assertEqual(result, "{}\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 if logged for non zero exit codes. """
|
|
self.assertTrue(notify.setup(self.hass, {
|
|
'notify': {
|
|
'name': 'test',
|
|
'platform': 'command_line',
|
|
'command': 'echo $(cat); exit 1'
|
|
}
|
|
}))
|
|
|
|
self.hass.services.call('notify', 'test', {'message': 'error'},
|
|
blocking=True)
|
|
self.assertEqual(1, mock_error.call_count)
|