summaryrefslogtreecommitdiff
path: root/lib/spack/spack/test/yaml.py
blob: f1b83e7b7170c65f4876bcb303e823d101ecb99e (plain) (blame)
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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
##############################################################################
# Copyright (c) 2013-2016, Lawrence Livermore National Security, LLC.
# Produced at the Lawrence Livermore National Laboratory.
#
# This file is part of Spack.
# Created by Todd Gamblin, tgamblin@llnl.gov, All rights reserved.
# LLNL-CODE-647188
#
# For details, see https://github.com/llnl/spack
# Please also see the LICENSE file for our notice and the LGPL.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License (as
# published by the Free Software Foundation) version 2.1, February 1999.
#
# 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 terms and
# conditions of the GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
##############################################################################
"""
Test Spack's custom YAML format.
"""
import unittest

import spack.util.spack_yaml as syaml

test_file = """\
config_file:
  x86_64:
    foo: /path/to/foo
    bar: /path/to/bar
    baz: /path/to/baz
  some_list:
    - item 1
    - item 2
    - item 3
  another_list:
    [ 1, 2, 3 ]
  some_key: some_string
"""

test_data = {
    'config_file' : syaml.syaml_dict([
        ('x86_64', syaml.syaml_dict([
            ('foo', '/path/to/foo'),
            ('bar', '/path/to/bar'),
            ('baz', '/path/to/baz' )])),
        ('some_list', [ 'item 1', 'item 2', 'item 3' ]),
        ('another_list', [ 1, 2, 3 ]),
        ('some_key', 'some_string')
    ])}

class YamlTest(unittest.TestCase):

    def setUp(self):
        self.data = syaml.load(test_file)


    def test_parse(self):
        self.assertEqual(test_data, self.data)


    def test_dict_order(self):
        self.assertEqual(
            ['x86_64', 'some_list', 'another_list', 'some_key'],
            self.data['config_file'].keys())

        self.assertEqual(
            ['foo', 'bar', 'baz'],
            self.data['config_file']['x86_64'].keys())


    def test_line_numbers(self):
        def check(obj, start_line, end_line):
            self.assertEqual(obj._start_mark.line, start_line)
            self.assertEqual(obj._end_mark.line, end_line)

        check(self.data,                                  0, 12)
        check(self.data['config_file'],                   1, 12)
        check(self.data['config_file']['x86_64'],         2,  5)
        check(self.data['config_file']['x86_64']['foo'],  2,  2)
        check(self.data['config_file']['x86_64']['bar'],  3,  3)
        check(self.data['config_file']['x86_64']['baz'],  4,  4)
        check(self.data['config_file']['some_list'],      6,  9)
        check(self.data['config_file']['some_list'][0],   6,  6)
        check(self.data['config_file']['some_list'][1],   7,  7)
        check(self.data['config_file']['some_list'][2],   8,  8)
        check(self.data['config_file']['another_list'],  10, 10)
        check(self.data['config_file']['some_key'],      11, 11)