diff options
Diffstat (limited to 'test/units/utils/kubevirt_fixtures.py')
-rw-r--r-- | test/units/utils/kubevirt_fixtures.py | 70 |
1 files changed, 0 insertions, 70 deletions
diff --git a/test/units/utils/kubevirt_fixtures.py b/test/units/utils/kubevirt_fixtures.py deleted file mode 100644 index fd2fdb95fe..0000000000 --- a/test/units/utils/kubevirt_fixtures.py +++ /dev/null @@ -1,70 +0,0 @@ -import pytest - -from units.compat.mock import MagicMock - -from ansible.module_utils.k8s.common import K8sAnsibleMixin -from ansible.module_utils.k8s.raw import KubernetesRawModule -from ansible.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() |