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
|
# Copyright 2015 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Tests for json_lib."""
from __future__ import print_function
from chromite.lib import cros_test_lib
from chromite.lib import json_lib
from chromite.lib import osutils
class JsonHelpersTest(cros_test_lib.MockTestCase):
"""Tests for chromite.lib.json_lib."""
def testAssertIsInstance(self):
"""Test that AssertIsInstance is correct."""
self.assertRaises(ValueError, json_lib.AssertIsInstance,
tuple(), list, 'a bad value')
self.assertRaises(ValueError, json_lib.AssertIsInstance,
1, float, 'a bad value')
self.assertRaises(ValueError, json_lib.AssertIsInstance,
1, bool, 'a bad value')
json_lib.AssertIsInstance([1], list, 'good value')
json_lib.AssertIsInstance(True, bool, 'good value')
json_lib.AssertIsInstance({'foo': 2}, dict, 'good value')
def testGetValueOfType(self):
"""Test that GetValueOfType is correct."""
self.assertRaises(
ValueError, json_lib.GetValueOfType,
{}, 'missing key', str, 'missing value')
self.assertRaises(
ValueError, json_lib.GetValueOfType,
{'key': 1}, 'key', bool, 'bad type')
self.assertRaises(
ValueError, json_lib.GetValueOfType,
{'key': [1]}, 'key', int, 'bad type')
self.assertEqual(
json_lib.GetValueOfType({'key': 1}, 'key', int, 'good value'),
1)
def testPopValueOfType(self):
"""Test that PopValueOfType is correct."""
input_dict = {'key': 'value'}
self.assertEqual(
'value',
json_lib.GetValueOfType(input_dict, 'key', str, 'value'))
self.assertEqual(
'value',
json_lib.PopValueOfType(input_dict, 'key', str, 'value'))
self.assertFalse(input_dict)
def testParseJsonFileWithComments(self):
"""Test that we can parse a JSON file with comments."""
JSON_WITH_COMMENTS = """
{
# I am a comment.
"foo": []
}
"""
self.PatchObject(osutils, 'ReadFile', return_value=JSON_WITH_COMMENTS)
self.assertEqual({u'foo': []},
json_lib.ParseJsonFileWithComments('fake path'))
self.PatchObject(osutils, 'ReadFile', return_value='')
self.assertRaises(ValueError,
json_lib.ParseJsonFileWithComments,
'fake path')
|
