"""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 """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.""" 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)