diff --git a/tests/unit/modules/cloud/openstack/test_baremetal_port_group.py b/tests/unit/modules/cloud/openstack/test_baremetal_port_group.py index 4f689552..93c5eb98 100644 --- a/tests/unit/modules/cloud/openstack/test_baremetal_port_group.py +++ b/tests/unit/modules/cloud/openstack/test_baremetal_port_group.py @@ -1,12 +1,13 @@ import importlib.util -import json -import unittest from pathlib import Path from unittest import mock -from unittest.mock import patch -from ansible.module_utils import basic -from ansible.module_utils._text import to_bytes +from ansible_collections.openstack.cloud.tests.unit.modules.utils import ( + AnsibleExitJson, + AnsibleFailJson, + ModuleTestCase, + set_module_args, +) def _load_module_under_test(): @@ -22,53 +23,6 @@ def _load_module_under_test(): baremetal_port_group = _load_module_under_test() -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): - mock_module = None - mock_sleep = None - - 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) - - class FakePortGroup(dict[str, object]): def to_dict(self, computed=False): diff --git a/tests/unit/modules/utils.py b/tests/unit/modules/utils.py index 057dcd1a..4f269207 100644 --- a/tests/unit/modules/utils.py +++ b/tests/unit/modules/utils.py @@ -14,6 +14,7 @@ def set_module_args(args): args = json.dumps({'ANSIBLE_MODULE_ARGS': args}) basic._ANSIBLE_ARGS = to_bytes(args) + basic._ANSIBLE_PROFILE = 'legacy' class AnsibleExitJson(Exception):