implemented unit test cases for nagios module

tests/unit/modules/nagios_test.py

@@ -0,0 +1,93 @@
+# -*- coding: utf-8 -*-
+'''
+    :codeauthor: :email:`Rupesh Tare <rupesht@saltstack.com>`
+'''
+# Import Python Libs
+from __future__ import absolute_import
+
+# Import Salt Testing Libs
+from salttesting import TestCase, skipIf
+from salttesting.mock import (
+    MagicMock,
+    patch,
+    NO_MOCK,
+    NO_MOCK_REASON
+)
+from salttesting.helpers import ensure_in_syspath
+
+ensure_in_syspath('../../')
+
+# Import Salt Libs
+from salt.modules import nagios
+import os
+
+
+# Globals
+nagios.__salt__ = {}
+
+
+@skipIf(NO_MOCK, NO_MOCK_REASON)
+class NagiosTestCase(TestCase):
+    '''
+    Test cases for salt.modules.nagios
+    '''
+    def test_run(self):
+        '''
+        Test for Run nagios plugin and return all
+         the data execution with cmd.run
+        '''
+        with patch.object(nagios, '_execute_cmd', return_value='A'):
+            self.assertEqual(nagios.run('plugin'), 'A')
+
+    def test_retcode(self):
+        '''
+        Test for Run one nagios plugin and return retcode of the execution
+        '''
+        with patch.object(nagios, '_execute_cmd', return_value='A'):
+            self.assertEqual(nagios.retcode('plugin', key_name='key'),
+                             {'key': {'status': 'A'}})
+
+    def test_run_all(self):
+        '''
+        Test for Run nagios plugin and return all
+         the data execution with cmd.run_all
+        '''
+        with patch.object(nagios, '_execute_cmd', return_value='A'):
+            self.assertEqual(nagios.run_all('plugin'), 'A')
+
+    def test_retcode_pillar(self):
+        '''
+        Test for Run one or more nagios plugins from pillar data and
+         get the result of cmd.retcode
+        '''
+        with patch.dict(nagios.__salt__,
+                        {'pillar.get': MagicMock(return_value={})}):
+            self.assertEqual(nagios.retcode_pillar('pillar_name'), {})
+
+    def test_run_pillar(self):
+        '''
+        Test for Run one or more nagios plugins from pillar data
+         and get the result of cmd.run
+        '''
+        with patch.object(nagios, '_execute_pillar', return_value='A'):
+            self.assertEqual(nagios.run_pillar('pillar'), 'A')
+
+    def test_run_all_pillar(self):
+        '''
+        Test for Run one or more nagios plugins from pillar data
+         and get the result of cmd.run
+        '''
+        with patch.object(nagios, '_execute_pillar', return_value='A'):
+            self.assertEqual(nagios.run_all_pillar('pillar'), 'A')
+
+    def test_list_plugins(self):
+        '''
+        Test for List all the nagios plugins
+        '''
+        with patch.object(os, 'listdir', return_value=[]):
+            self.assertEqual(nagios.list_plugins(), [])
+
+
+if __name__ == '__main__':
+    from integration import run_tests
+    run_tests(NagiosTestCase, needs_daemon=False)