summaryrefslogtreecommitdiff
path: root/morphlib/yamlparse_tests.py
blob: 38815168b96bc964b2320df67ae3cefb7dd9a9e5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
# Copyright (C) 2013-2014  Codethink Limited
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; version 2 of the License.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

import unittest

import morphlib
import morphlib.yamlparse as yamlparse
from morphlib.util import OrderedDict

if morphlib.got_yaml:
    yaml = morphlib.yaml


class YAMLParseTests(unittest.TestCase):

    def run(self, *args, **kwargs):
        if morphlib.got_yaml:
            return unittest.TestCase.run(self, *args, **kwargs)

    example_text = '''\
name: foo
kind: chunk
build-system: manual
'''

    example_dict = OrderedDict([
        ('name', 'foo'),
        ('kind', 'chunk'),
        ('build-system', 'manual'),
    ])

    def test_non_map_raises(self):
        incorrect_type = '''\
!!map
- foo
- bar
'''
        self.assertRaises(yaml.YAMLError, yamlparse.load, incorrect_type)

    def test_complex_key_fails_KNOWNFAILURE(self):
        complex_key = '? { foo: bar, baz: qux }: True'
        self.assertRaises(yaml.YAMLError, yamlparse.load, complex_key)

    def test_represents_non_scalar_nodes(self):
        self.assertTrue(
            yamlparse.dump(
                {
                    ('a', 'b'): {
                        "foo": 1,
                        "bar": 2,
                    }
                }, default_flow_style=None))