Added some unittests for the command_line notification

tests/components/notify/test_command_line.py

@@ -0,0 +1,58 @@
+"""
+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)