mirror of
https://github.com/ansible-collections/community.general.git
synced 2024-09-14 20:13:21 +02:00
74 lines
2.5 KiB
Python
74 lines
2.5 KiB
Python
|
from __future__ import (absolute_import, division, print_function)
|
||
|
|
__metaclass__ = type
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
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(
|
||
|
|
KubernetesRawModule, "exit_json", exit_json)
|
||
|
|
monkeypatch.setattr(
|
||
|
|
KubernetesRawModule, "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()
|