Fix yaml loading in tests

canonical · Nov 1, 2021 · e372970 · e372970
1 parent 6699a5d
commit e372970
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/test-addons.py b/tests/test-addons.py
@@ -58,7 +58,7 @@ def test_invalid_addon(self):
             sh.microk8s.enable.foo()
 
     def test_help_text(self):
-        status = yaml.load(sh.microk8s.status(format="yaml").stdout)
+        status = yaml.load_safe(sh.microk8s.status(format="yaml").stdout)
         expected = {a["name"]: "disabled" for a in status["addons"]}
         expected["ha-cluster"] = "enabled"
 

diff --git a/tests/utils.py b/tests/utils.py
@@ -79,7 +79,7 @@ def kubectl_get(target, timeout_insec=300):
     """
     cmd = "get -o yaml " + target
     output = kubectl(cmd, timeout_insec)
-    return yaml.load(output)
+    return yaml.load_safe(output)
 
 
 def wait_for_pod_state(