aboutsummaryrefslogtreecommitdiffstats
path: root/test/security/check_versions/tests/test_list_all_containers.py
blob: 4178077c3620584bed490471285688c52174b04e (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
#!/usr/bin/env python3

import k8s_bin_versions_inspector as kbvi
import kubernetes


def exec_list_all_containers(pod_name_trimmer, field_selector):
    kubernetes.config.load_kube_config()
    api = kubernetes.client.CoreV1Api()
    containers = kbvi.list_all_containers(api, field_selector)
    extracted = ((c.namespace, c.pod, c.container) for c in containers)
    trimmed = ((n, pod_name_trimmer(p), c) for n, p, c in extracted)
    result = sorted(trimmed)
    return result


def test_list_all_containers(pod_name_trimmer):
    result = exec_list_all_containers(pod_name_trimmer, "")
    assert result == [
        ("default", "kbvi-test-java-keycloak", "keycloak"),
        ("default", "kbvi-test-java-keycloak-old", "keycloak-old"),
        ("default", "kbvi-test-java-keycloak-very-old", "keycloak-very-old"),
        ("default", "kbvi-test-python-jupyter", "jupyter"),
        ("default", "kbvi-test-python-jupyter-old", "jupyter-old"),
        ("default", "kbvi-test-python-stderr-filebeat", "filebeat"),
        ("default", "kbvi-test-terminated", "python"),
        ("ingress-nginx", "kbvi-test-ingress-nginx", "echo-server"),
        ("kube-system", "kbvi-test-kube-system", "echo-server"),
    ]


def test_list_all_containers_not_default(pod_name_trimmer):
    field_selector = "metadata.namespace!=default"
    result = exec_list_all_containers(pod_name_trimmer, field_selector)
    assert result == [
        ("ingress-nginx", "kbvi-test-ingress-nginx", "echo-server"),
        ("kube-system", "kbvi-test-kube-system", "echo-server"),
    ]


def test_list_all_containers_conjunction(pod_name_trimmer):
    field_selector = "metadata.namespace!=kube-system,metadata.namespace!=ingress-nginx"
    result = exec_list_all_containers(pod_name_trimmer, field_selector)
    assert result == [
        ("default", "kbvi-test-java-keycloak", "keycloak"),
        ("default", "kbvi-test-java-keycloak-old", "keycloak-old"),
        ("default", "kbvi-test-java-keycloak-very-old", "keycloak-very-old"),
        ("default", "kbvi-test-python-jupyter", "jupyter"),
        ("default", "kbvi-test-python-jupyter-old", "jupyter-old"),
        ("default", "kbvi-test-python-stderr-filebeat", "filebeat"),
        ("default", "kbvi-test-terminated", "python"),
    ]