mirror of
https://github.com/ansible-collections/kubernetes.core.git
synced 2026-03-27 05:43:02 +00:00
83b3a1aa39
[backport/2.2] Use yaml.safe_load in unit tests (#265)
Use yaml.safe_load in unit tests
SUMMARY
The function signature in pyyaml 6 for yaml.load changed. Using
safe_load fixes this.
ISSUE TYPE
Bugfix Pull Request
COMPONENT NAME
ADDITIONAL INFORMATION
Reviewed-by: Jill R
Reviewed-by: None
Reviewed-by: Gonéri Le Bouder goneri@lebouder.net
(cherry picked from commit 281ff56)
Reviewed-by: None <None>
Reviewed-by: None <None>
101 lines
2.8 KiB
Python
101 lines
2.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright: (c) 2020, Ansible Project
|
|
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
|
|
|
|
from __future__ import absolute_import, division, print_function
|
|
|
|
__metaclass__ = type
|
|
|
|
import os.path
|
|
|
|
import yaml
|
|
|
|
|
|
from ansible_collections.kubernetes.core.plugins.module_utils.helm import (
|
|
run_helm,
|
|
write_temp_kubeconfig,
|
|
)
|
|
|
|
|
|
class MockedModule:
|
|
def __init__(self):
|
|
|
|
self.params = {
|
|
"api_key": None,
|
|
"ca_cert": None,
|
|
"host": None,
|
|
"kube_context": None,
|
|
"kubeconfig": None,
|
|
"release_namespace": None,
|
|
"validate_certs": None,
|
|
}
|
|
|
|
self.r = {}
|
|
|
|
def run_command(self, command, environ_update=None):
|
|
self.r = {"command": command, "environ_update": environ_update}
|
|
return 0, "", ""
|
|
|
|
|
|
def test_write_temp_kubeconfig_server_only():
|
|
file_name = write_temp_kubeconfig("ff")
|
|
try:
|
|
with open(file_name, "r") as fd:
|
|
content = yaml.safe_load(fd)
|
|
finally:
|
|
os.remove(file_name)
|
|
|
|
assert content == {
|
|
"apiVersion": "v1",
|
|
"clusters": [{"cluster": {"server": "ff"}, "name": "generated-cluster"}],
|
|
"contexts": [
|
|
{"context": {"cluster": "generated-cluster"}, "name": "generated-context"}
|
|
],
|
|
"current-context": "generated-context",
|
|
"kind": "Config",
|
|
}
|
|
|
|
|
|
def test_write_temp_kubeconfig_server_inscure_certs():
|
|
file_name = write_temp_kubeconfig("ff", False, "my-certificate")
|
|
try:
|
|
with open(file_name, "r") as fd:
|
|
content = yaml.safe_load(fd)
|
|
finally:
|
|
os.remove(file_name)
|
|
|
|
assert content["clusters"][0]["cluster"]["insecure-skip-tls-verify"] is True
|
|
assert (
|
|
content["clusters"][0]["cluster"]["certificate-authority"] == "my-certificate"
|
|
)
|
|
|
|
|
|
def test_run_helm_naked():
|
|
module = MockedModule()
|
|
run_helm(module, "helm foo")
|
|
|
|
assert module.r["command"] == "helm foo"
|
|
assert module.r["environ_update"] == {}
|
|
|
|
|
|
def test_run_helm_with_params():
|
|
module = MockedModule()
|
|
module.params = {
|
|
"api_key": "my-api-key",
|
|
"ca_cert": "my-ca-cert",
|
|
"host": "some-host",
|
|
"context": "my-context",
|
|
"release_namespace": "a-release-namespace",
|
|
"validate_certs": False,
|
|
}
|
|
|
|
run_helm(module, "helm foo")
|
|
|
|
assert module.r["command"] == "helm foo"
|
|
assert module.r["environ_update"]["HELM_KUBEAPISERVER"] == "some-host"
|
|
assert module.r["environ_update"]["HELM_KUBECONTEXT"] == "my-context"
|
|
assert module.r["environ_update"]["HELM_KUBETOKEN"] == "my-api-key"
|
|
assert module.r["environ_update"]["HELM_NAMESPACE"] == "a-release-namespace"
|
|
assert module.r["environ_update"]["KUBECONFIG"]
|
|
assert not os.path.exists(module.r["environ_update"]["KUBECONFIG"])
|