# 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.tests import TestCase

import tempfile


class TestSafeYaml(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)

    def test_dumps_returns_string(self):
        self.assertTrue(
            isinstance(yaml.dumps(867 - 5309), (str,)))

    def test_dumps_is_loadable(self):
        mydata = {'a': 'hey', 'b': ['bee', 'Bea']}
        self.assertEqual(yaml.loads(yaml.dumps(mydata)), mydata)

    def test_load(self):
        valid_yaml = "foo: bar"
        expected = {'foo': 'bar'}
        with tempfile.NamedTemporaryFile(mode='w', delete=False) as tmpf:
            tmpf.write(valid_yaml)
            tmpf.close()
            self.assertEqual(yaml.load(tmpf.name), expected)