summaryrefslogtreecommitdiffstats
path: root/ansible_collections/cisco/aci/tests/unit/modules
diff options
context:
space:
mode:
Diffstat (limited to 'ansible_collections/cisco/aci/tests/unit/modules')
-rw-r--r--ansible_collections/cisco/aci/tests/unit/modules/__init__.py0
-rw-r--r--ansible_collections/cisco/aci/tests/unit/modules/utils.py50
2 files changed, 50 insertions, 0 deletions
diff --git a/ansible_collections/cisco/aci/tests/unit/modules/__init__.py b/ansible_collections/cisco/aci/tests/unit/modules/__init__.py
new file mode 100644
index 000000000..e69de29bb
--- /dev/null
+++ b/ansible_collections/cisco/aci/tests/unit/modules/__init__.py
diff --git a/ansible_collections/cisco/aci/tests/unit/modules/utils.py b/ansible_collections/cisco/aci/tests/unit/modules/utils.py
new file mode 100644
index 000000000..80a77fdc1
--- /dev/null
+++ b/ansible_collections/cisco/aci/tests/unit/modules/utils.py
@@ -0,0 +1,50 @@
+from __future__ import absolute_import, division, print_function
+
+__metaclass__ = type
+
+import json
+
+from ansible_collections.cisco.aci.tests.unit.compat import unittest
+from ansible_collections.cisco.aci.tests.unit.compat.mock import patch
+from ansible.module_utils import basic
+from ansible.module_utils._text import to_bytes
+
+
+def set_module_args(args):
+ if "_ansible_remote_tmp" not in args:
+ args["_ansible_remote_tmp"] = "/tmp"
+ if "_ansible_keep_remote_files" not in args:
+ args["_ansible_keep_remote_files"] = False
+
+ args = json.dumps({"ANSIBLE_MODULE_ARGS": args})
+ basic._ANSIBLE_ARGS = to_bytes(args)
+
+
+class AnsibleExitJson(Exception):
+ pass
+
+
+class AnsibleFailJson(Exception):
+ pass
+
+
+def exit_json(*args, **kwargs):
+ if "changed" not in kwargs:
+ kwargs["changed"] = False
+ raise AnsibleExitJson(kwargs)
+
+
+def fail_json(*args, **kwargs):
+ kwargs["failed"] = True
+ raise AnsibleFailJson(kwargs)
+
+
+class ModuleTestCase(unittest.TestCase):
+ def setUp(self):
+ self.mock_module = patch.multiple(basic.AnsibleModule, exit_json=exit_json, fail_json=fail_json)
+ self.mock_module.start()
+ self.mock_sleep = patch("time.sleep")
+ self.mock_sleep.start()
+ set_module_args({})
+ self.addCleanup(self.mock_module.stop)
+ self.addCleanup(self.mock_sleep.stop)