TT#81700 remove warning

> 15:09:50 /code/./tests/test_yaml_format.py:66: YAMLLoadWarning: calling
> yaml.load() without Loader=... is deprecated, as the default Loader is
> unsafe. Please read https://msg.pyyaml.org/load for full details.

Change-Id: I56c6b11bee0ed2c42b61d4a20ac97327e0fb725d
mr10.0
Victor Seva 5 years ago
parent 39930c48ef
commit aa109d5fb9

@ -24,6 +24,7 @@ import os
import sys import sys
import unittest import unittest
import fnmatch import fnmatch
try: try:
from yaml import CLoader as Loader from yaml import CLoader as Loader
except ImportError: except ImportError:
@ -38,14 +39,16 @@ class ParametrizedTestCase(unittest.TestCase):
2011/08/02/python-unit-testing-parametrized-test-cases 2011/08/02/python-unit-testing-parametrized-test-cases
""" """
def __init__(self, methodName='runTest', param=None): def __init__(self, methodName="runTest", param=None):
super(ParametrizedTestCase, self).__init__(methodName) super(ParametrizedTestCase, self).__init__(methodName)
self.param = param self.param = param
self.scenario = os.path.dirname(self.param) self.scenario = os.path.dirname(self.param)
def id(self): def id(self):
return "%s_%s" % (super(ParametrizedTestCase, self).id(), return "%s_%s" % (
self.scenario) super(ParametrizedTestCase, self).id(),
self.scenario,
)
@staticmethod @staticmethod
def parametrize(testcase_klass, param=None): def parametrize(testcase_klass, param=None):
@ -61,30 +64,29 @@ class ParametrizedTestCase(unittest.TestCase):
class TestYmlLint(ParametrizedTestCase): class TestYmlLint(ParametrizedTestCase):
def setUp(self): def setUp(self):
self.yaml = load(open(self.param, 'r')) self.yaml = load(open(self.param, "r"), Loader=Loader)
def testFlow(self): def testFlow(self):
self.assertTrue('flow' in self.yaml) self.assertTrue("flow" in self.yaml)
self.assertIsInstance(self.yaml['flow'], list) self.assertIsInstance(self.yaml["flow"], list)
def testSipIn(self): def testSipIn(self):
self.assertTrue('sip_in' in self.yaml) self.assertTrue("sip_in" in self.yaml)
self.assertIsInstance(self.yaml['sip_in'], list) self.assertIsInstance(self.yaml["sip_in"], list)
def testSipOut(self): def testSipOut(self):
self.assertTrue('sip_out' in self.yaml) self.assertTrue("sip_out" in self.yaml)
self.assertIsInstance(self.yaml['sip_out'], list) self.assertIsInstance(self.yaml["sip_out"], list)
if __name__ == '__main__': if __name__ == "__main__":
assert len(sys.argv) == 2 assert len(sys.argv) == 2
assert os.path.exists(sys.argv[1]) assert os.path.exists(sys.argv[1])
suite = unittest.TestSuite() suite = unittest.TestSuite()
suite.addTest( suite.addTest(
ParametrizedTestCase.parametrize(TestYmlLint, ParametrizedTestCase.parametrize(TestYmlLint, param=sys.argv[1])
param=sys.argv[1])) )
result = junitxml.JUnitXmlResult(sys.stdout) result = junitxml.JUnitXmlResult(sys.stdout)
result.startTestRun() result.startTestRun()
suite.run(result) suite.run(result)

Loading…
Cancel
Save