[squash] Add add unittests

Author: martialblog

check_monit.py

@@ -72,7 +72,7 @@ def print_output(status, count_ok, count_all, items):
         for item in items:
             s = "OK" if item.get('status') == 0 else "CRITICAL"
             print(' \\_ [{0}]: {1}'.format(s, item.get('name', 'No Name')))
-            print('  ' + item.get('output', 'None'))
+            print('  {}' .format(item.get('output', 'None')))
 
 def get_service_output(service_type, element):
     # Service Type Filesystem

test_check_monit.py

@@ -64,6 +64,24 @@ def test_return_plugin(self, mock_print):
 
         mock_print.assert_has_calls(calls)
 
+    @mock.patch('builtins.print')
+    def test_return_plugin_with_no_output(self, mock_print):
+        actual = print_output(1, 2, 3, [{'name': 'foo', 'output': None, 'status': 1}])
+
+        calls = [mock.call('[WARNING]: Monit Service Status 2/3'),
+                 mock.call(' \\_ [CRITICAL]: foo'),
+                 mock.call('  None')]
+
+        mock_print.assert_has_calls(calls)
+
+        actual = print_output(1, 2, 3, [{'name': 'foo', 'status': 1}])
+
+        calls = [mock.call('[WARNING]: Monit Service Status 2/3'),
+                 mock.call(' \\_ [CRITICAL]: foo'),
+                 mock.call('  None')]
+
+        mock_print.assert_has_calls(calls)
+
 class MainTesting(unittest.TestCase):
 
     @mock.patch('builtins.print')