mirror of
https://github.com/ansible-collections/community.general.git
synced 2024-09-14 20:13:21 +02:00
2cb3cec659
Fixes: #1064
Signed-off-by: Abhijeet Kasurde <akasurde@redhat.com>
(cherry picked from commit f7656ac0d3)
Co-authored-by: Abhijeet Kasurde <akasurde@redhat.com>
74 lines
2.6 KiB
Python
74 lines
2.6 KiB
Python
from __future__ import (absolute_import, division, print_function)
|
|
__metaclass__ = type
|
|
|
|
import pytest
|
|
|
|
from ansible.module_utils.basic import AnsibleModule
|
|
from ansible_collections.community.general.tests.unit.compat.mock import MagicMock
|
|
|
|
from ansible_collections.community.kubernetes.plugins.module_utils.common import K8sAnsibleMixin
|
|
from ansible_collections.community.kubernetes.plugins.module_utils.raw import KubernetesRawModule
|
|
from ansible_collections.community.general.plugins.module_utils.kubevirt import KubeVirtRawModule
|
|
|
|
import openshift.dynamic
|
|
|
|
RESOURCE_DEFAULT_ARGS = {'api_version': 'v1alpha3', 'group': 'kubevirt.io',
|
|
'prefix': 'apis', 'namespaced': True}
|
|
|
|
|
|
class AnsibleExitJson(Exception):
|
|
"""Exception class to be raised by module.exit_json and caught
|
|
by the test case"""
|
|
def __init__(self, **kwargs):
|
|
for k in kwargs:
|
|
setattr(self, k, kwargs[k])
|
|
|
|
def __getitem__(self, attr):
|
|
return getattr(self, attr)
|
|
|
|
|
|
class AnsibleFailJson(Exception):
|
|
"""Exception class to be raised by module.fail_json and caught
|
|
by the test case"""
|
|
def __init__(self, **kwargs):
|
|
for k in kwargs:
|
|
setattr(self, k, kwargs[k])
|
|
|
|
def __getitem__(self, attr):
|
|
return getattr(self, attr)
|
|
|
|
|
|
def exit_json(*args, **kwargs):
|
|
kwargs['success'] = True
|
|
if 'changed' not in kwargs:
|
|
kwargs['changed'] = False
|
|
raise AnsibleExitJson(**kwargs)
|
|
|
|
|
|
def fail_json(*args, **kwargs):
|
|
kwargs['success'] = False
|
|
raise AnsibleFailJson(**kwargs)
|
|
|
|
|
|
@pytest.fixture()
|
|
def base_fixture(monkeypatch):
|
|
monkeypatch.setattr(
|
|
AnsibleModule, "exit_json", exit_json)
|
|
monkeypatch.setattr(
|
|
AnsibleModule, "fail_json", fail_json)
|
|
# Create mock methods in Resource directly, otherwise dyn client
|
|
# tries binding those to corresponding methods in DynamicClient
|
|
# (with partial()), which is more problematic to intercept
|
|
openshift.dynamic.Resource.get = MagicMock()
|
|
openshift.dynamic.Resource.create = MagicMock()
|
|
openshift.dynamic.Resource.delete = MagicMock()
|
|
openshift.dynamic.Resource.patch = MagicMock()
|
|
openshift.dynamic.Resource.search = MagicMock()
|
|
openshift.dynamic.Resource.watch = MagicMock()
|
|
# Globally mock some methods, since all tests will use this
|
|
KubernetesRawModule.patch_resource = MagicMock()
|
|
KubernetesRawModule.patch_resource.return_value = ({}, None)
|
|
K8sAnsibleMixin.get_api_client = MagicMock()
|
|
K8sAnsibleMixin.get_api_client.return_value = None
|
|
K8sAnsibleMixin.find_resource = MagicMock()
|
|
KubeVirtRawModule.find_supported_resource = MagicMock()
|