Merge "Fix the yaml load warning."

This commit is contained in:
Zuul 2019-12-16 15:00:56 +00:00 committed by Gerrit Code Review
commit 63d877945d
5 changed files with 10 additions and 10 deletions

View File

@ -59,7 +59,7 @@ class JobCache(object):
self.data = {}
else:
with io.open(self.cachefilename, "r", encoding="utf-8") as yfile:
self.data = yaml.load(yfile)
self.data = yaml.safe_load(yfile)
logger.debug("Using cache: '{0}'".format(self.cachefilename))
def _lock(self):

View File

@ -101,7 +101,7 @@ class JenkinsJobs(object):
with io.open(
self.options.plugins_info_path, "r", encoding="utf-8"
) as yaml_file:
plugins_info = yaml.load(yaml_file)
plugins_info = yaml.safe_load(yaml_file)
if not isinstance(plugins_info, list):
self.parser.error(
"{0} must contain a Yaml list!".format(

View File

@ -40,6 +40,6 @@ class TestCaseJobCache(base.BaseTestCase):
"""
test_file = os.path.abspath(__file__)
with mock.patch("os.path.join", return_value=test_file):
with mock.patch("yaml.load"):
with mock.patch("yaml.safe_load"):
with mock.patch("jenkins_jobs.builder.JobCache._lock"):
jenkins_jobs.builder.JobCache("dummy").data = None

View File

@ -213,7 +213,7 @@ class TestTests(CmdTestsBase):
self.execute_jenkins_jobs_with_args(args)
with io.open(plugins_info_stub_yaml_file, "r", encoding="utf-8") as yaml_file:
plugins_info_list = yaml.load(yaml_file)
plugins_info_list = yaml.safe_load(yaml_file)
registry_mock.assert_called_with(mock.ANY, plugins_info_list)

View File

@ -31,7 +31,7 @@ class TestCaseTestHelpers(base.BaseTestCase):
# Test default values
default_root = XML.Element("testdefault")
default_data = yaml.load("string: hello")
default_data = yaml.safe_load("string: hello")
default_mappings = [("default-string", "defaultString", "default")]
convert_mapping_to_xml(
@ -42,7 +42,7 @@ class TestCaseTestHelpers(base.BaseTestCase):
# Test user input
user_input_root = XML.Element("testUserInput")
user_input_data = yaml.load("user-input-string: hello")
user_input_data = yaml.safe_load("user-input-string: hello")
user_input_mappings = [("user-input-string", "userInputString", "user-input")]
convert_mapping_to_xml(
@ -53,7 +53,7 @@ class TestCaseTestHelpers(base.BaseTestCase):
# Test missing required input
required_root = XML.Element("testrequired")
required_data = yaml.load("string: hello")
required_data = yaml.safe_load("string: hello")
required_mappings = [("required-string", "requiredString", None)]
self.assertRaises(
@ -67,7 +67,7 @@ class TestCaseTestHelpers(base.BaseTestCase):
# Test invalid user input for list
user_input_root = XML.Element("testUserInput")
user_input_data = yaml.load("user-input-string: bye")
user_input_data = yaml.safe_load("user-input-string: bye")
valid_inputs = ["hello"]
user_input_mappings = [
("user-input-string", "userInputString", "user-input", valid_inputs)
@ -83,7 +83,7 @@ class TestCaseTestHelpers(base.BaseTestCase):
# Test invalid user input for dict
user_input_root = XML.Element("testUserInput")
user_input_data = yaml.load("user-input-string: later")
user_input_data = yaml.safe_load("user-input-string: later")
valid_inputs = {"hello": "world"}
user_input_mappings = [
("user-input-string", "userInputString", "user-input", valid_inputs)
@ -99,7 +99,7 @@ class TestCaseTestHelpers(base.BaseTestCase):
# Test invalid key for dict
user_input_root = XML.Element("testUserInput")
user_input_data = yaml.load("user-input-string: world")
user_input_data = yaml.safe_load("user-input-string: world")
valid_inputs = {"hello": "world"}
user_input_mappings = [
("user-input-string", "userInputString", "user-input", valid_inputs)