792564e401
This will likely also be very useful to use since it ensures we safely use/parse yaml. Change-Id: Ibc0f7dee46d2f199e2d65d9bc6444bfa72383704
30 lines
1.0 KiB
Python
30 lines
1.0 KiB
Python
# Copyright 2015 Canonical Ltd.
|
|
# This file is part of cloud-init. See LICENCE file for license information.
|
|
#
|
|
# vi: ts=4 expandtab
|
|
|
|
from cloudinit import safeyaml as yaml
|
|
from cloudinit import test
|
|
|
|
|
|
class TestSafeYaml(test.TestCase):
|
|
def test_simple(self):
|
|
blob = '\nk1: one\nk2: two'
|
|
expected = {'k1': "one", 'k2': "two"}
|
|
self.assertEqual(yaml.loads(blob), expected)
|
|
|
|
def test_bogus_raises_exception(self):
|
|
badyaml = "1\n 2:"
|
|
self.assertRaises(yaml.YAMLError, yaml.loads, badyaml)
|
|
|
|
def test_unsafe_types(self):
|
|
# should not load complex types
|
|
unsafe_yaml = "!!python/object:__builtin__.object {}"
|
|
self.assertRaises(yaml.YAMLError, yaml.loads, unsafe_yaml)
|
|
|
|
def test_python_unicode_not_allowed(self):
|
|
# python/unicode is not allowed
|
|
# in the past this type was allowed, but not now, so explicit test.
|
|
blob = "{k1: !!python/unicode 'my unicode', k2: my string}"
|
|
self.assertRaises(yaml.YAMLError, yaml.loads, blob)
|