Fix yaml loading in tests (#2698)

canonical · Nov 1, 2021 · 14ada87 · 14ada87
1 parent eb83acd
commit 14ada87
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
@@ -53,7 +53,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.safe_load(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.safe_load(output)
 
 
 def wait_for_pod_state(