summaryrefslogtreecommitdiffstats
path: root/ansible_collections/kubernetes/core/tests/unit/module_utils/test_core.py
blob: a49ea12cc729fc1b02a5534882c42bed1657d8a1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
from __future__ import absolute_import, division, print_function

__metaclass__ = type

import json

import kubernetes
import pytest
from ansible_collections.kubernetes.core.plugins.module_utils.k8s.core import (
    AnsibleK8SModule,
)

MINIMAL_K8S_VERSION = "24.2.0"
UNSUPPORTED_K8S_VERSION = "11.0.0"


@pytest.mark.parametrize("stdin", [{}], indirect=["stdin"])
def test_no_warn(monkeypatch, stdin, capfd):
    monkeypatch.setattr(kubernetes, "__version__", MINIMAL_K8S_VERSION)

    module = AnsibleK8SModule(argument_spec={})
    with pytest.raises(SystemExit):
        module.exit_json()
    out, err = capfd.readouterr()

    return_value = json.loads(out)

    assert return_value.get("exception") is None
    assert return_value.get("warnings") is None
    assert return_value.get("failed") is None


@pytest.mark.parametrize("stdin", [{}], indirect=["stdin"])
def test_warn_on_k8s_version(monkeypatch, stdin, capfd):
    monkeypatch.setattr(kubernetes, "__version__", UNSUPPORTED_K8S_VERSION)

    module = AnsibleK8SModule(argument_spec={})
    with pytest.raises(SystemExit):
        module.exit_json()
    out, err = capfd.readouterr()

    return_value = json.loads(out)

    assert return_value.get("warnings") is not None
    warnings = return_value["warnings"]
    assert len(warnings) == 1
    assert "kubernetes" in warnings[0]
    assert MINIMAL_K8S_VERSION in warnings[0]


dependencies = [
    ["28.20.0", "24.2.1", False],
    ["28.20.0", "28.20.0", True],
    ["24.2.1", "28.20.0", True],
]


@pytest.mark.parametrize(
    "stdin,desired,actual,result", [({}, *d) for d in dependencies], indirect=["stdin"]
)
def test_has_at_least(monkeypatch, stdin, desired, actual, result, capfd):
    monkeypatch.setattr(kubernetes, "__version__", actual)

    module = AnsibleK8SModule(argument_spec={})

    assert module.has_at_least("kubernetes", desired) is result


dependencies = [
    ["kubernetes", "28.20.0", "(kubernetes>=28.20.0)"],
    ["foobar", "1.0.0", "(foobar>=1.0.0)"],
    ["foobar", None, "(foobar)"],
]


@pytest.mark.parametrize(
    "stdin,dependency,version,msg", [({}, *d) for d in dependencies], indirect=["stdin"]
)
def test_requires_fails_with_message(
    monkeypatch, stdin, dependency, version, msg, capfd
):
    monkeypatch.setattr(kubernetes, "__version__", "24.2.0")
    module = AnsibleK8SModule(argument_spec={})
    with pytest.raises(SystemExit):
        module.requires(dependency, version)
    out, err = capfd.readouterr()
    return_value = json.loads(out)

    assert return_value.get("failed")
    assert msg in return_value.get("msg")