ansible-collections/community.general
1
0
Fork 0
mirror of https://github.com/ansible-collections/community.general.git synced 2025-08-24 06:51:44 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

use exception classes from utils

Browse source
This commit is contained in:
Simon Kelly 2020-10-15 09:24:56 +02:00
commit f2c220a925
1 changed files with 9 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

26
tests/unit/plugins/modules/monitoring/test_monit.py
View file

@ -6,22 +6,14 @@ __metaclass__ = type
import mock
from ansible_collections.community.general.tests.unit.compat import unittest
from ansible_collections.community.general.plugins.modules.monitoring import monit
class AnsibleExitJson(Exception):
"""Exception class to be raised by module.exit_json and caught by the test case"""
pass
class AnsibleFailJson(Exception):
"""Exception class to be raised by module.fail_json and caught by the test case"""
pass
from ansible_collections.community.general.tests.unit.plugins.modules.utils import AnsibleExitJson, AnsibleFailJson
class MonitTest(unittest.TestCase):
def setUp(self):
self.module = mock.MagicMock()
self.module.exit_json.side_effect = AnsibleExitJson(Exception)
self.module.exit_json.side_effect = AnsibleExitJson
self.module.fail_json.side_effect = AnsibleFailJson
self.monit = monit.Monit(self.module, 'monit', 'processX', 1)
def patch_status(self, side_effect):
@ -29,8 +21,8 @@ class MonitTest(unittest.TestCase):
def test_min_version(self):
with mock.patch.object(self.monit, 'monit_version', return_value=(5, 20)):
self.monit.check_version()
self.module.fail_json.assert_called_once()
with self.assertRaises(AnsibleFailJson):
self.monit.check_version()
def test_change_state_success(self):
with self.patch_status(['not monitored']):
@ -41,8 +33,8 @@ class MonitTest(unittest.TestCase):
def test_change_state_fail(self):
with self.patch_status(['monitored']):
self.monit.stop()
self.module.fail_json.assert_called_once()
with self.assertRaises(AnsibleFailJson):
self.monit.stop()
def test_reload_fail(self):
self.module.run_command.return_value = (1, 'stdout', 'stderr')
@ -64,8 +56,8 @@ class MonitTest(unittest.TestCase):
def test_monitor_fail(self):
with self.patch_status(['not monitored']):
self.monit.monitor()
self.module.fail_json.assert_called_once()
with self.assertRaises(AnsibleFailJson):
self.monit.monitor()
def test_timeout(self):
self.monit.timeout = 0