mirror of
https://github.com/valitydev/salt.git
synced 2024-11-07 08:58:59 +00:00
94 lines
2.7 KiB
Python
94 lines
2.7 KiB
Python
# -*- 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)
|