1
0
Fork 0
mirror of https://github.com/ansible-collections/community.general.git synced 2026-02-04 07:51:50 +00:00
community.general/tests/unit/plugins/modules/test_icinga2_feature.py
2025-11-01 13:46:53 +01:00

105 lines
4 KiB
Python

# Copyright (c) 2018, Ansible Project
# Copyright (c) 2018, Abhijeet Kasurde <akasurde@redhat.com>
#
# GNU General Public License v3.0+ (see LICENSES/GPL-3.0-or-later.txt or https://www.gnu.org/licenses/gpl-3.0.txt)
# SPDX-License-Identifier: GPL-3.0-or-later
from __future__ import annotations
from unittest.mock import patch
from ansible_collections.community.general.plugins.modules import icinga2_feature
from ansible_collections.community.internal_test_tools.tests.unit.plugins.modules.utils import (
AnsibleExitJson,
AnsibleFailJson,
ModuleTestCase,
set_module_args,
)
from ansible.module_utils import basic
def get_bin_path(*args, **kwargs):
"""Function to return path of icinga2 binary."""
return "/bin/icinga2"
class TestIcinga2Feature(ModuleTestCase):
"""Main class for testing icinga2_feature module."""
def setUp(self):
"""Setup."""
super().setUp()
self.module = icinga2_feature
self.mock_get_bin_path = patch.object(basic.AnsibleModule, "get_bin_path", get_bin_path)
self.mock_get_bin_path.start()
self.addCleanup(self.mock_get_bin_path.stop) # ensure that the patching is 'undone'
def tearDown(self):
"""Teardown."""
super().tearDown()
def test_without_required_parameters(self):
"""Failure must occurs when all parameters are missing."""
with self.assertRaises(AnsibleFailJson):
with set_module_args({}):
self.module.main()
def test_enable_feature(self):
"""Check that result is changed."""
with set_module_args(
{
"name": "api",
}
):
with patch.object(basic.AnsibleModule, "run_command") as run_command:
run_command.return_value = 0, "", "" # successful execution, no output
with self.assertRaises(AnsibleExitJson) as result:
icinga2_feature.main()
self.assertTrue(result.exception.args[0]["changed"])
self.assertEqual(run_command.call_count, 2)
self.assertEqual(run_command.call_args[0][0][-1], "api")
def test_enable_feature_with_check_mode(self):
"""Check that result is changed in check mode."""
with set_module_args(
{
"name": "api",
"_ansible_check_mode": True,
}
):
with patch.object(basic.AnsibleModule, "run_command") as run_command:
run_command.return_value = 0, "", "" # successful execution, no output
with self.assertRaises(AnsibleExitJson) as result:
icinga2_feature.main()
self.assertTrue(result.exception.args[0]["changed"])
self.assertEqual(run_command.call_count, 1)
def test_disable_feature(self):
"""Check that result is changed."""
with set_module_args({"name": "api", "state": "absent"}):
with patch.object(basic.AnsibleModule, "run_command") as run_command:
run_command.return_value = 0, "", "" # successful execution, no output
with self.assertRaises(AnsibleExitJson) as result:
icinga2_feature.main()
self.assertTrue(result.exception.args[0]["changed"])
self.assertEqual(run_command.call_count, 2)
self.assertEqual(run_command.call_args[0][0][-1], "api")
def test_disable_feature_with_check_mode(self):
"""Check that result is changed in check mode."""
with set_module_args(
{
"name": "api",
"state": "absent",
"_ansible_check_mode": True,
}
):
with patch.object(basic.AnsibleModule, "run_command") as run_command:
run_command.return_value = 0, "", "" # successful execution, no output
with self.assertRaises(AnsibleExitJson) as result:
icinga2_feature.main()
self.assertTrue(result.exception.args[0]["changed"])
self.assertEqual(run_command.call_count, 1)