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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
|
"""Tests for `openid.kvform` module."""
from __future__ import unicode_literals
import unittest
import six
from testfixtures import LogCapture
from openid import kvform
class KVDictTest(unittest.TestCase):
def runTest(self):
for kv_data, result, expected_warnings in kvdict_cases:
# Convert KVForm to dict
with LogCapture() as logbook:
d = kvform.kvToDict(kv_data)
# make sure it parses to expected dict
self.assertEqual(d, result)
# Check to make sure we got the expected number of warnings
self.assertEqual(len(logbook.records), expected_warnings)
# Convert back to KVForm and round-trip back to dict to make
# sure that *** dict -> kv -> dict is identity. ***
kv = kvform.dictToKV(d)
d2 = kvform.kvToDict(kv)
self.assertEqual(d, d2)
class KVSeqTest(unittest.TestCase):
def cleanSeq(self, seq):
"""Create a new sequence by stripping whitespace from start
and end of each value of each pair"""
clean = []
for k, v in seq:
clean.append((k.strip(), v.strip()))
return clean
def runTest(self):
for kv_data, result, expected_warnings in kvseq_cases:
# seq serializes to expected kvform
with LogCapture() as logbook:
actual = kvform.seqToKV(kv_data)
self.assertEqual(actual, result)
self.assertIsInstance(actual, six.text_type)
# Parse back to sequence. Expected to be unchanged, except
# stripping whitespace from start and end of values
# (i. e. ordering, case, and internal whitespace is preserved)
seq = kvform.kvToSeq(actual)
clean_seq = self.cleanSeq(seq)
self.assertEqual(seq, clean_seq)
self.assertEqual(len(logbook.records), expected_warnings,
"Invalid warnings for {}: {}".format(kv_data, [r.getMessage() for r in logbook.records]))
kvdict_cases = [
# (kvform, parsed dictionary, expected warnings)
('', {}, 0),
('college:harvey mudd\n', {'college': 'harvey mudd'}, 0),
('city:claremont\nstate:CA\n', {'city': 'claremont', 'state': 'CA'}, 0),
('is_valid:true\ninvalidate_handle:{HMAC-SHA1:2398410938412093}\n',
{'is_valid': 'true', 'invalidate_handle': '{HMAC-SHA1:2398410938412093}'}, 0),
# Warnings from lines with no colon:
('x\n', {}, 1),
('x\nx\n', {}, 2),
('East is least\n', {}, 1),
# But not from blank lines (because LJ generates them)
('x\n\n', {}, 1),
# Warning from empty key
(':\n', {'': ''}, 1),
(':missing key\n', {'': 'missing key'}, 1),
# Warnings from leading or trailing whitespace in key or value
(' street:foothill blvd\n', {'street': 'foothill blvd'}, 1),
('major: computer science\n', {'major': 'computer science'}, 1),
(' dorm : east \n', {'dorm': 'east'}, 2),
# Warnings from missing trailing newline
('e^(i*pi)+1:0', {'e^(i*pi)+1': '0'}, 1),
('east:west\nnorth:south', {'east': 'west', 'north': 'south'}, 1),
]
kvseq_cases = [
([], '', 0),
# Make sure that we handle unicode characters
([('\u03bbx', 'x')], '\u03bbx:x\n', 0),
# If it's a UTF-8 str, make sure that it's equivalent to the same
# string, decoded.
([(b'\xce\xbbx', b'x')], '\u03bbx:x\n', 0),
([('openid', 'useful'), ('a', 'b')], 'openid:useful\na:b\n', 0),
# Warnings about leading whitespace
([(' openid', 'useful'), ('a', 'b')], ' openid:useful\na:b\n', 1),
# Warnings about leading and trailing whitespace
([(' openid ', ' useful '),
(' a ', ' b ')], ' openid : useful \n a : b \n', 4),
# warnings about leading and trailing whitespace, but not about
# internal whitespace.
([(' open id ', ' use ful '),
(' a ', ' b ')], ' open id : use ful \n a : b \n', 4),
([('foo', 'bar')], 'foo:bar\n', 0),
]
kvexc_cases = [
[('openid', 'use\nful')],
[('open\nid', 'useful')],
[('open\nid', 'use\nful')],
[('open:id', 'useful')],
[('foo', 'bar'), ('ba\n d', 'seed')],
[('foo', 'bar'), ('bad:', 'seed')],
]
class KVExcTest(unittest.TestCase):
def runTest(self):
for kv_data in kvexc_cases:
self.assertRaises(ValueError, kvform.seqToKV, kv_data)
class GeneralTest(unittest.TestCase):
kvform = '<None>'
def test_convert(self):
with LogCapture() as logbook:
result = kvform.seqToKV([(1, 1)])
self.assertEqual(result, '1:1\n')
self.assertEqual(len(logbook.records), 2)
|
