4076ccf639
* Alarm Control Panel Manual - use setup_component * Update automation - zone tests * Update climate - demo tests * Update climate - generic thermostat tests * Update cover - command line tests * Update cover - demo tests * Update device tracker tests * Update device tracker - owntracks tests * Update fan - demo tests * Update garage door - demo tests * Update light tests * Update lock - demo tests * Update media player - demo tests * Update notify - command line tests * Update notify - demo tests * Update notify - file tests * Update notify - group tests * Update sensor - mfi tests * Update sensor - moldindicator tests * Update sensor - mqtt room tests * Update switch - command line * Update switch - flux * Update switch tests * Update scene tests * Fix wrong default port for mfi switch
78 lines
2.7 KiB
Python
78 lines
2.7 KiB
Python
"""The tests for the command line notification platform."""
|
|
import os
|
|
import tempfile
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
from homeassistant.bootstrap import setup_component
|
|
import homeassistant.components.notify as notify
|
|
from tests.common import get_test_home_assistant
|
|
|
|
|
|
class TestCommandLine(unittest.TestCase):
|
|
"""Test the command line notifications."""
|
|
|
|
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_setup(self):
|
|
"""Test setup."""
|
|
assert setup_component(self.hass, 'notify', {
|
|
'notify': {
|
|
'name': 'test',
|
|
'platform': 'command_line',
|
|
'command': 'echo $(cat); exit 1',
|
|
}})
|
|
|
|
def test_bad_config(self):
|
|
"""Test set up the platform with bad/missing configuration."""
|
|
self.assertFalse(setup_component(self.hass, notify.DOMAIN, {
|
|
'notify': {
|
|
'name': 'test',
|
|
'platform': 'bad_platform',
|
|
}
|
|
}))
|
|
|
|
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'
|
|
self.assertTrue(setup_component(self.hass, notify.DOMAIN, {
|
|
'notify': {
|
|
'name': 'test',
|
|
'platform': 'command_line',
|
|
'command': 'echo $(cat) > {}'.format(filename)
|
|
}
|
|
}))
|
|
|
|
self.assertTrue(
|
|
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 is logged for non zero exit codes."""
|
|
self.assertTrue(setup_component(self.hass, notify.DOMAIN, {
|
|
'notify': {
|
|
'name': 'test',
|
|
'platform': 'command_line',
|
|
'command': 'echo $(cat); exit 1'
|
|
}
|
|
}))
|
|
|
|
self.assertTrue(
|
|
self.hass.services.call('notify', 'test', {'message': 'error'},
|
|
blocking=True)
|
|
)
|
|
self.assertEqual(1, mock_error.call_count)
|