-
Notifications
You must be signed in to change notification settings - Fork 36
/
Copy pathtest_binary_vdf.py
194 lines (147 loc) · 6.98 KB
/
test_binary_vdf.py
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
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
import sys
import unittest
import vdf
from io import BytesIO
from collections import OrderedDict
u = str if sys.version_info >= (3,) else unicode
class BinaryVDF(unittest.TestCase):
def test_BASE_INT(self):
repr(vdf.BASE_INT())
def test_simple(self):
pairs = [
('a', 'test'),
('a2', b'\xd0\xb0\xd0\xb1\xd0\xb2\xd0\xb3'.decode('utf-8')),
('bb', 1),
('bb2', -500),
('ccc', 1.0),
('dddd', vdf.POINTER(1234)),
('fffff', vdf.COLOR(1234)),
('gggggg', vdf.UINT_64(1234)),
('hhhhhhh', vdf.INT_64(-1234)),
]
data = OrderedDict(pairs)
data['level1-1'] = OrderedDict(pairs)
data['level1-1']['level2-1'] = OrderedDict(pairs)
data['level1-1']['level2-2'] = OrderedDict(pairs)
data['level1-2'] = OrderedDict(pairs)
result = vdf.binary_loads(vdf.binary_dumps(data), mapper=OrderedDict)
self.assertEqual(data, result)
result = vdf.binary_loads(vdf.binary_dumps(data, alt_format=True), mapper=OrderedDict, alt_format=True)
self.assertEqual(data, result)
result = vdf.vbkv_loads(vdf.vbkv_dumps(data), mapper=OrderedDict)
self.assertEqual(data, result)
def test_vbkv_empty(self):
with self.assertRaises(ValueError):
vdf.vbkv_loads(b'')
def test_loads_empty(self):
self.assertEqual(vdf.binary_loads(b''), {})
self.assertEqual(vdf.binary_load(BytesIO(b'')), {})
def test_dumps_empty(self):
self.assertEqual(vdf.binary_dumps({}), b'')
buf = BytesIO()
vdf.binary_dump({}, buf)
self.assertEqual(buf.getvalue(), b'')
def test_dumps_unicode(self):
self.assertEqual(vdf.binary_dumps({u('a'): u('b')}), b'\x01a\x00b\x00\x08')
def test_dumps_unicode_alternative(self):
self.assertEqual(vdf.binary_dumps({u('a'): u('b')}, alt_format=True), b'\x01a\x00b\x00\x0b')
def test_dump_params_invalid(self):
with self.assertRaises(TypeError):
vdf.binary_dump([], BytesIO())
with self.assertRaises(TypeError):
vdf.binary_dump({}, b'aaaa')
def test_dumps_params_invalid(self):
with self.assertRaises(TypeError):
vdf.binary_dumps([])
with self.assertRaises(TypeError):
vdf.binary_dumps(b'aaaa')
def test_dumps_key_invalid_type(self):
with self.assertRaises(TypeError):
vdf.binary_dumps({1:1})
with self.assertRaises(TypeError):
vdf.binary_dumps({None:1})
def test_dumps_value_invalid_type(self):
with self.assertRaises(TypeError):
vdf.binary_dumps({'': None})
def test_alternative_format(self):
with self.assertRaises(SyntaxError):
vdf.binary_loads(b'\x00a\x00\x00b\x00\x0b\x0b')
with self.assertRaises(SyntaxError):
vdf.binary_loads(b'\x00a\x00\x00b\x00\x08\x08', alt_format=True)
def test_load_params_invalid(self):
with self.assertRaises(TypeError):
vdf.binary_load(b'aaaa')
with self.assertRaises(TypeError):
vdf.binary_load(1234)
with self.assertRaises(TypeError):
vdf.binary_load(BytesIO(b'aaaa'), b'bbbb')
def test_loads_params_invalid(self):
with self.assertRaises(TypeError):
vdf.binary_loads([])
with self.assertRaises(TypeError):
vdf.binary_loads(11111)
with self.assertRaises(TypeError):
vdf.binary_loads(BytesIO())
with self.assertRaises(TypeError):
vdf.binary_load(b'', b'bbbb')
def test_loads_unbalanced_nesting(self):
with self.assertRaises(SyntaxError):
vdf.binary_loads(b'\x00a\x00\x00b\x00\x08')
with self.assertRaises(SyntaxError):
vdf.binary_loads(b'\x00a\x00\x00b\x00\x08\x08\x08\x08')
def test_loads_unknown_type(self):
with self.assertRaises(SyntaxError):
vdf.binary_loads(b'\x33a\x00\x08')
def test_loads_unterminated_string(self):
with self.assertRaises(SyntaxError):
vdf.binary_loads(b'\x01abbbb')
def test_loads_type_checks(self):
with self.assertRaises(TypeError):
vdf.binary_loads(None)
with self.assertRaises(TypeError):
vdf.binary_loads(b'', mapper=list)
def test_merge_multiple_keys_on(self):
# VDFDict([('a', VDFDict([('a', '1'), ('b', '2')])), ('a', VDFDict([('a', '3'), ('c', '4')]))])
test = b'\x00a\x00\x01a\x001\x00\x01b\x002\x00\x08\x00a\x00\x01a\x003\x00\x01c\x004\x00\x08\x08'
result = {'a': {'a': '3', 'b': '2', 'c': '4'}}
self.assertEqual(vdf.binary_loads(test, merge_duplicate_keys=True), result)
def test_merge_multiple_keys_off(self):
# VDFDict([('a', VDFDict([('a', '1'), ('b', '2')])), ('a', VDFDict([('a', '3'), ('c', '4')]))])
test = b'\x00a\x00\x01a\x001\x00\x01b\x002\x00\x08\x00a\x00\x01a\x003\x00\x01c\x004\x00\x08\x08'
result = {'a': {'a': '3', 'c': '4'}}
self.assertEqual(vdf.binary_loads(test, merge_duplicate_keys=False), result)
def test_raise_on_remaining(self):
# default binary_loads is to raise
with self.assertRaises(SyntaxError):
vdf.binary_loads(b'\x01key\x00value\x00\x08' + b'aaaa')
# do not raise
self.assertEqual(vdf.binary_loads(b'\x01key\x00value\x00\x08' + b'aaaa', raise_on_remaining=False), {'key': 'value'})
def test_raise_on_remaining_with_file(self):
buf = BytesIO(b'\x01key\x00value\x00\x08' + b'aaaa')
# binary_load doesn't raise by default
self.assertEqual(vdf.binary_load(buf), {'key': 'value'})
self.assertEqual(buf.read(), b'aaaa')
# raise when extra data remains
buf.seek(0)
with self.assertRaises(SyntaxError):
vdf.binary_load(buf, raise_on_remaining=True)
self.assertEqual(buf.read(), b'aaaa')
def test_key_table(self):
test = b'\x01\x00\x00\x00\x00value\x00\x01\x02\x00\x00\x00value3\x00\x08'
key_table = ['key', 'key2', 'key3', 'key4']
self.assertEqual({'key': 'value', 'key3': 'value3'}, vdf.binary_loads(test, key_table=key_table))
def test_vbkv_loads_empty(self):
with self.assertRaises(ValueError):
vdf.vbkv_loads(b'')
def test_vbkv_dumps_empty(self):
self.assertEqual(vdf.vbkv_dumps({}), b'VBKV\x00\x00\x00\x00')
def test_vbkv_loads_invalid_header(self):
with self.assertRaises(ValueError):
vdf.vbkv_loads(b'DD1235764tdffhghsdf')
def test_vbkv_loads_invalid_checksum(self):
with self.assertRaises(ValueError):
vdf.vbkv_loads(b'VBKV\x01\x02\x03\x04\x00a\x00\x0b\x0b')
def test_loads_utf8_invalmid(self):
self.assertEqual({'aaa': b'bb\xef\xbf\xbdbb'.decode('utf-8')}, vdf.binary_loads(b'\x01aaa\x00bb\xffbb\x00\x08'))
def test_loads_utf16(self):
self.assertEqual({'aaa': b'b\x00b\x00\xff\xffb\x00b\x00'.decode('utf-16le')}, vdf.binary_loads(b'\x05aaa\x00b\x00b\x00\xff\xffb\x00b\x00\x00\x00\x08'))