Mercurial > hg
annotate tests/test-cbor.py @ 49742:30eb36d93072
path: use `get_clone_path_obj` in _getlocal
We don't need to feed the `path` object to a `peer` object, but using an higher
level function is simpler here (e.g. no subscript access, explicit attribute
access).
author | Pierre-Yves David <pierre-yves.david@octobus.net> |
---|---|
date | Fri, 02 Dec 2022 16:36:43 +0100 |
parents | 6000f5b25c9b |
children | 493034cc3265 |
rev | line source |
---|---|
41160
a9905045b992
tests: get access to thirdparty.cbor without requiring it to be installed
Augie Fackler <augie@google.com>
parents:
40124
diff
changeset
|
1 import os |
a9905045b992
tests: get access to thirdparty.cbor without requiring it to be installed
Augie Fackler <augie@google.com>
parents:
40124
diff
changeset
|
2 import sys |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
3 import unittest |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
4 |
41160
a9905045b992
tests: get access to thirdparty.cbor without requiring it to be installed
Augie Fackler <augie@google.com>
parents:
40124
diff
changeset
|
5 # TODO migrate to canned cbor test strings and stop using thirdparty.cbor |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
6 tpp = os.path.normpath( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
7 os.path.join(os.path.dirname(__file__), '..', 'mercurial', 'thirdparty') |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
8 ) |
41160
a9905045b992
tests: get access to thirdparty.cbor without requiring it to be installed
Augie Fackler <augie@google.com>
parents:
40124
diff
changeset
|
9 if not os.path.exists(tpp): |
a9905045b992
tests: get access to thirdparty.cbor without requiring it to be installed
Augie Fackler <augie@google.com>
parents:
40124
diff
changeset
|
10 # skip, not in a repo |
a9905045b992
tests: get access to thirdparty.cbor without requiring it to be installed
Augie Fackler <augie@google.com>
parents:
40124
diff
changeset
|
11 sys.exit(80) |
a9905045b992
tests: get access to thirdparty.cbor without requiring it to be installed
Augie Fackler <augie@google.com>
parents:
40124
diff
changeset
|
12 sys.path[0:0] = [tpp] |
a9905045b992
tests: get access to thirdparty.cbor without requiring it to be installed
Augie Fackler <augie@google.com>
parents:
40124
diff
changeset
|
13 import cbor |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
14 |
41160
a9905045b992
tests: get access to thirdparty.cbor without requiring it to be installed
Augie Fackler <augie@google.com>
parents:
40124
diff
changeset
|
15 del sys.path[0] |
a9905045b992
tests: get access to thirdparty.cbor without requiring it to be installed
Augie Fackler <augie@google.com>
parents:
40124
diff
changeset
|
16 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
17 from mercurial.utils import cborutil |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
18 |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
19 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
20 class TestCase(unittest.TestCase): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
21 if not getattr(unittest.TestCase, 'assertRaisesRegex', False): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
22 # Python 3.7 deprecates the regex*p* version, but 2.7 lacks |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
23 # the regex version. |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
24 assertRaisesRegex = ( # camelcase-required |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
25 unittest.TestCase.assertRaisesRegexp |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
26 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
27 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
28 |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
29 def loadit(it): |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
30 return cbor.loads(b''.join(it)) |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
31 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
32 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
33 class BytestringTests(TestCase): |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
34 def testsimple(self): |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
35 self.assertEqual( |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
36 list(cborutil.streamencode(b'foobar')), [b'\x46', b'foobar'] |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
37 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
38 |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
39 self.assertEqual(loadit(cborutil.streamencode(b'foobar')), b'foobar') |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
40 |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
41 self.assertEqual(cborutil.decodeall(b'\x46foobar'), [b'foobar']) |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
42 |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
43 self.assertEqual( |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
44 cborutil.decodeall(b'\x46foobar\x45fizbi'), [b'foobar', b'fizbi'] |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
45 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
46 |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
47 def testlong(self): |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
48 source = b'x' * 1048576 |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
49 |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
50 self.assertEqual(loadit(cborutil.streamencode(source)), source) |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
51 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
52 encoded = b''.join(cborutil.streamencode(source)) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
53 self.assertEqual(cborutil.decodeall(encoded), [source]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
54 |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
55 def testfromiter(self): |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
56 # This is the example from RFC 7049 Section 2.2.2. |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
57 source = [b'\xaa\xbb\xcc\xdd', b'\xee\xff\x99'] |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
58 |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
59 self.assertEqual( |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
60 list(cborutil.streamencodebytestringfromiter(source)), |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
61 [ |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
62 b'\x5f', |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
63 b'\x44', |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
64 b'\xaa\xbb\xcc\xdd', |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
65 b'\x43', |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
66 b'\xee\xff\x99', |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
67 b'\xff', |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
68 ], |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
69 ) |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
70 |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
71 self.assertEqual( |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
72 loadit(cborutil.streamencodebytestringfromiter(source)), |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
73 b''.join(source), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
74 ) |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
75 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
76 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
77 cborutil.decodeall( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
78 b'\x5f\x44\xaa\xbb\xcc\xdd' b'\x43\xee\xff\x99\xff' |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
79 ), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
80 [b'\xaa\xbb\xcc\xdd', b'\xee\xff\x99', b''], |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
81 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
82 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
83 for i, chunk in enumerate( |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
84 cborutil.decodeall( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
85 b'\x5f\x44\xaa\xbb\xcc\xdd' b'\x43\xee\xff\x99\xff' |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
86 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
87 ): |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
88 self.assertIsInstance(chunk, cborutil.bytestringchunk) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
89 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
90 if i == 0: |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
91 self.assertTrue(chunk.isfirst) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
92 else: |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
93 self.assertFalse(chunk.isfirst) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
94 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
95 if i == 2: |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
96 self.assertTrue(chunk.islast) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
97 else: |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
98 self.assertFalse(chunk.islast) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
99 |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
100 def testfromiterlarge(self): |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
101 source = [b'a' * 16, b'b' * 128, b'c' * 1024, b'd' * 1048576] |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
102 |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
103 self.assertEqual( |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
104 loadit(cborutil.streamencodebytestringfromiter(source)), |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
105 b''.join(source), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
106 ) |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
107 |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
108 def testindefinite(self): |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
109 source = b'\x00\x01\x02\x03' + b'\xff' * 16384 |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
110 |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
111 it = cborutil.streamencodeindefinitebytestring(source, chunksize=2) |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
112 |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
113 self.assertEqual(next(it), b'\x5f') |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
114 self.assertEqual(next(it), b'\x42') |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
115 self.assertEqual(next(it), b'\x00\x01') |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
116 self.assertEqual(next(it), b'\x42') |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
117 self.assertEqual(next(it), b'\x02\x03') |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
118 self.assertEqual(next(it), b'\x42') |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
119 self.assertEqual(next(it), b'\xff\xff') |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
120 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
121 dest = b''.join( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
122 cborutil.streamencodeindefinitebytestring(source, chunksize=42) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
123 ) |
37899
2b3b6187c316
tests: port test-cbor.py to Python 3
Augie Fackler <augie@google.com>
parents:
37711
diff
changeset
|
124 self.assertEqual(cbor.loads(dest), source) |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
125 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
126 self.assertEqual(b''.join(cborutil.decodeall(dest)), source) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
127 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
128 for chunk in cborutil.decodeall(dest): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
129 self.assertIsInstance(chunk, cborutil.bytestringchunk) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
130 self.assertIn(len(chunk), (0, 8, 42)) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
131 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
132 encoded = b'\x5f\xff' |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
133 b = cborutil.decodeall(encoded) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
134 self.assertEqual(b, [b'']) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
135 self.assertTrue(b[0].isfirst) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
136 self.assertTrue(b[0].islast) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
137 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
138 def testdecodevariouslengths(self): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
139 for i in (0, 1, 22, 23, 24, 25, 254, 255, 256, 65534, 65535, 65536): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
140 source = b'x' * i |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
141 encoded = b''.join(cborutil.streamencode(source)) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
142 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
143 if len(source) < 24: |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
144 hlen = 1 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
145 elif len(source) < 256: |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
146 hlen = 2 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
147 elif len(source) < 65536: |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
148 hlen = 3 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
149 elif len(source) < 1048576: |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
150 hlen = 5 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
151 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
152 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
153 cborutil.decodeitem(encoded), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
154 (True, source, hlen + len(source), cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
155 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
156 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
157 def testpartialdecode(self): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
158 encoded = b''.join(cborutil.streamencode(b'foobar')) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
159 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
160 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
161 cborutil.decodeitem(encoded[0:1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
162 (False, None, -6, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
163 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
164 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
165 cborutil.decodeitem(encoded[0:2]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
166 (False, None, -5, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
167 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
168 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
169 cborutil.decodeitem(encoded[0:3]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
170 (False, None, -4, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
171 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
172 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
173 cborutil.decodeitem(encoded[0:4]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
174 (False, None, -3, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
175 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
176 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
177 cborutil.decodeitem(encoded[0:5]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
178 (False, None, -2, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
179 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
180 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
181 cborutil.decodeitem(encoded[0:6]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
182 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
183 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
184 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
185 cborutil.decodeitem(encoded[0:7]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
186 (True, b'foobar', 7, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
187 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
188 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
189 def testpartialdecodevariouslengths(self): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
190 lens = [ |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
191 2, |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
192 3, |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
193 10, |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
194 23, |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
195 24, |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
196 25, |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
197 31, |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
198 100, |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
199 254, |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
200 255, |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
201 256, |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
202 257, |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
203 16384, |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
204 65534, |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
205 65535, |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
206 65536, |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
207 65537, |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
208 131071, |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
209 131072, |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
210 131073, |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
211 1048575, |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
212 1048576, |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
213 1048577, |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
214 ] |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
215 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
216 for size in lens: |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
217 if size < 24: |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
218 hlen = 1 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
219 elif size < 2 ** 8: |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
220 hlen = 2 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
221 elif size < 2 ** 16: |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
222 hlen = 3 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
223 elif size < 2 ** 32: |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
224 hlen = 5 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
225 else: |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
226 assert False |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
227 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
228 source = b'x' * size |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
229 encoded = b''.join(cborutil.streamencode(source)) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
230 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
231 res = cborutil.decodeitem(encoded[0:1]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
232 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
233 if hlen > 1: |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
234 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
235 res, (False, None, -(hlen - 1), cborutil.SPECIAL_NONE) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
236 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
237 else: |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
238 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
239 res, |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
240 (False, None, -(size + hlen - 1), cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
241 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
242 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
243 # Decoding partial header reports remaining header size. |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
244 for i in range(hlen - 1): |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
245 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
246 cborutil.decodeitem(encoded[0 : i + 1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
247 (False, None, -(hlen - i - 1), cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
248 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
249 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
250 # Decoding complete header reports item size. |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
251 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
252 cborutil.decodeitem(encoded[0:hlen]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
253 (False, None, -size, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
254 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
255 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
256 # Decoding single byte after header reports item size - 1 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
257 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
258 cborutil.decodeitem(encoded[0 : hlen + 1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
259 (False, None, -(size - 1), cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
260 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
261 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
262 # Decoding all but the last byte reports -1 needed. |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
263 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
264 cborutil.decodeitem(encoded[0 : hlen + size - 1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
265 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
266 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
267 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
268 # Decoding last byte retrieves value. |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
269 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
270 cborutil.decodeitem(encoded[0 : hlen + size]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
271 (True, source, hlen + size, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
272 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
273 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
274 def testindefinitepartialdecode(self): |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
275 encoded = b''.join( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
276 cborutil.streamencodebytestringfromiter([b'foobar', b'biz']) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
277 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
278 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
279 # First item should be begin of bytestring special. |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
280 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
281 cborutil.decodeitem(encoded[0:1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
282 (True, None, 1, cborutil.SPECIAL_START_INDEFINITE_BYTESTRING), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
283 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
284 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
285 # Second item should be the first chunk. But only available when |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
286 # we give it 7 bytes (1 byte header + 6 byte chunk). |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
287 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
288 cborutil.decodeitem(encoded[1:2]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
289 (False, None, -6, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
290 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
291 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
292 cborutil.decodeitem(encoded[1:3]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
293 (False, None, -5, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
294 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
295 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
296 cborutil.decodeitem(encoded[1:4]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
297 (False, None, -4, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
298 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
299 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
300 cborutil.decodeitem(encoded[1:5]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
301 (False, None, -3, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
302 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
303 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
304 cborutil.decodeitem(encoded[1:6]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
305 (False, None, -2, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
306 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
307 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
308 cborutil.decodeitem(encoded[1:7]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
309 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
310 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
311 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
312 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
313 cborutil.decodeitem(encoded[1:8]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
314 (True, b'foobar', 7, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
315 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
316 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
317 # Third item should be second chunk. But only available when |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
318 # we give it 4 bytes (1 byte header + 3 byte chunk). |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
319 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
320 cborutil.decodeitem(encoded[8:9]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
321 (False, None, -3, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
322 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
323 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
324 cborutil.decodeitem(encoded[8:10]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
325 (False, None, -2, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
326 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
327 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
328 cborutil.decodeitem(encoded[8:11]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
329 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
330 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
331 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
332 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
333 cborutil.decodeitem(encoded[8:12]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
334 (True, b'biz', 4, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
335 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
336 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
337 # Fourth item should be end of indefinite stream marker. |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
338 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
339 cborutil.decodeitem(encoded[12:13]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
340 (True, None, 1, cborutil.SPECIAL_INDEFINITE_BREAK), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
341 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
342 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
343 # Now test the behavior when going through the decoder. |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
344 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
345 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
346 cborutil.sansiodecoder().decode(encoded[0:1]), (False, 1, 0) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
347 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
348 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
349 cborutil.sansiodecoder().decode(encoded[0:2]), (False, 1, 6) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
350 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
351 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
352 cborutil.sansiodecoder().decode(encoded[0:3]), (False, 1, 5) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
353 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
354 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
355 cborutil.sansiodecoder().decode(encoded[0:4]), (False, 1, 4) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
356 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
357 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
358 cborutil.sansiodecoder().decode(encoded[0:5]), (False, 1, 3) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
359 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
360 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
361 cborutil.sansiodecoder().decode(encoded[0:6]), (False, 1, 2) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
362 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
363 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
364 cborutil.sansiodecoder().decode(encoded[0:7]), (False, 1, 1) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
365 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
366 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
367 cborutil.sansiodecoder().decode(encoded[0:8]), (True, 8, 0) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
368 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
369 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
370 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
371 cborutil.sansiodecoder().decode(encoded[0:9]), (True, 8, 3) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
372 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
373 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
374 cborutil.sansiodecoder().decode(encoded[0:10]), (True, 8, 2) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
375 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
376 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
377 cborutil.sansiodecoder().decode(encoded[0:11]), (True, 8, 1) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
378 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
379 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
380 cborutil.sansiodecoder().decode(encoded[0:12]), (True, 12, 0) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
381 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
382 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
383 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
384 cborutil.sansiodecoder().decode(encoded[0:13]), (True, 13, 0) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
385 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
386 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
387 decoder = cborutil.sansiodecoder() |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
388 decoder.decode(encoded[0:8]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
389 values = decoder.getavailable() |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
390 self.assertEqual(values, [b'foobar']) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
391 self.assertTrue(values[0].isfirst) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
392 self.assertFalse(values[0].islast) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
393 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
394 self.assertEqual(decoder.decode(encoded[8:12]), (True, 4, 0)) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
395 values = decoder.getavailable() |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
396 self.assertEqual(values, [b'biz']) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
397 self.assertFalse(values[0].isfirst) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
398 self.assertFalse(values[0].islast) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
399 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
400 self.assertEqual(decoder.decode(encoded[12:]), (True, 1, 0)) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
401 values = decoder.getavailable() |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
402 self.assertEqual(values, [b'']) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
403 self.assertFalse(values[0].isfirst) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
404 self.assertTrue(values[0].islast) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
405 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
406 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
407 class StringTests(TestCase): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
408 def testdecodeforbidden(self): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
409 encoded = b'\x63foo' |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
410 with self.assertRaisesRegex( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
411 cborutil.CBORDecodeError, 'string major type not supported' |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
412 ): |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
413 cborutil.decodeall(encoded) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
414 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
415 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
416 class IntTests(TestCase): |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
417 def testsmall(self): |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
418 self.assertEqual(list(cborutil.streamencode(0)), [b'\x00']) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
419 self.assertEqual(cborutil.decodeall(b'\x00'), [0]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
420 |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
421 self.assertEqual(list(cborutil.streamencode(1)), [b'\x01']) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
422 self.assertEqual(cborutil.decodeall(b'\x01'), [1]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
423 |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
424 self.assertEqual(list(cborutil.streamencode(2)), [b'\x02']) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
425 self.assertEqual(cborutil.decodeall(b'\x02'), [2]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
426 |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
427 self.assertEqual(list(cborutil.streamencode(3)), [b'\x03']) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
428 self.assertEqual(cborutil.decodeall(b'\x03'), [3]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
429 |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
430 self.assertEqual(list(cborutil.streamencode(4)), [b'\x04']) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
431 self.assertEqual(cborutil.decodeall(b'\x04'), [4]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
432 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
433 # Multiple value decode works. |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
434 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
435 cborutil.decodeall(b'\x00\x01\x02\x03\x04'), [0, 1, 2, 3, 4] |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
436 ) |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
437 |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
438 def testnegativesmall(self): |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
439 self.assertEqual(list(cborutil.streamencode(-1)), [b'\x20']) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
440 self.assertEqual(cborutil.decodeall(b'\x20'), [-1]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
441 |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
442 self.assertEqual(list(cborutil.streamencode(-2)), [b'\x21']) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
443 self.assertEqual(cborutil.decodeall(b'\x21'), [-2]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
444 |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
445 self.assertEqual(list(cborutil.streamencode(-3)), [b'\x22']) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
446 self.assertEqual(cborutil.decodeall(b'\x22'), [-3]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
447 |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
448 self.assertEqual(list(cborutil.streamencode(-4)), [b'\x23']) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
449 self.assertEqual(cborutil.decodeall(b'\x23'), [-4]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
450 |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
451 self.assertEqual(list(cborutil.streamencode(-5)), [b'\x24']) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
452 self.assertEqual(cborutil.decodeall(b'\x24'), [-5]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
453 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
454 # Multiple value decode works. |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
455 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
456 cborutil.decodeall(b'\x20\x21\x22\x23\x24'), [-1, -2, -3, -4, -5] |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
457 ) |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
458 |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
459 def testrange(self): |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
460 for i in range(-70000, 70000, 10): |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
461 encoded = b''.join(cborutil.streamencode(i)) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
462 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
463 self.assertEqual(encoded, cbor.dumps(i)) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
464 self.assertEqual(cborutil.decodeall(encoded), [i]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
465 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
466 def testdecodepartialubyte(self): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
467 encoded = b''.join(cborutil.streamencode(250)) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
468 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
469 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
470 cborutil.decodeitem(encoded[0:1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
471 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
472 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
473 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
474 cborutil.decodeitem(encoded[0:2]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
475 (True, 250, 2, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
476 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
477 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
478 def testdecodepartialbyte(self): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
479 encoded = b''.join(cborutil.streamencode(-42)) |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
480 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
481 cborutil.decodeitem(encoded[0:1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
482 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
483 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
484 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
485 cborutil.decodeitem(encoded[0:2]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
486 (True, -42, 2, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
487 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
488 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
489 def testdecodepartialushort(self): |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
490 encoded = b''.join(cborutil.streamencode(2 ** 15)) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
491 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
492 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
493 cborutil.decodeitem(encoded[0:1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
494 (False, None, -2, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
495 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
496 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
497 cborutil.decodeitem(encoded[0:2]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
498 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
499 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
500 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
501 cborutil.decodeitem(encoded[0:5]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
502 (True, 2 ** 15, 3, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
503 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
504 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
505 def testdecodepartialshort(self): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
506 encoded = b''.join(cborutil.streamencode(-1024)) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
507 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
508 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
509 cborutil.decodeitem(encoded[0:1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
510 (False, None, -2, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
511 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
512 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
513 cborutil.decodeitem(encoded[0:2]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
514 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
515 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
516 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
517 cborutil.decodeitem(encoded[0:3]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
518 (True, -1024, 3, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
519 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
520 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
521 def testdecodepartialulong(self): |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
522 encoded = b''.join(cborutil.streamencode(2 ** 28)) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
523 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
524 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
525 cborutil.decodeitem(encoded[0:1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
526 (False, None, -4, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
527 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
528 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
529 cborutil.decodeitem(encoded[0:2]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
530 (False, None, -3, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
531 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
532 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
533 cborutil.decodeitem(encoded[0:3]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
534 (False, None, -2, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
535 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
536 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
537 cborutil.decodeitem(encoded[0:4]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
538 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
539 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
540 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
541 cborutil.decodeitem(encoded[0:5]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
542 (True, 2 ** 28, 5, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
543 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
544 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
545 def testdecodepartiallong(self): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
546 encoded = b''.join(cborutil.streamencode(-1048580)) |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
547 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
548 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
549 cborutil.decodeitem(encoded[0:1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
550 (False, None, -4, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
551 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
552 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
553 cborutil.decodeitem(encoded[0:2]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
554 (False, None, -3, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
555 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
556 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
557 cborutil.decodeitem(encoded[0:3]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
558 (False, None, -2, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
559 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
560 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
561 cborutil.decodeitem(encoded[0:4]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
562 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
563 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
564 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
565 cborutil.decodeitem(encoded[0:5]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
566 (True, -1048580, 5, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
567 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
568 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
569 def testdecodepartialulonglong(self): |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
570 encoded = b''.join(cborutil.streamencode(2 ** 32)) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
571 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
572 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
573 cborutil.decodeitem(encoded[0:1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
574 (False, None, -8, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
575 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
576 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
577 cborutil.decodeitem(encoded[0:2]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
578 (False, None, -7, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
579 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
580 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
581 cborutil.decodeitem(encoded[0:3]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
582 (False, None, -6, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
583 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
584 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
585 cborutil.decodeitem(encoded[0:4]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
586 (False, None, -5, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
587 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
588 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
589 cborutil.decodeitem(encoded[0:5]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
590 (False, None, -4, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
591 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
592 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
593 cborutil.decodeitem(encoded[0:6]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
594 (False, None, -3, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
595 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
596 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
597 cborutil.decodeitem(encoded[0:7]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
598 (False, None, -2, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
599 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
600 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
601 cborutil.decodeitem(encoded[0:8]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
602 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
603 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
604 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
605 cborutil.decodeitem(encoded[0:9]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
606 (True, 2 ** 32, 9, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
607 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
608 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
609 with self.assertRaisesRegex( |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
610 cborutil.CBORDecodeError, 'input data not fully consumed' |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
611 ): |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
612 cborutil.decodeall(encoded[0:1]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
613 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
614 with self.assertRaisesRegex( |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
615 cborutil.CBORDecodeError, 'input data not fully consumed' |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
616 ): |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
617 cborutil.decodeall(encoded[0:2]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
618 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
619 def testdecodepartiallonglong(self): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
620 encoded = b''.join(cborutil.streamencode(-7000000000)) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
621 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
622 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
623 cborutil.decodeitem(encoded[0:1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
624 (False, None, -8, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
625 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
626 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
627 cborutil.decodeitem(encoded[0:2]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
628 (False, None, -7, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
629 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
630 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
631 cborutil.decodeitem(encoded[0:3]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
632 (False, None, -6, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
633 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
634 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
635 cborutil.decodeitem(encoded[0:4]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
636 (False, None, -5, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
637 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
638 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
639 cborutil.decodeitem(encoded[0:5]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
640 (False, None, -4, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
641 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
642 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
643 cborutil.decodeitem(encoded[0:6]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
644 (False, None, -3, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
645 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
646 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
647 cborutil.decodeitem(encoded[0:7]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
648 (False, None, -2, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
649 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
650 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
651 cborutil.decodeitem(encoded[0:8]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
652 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
653 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
654 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
655 cborutil.decodeitem(encoded[0:9]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
656 (True, -7000000000, 9, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
657 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
658 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
659 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
660 class ArrayTests(TestCase): |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
661 def testempty(self): |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
662 self.assertEqual(list(cborutil.streamencode([])), [b'\x80']) |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
663 self.assertEqual(loadit(cborutil.streamencode([])), []) |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
664 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
665 self.assertEqual(cborutil.decodeall(b'\x80'), [[]]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
666 |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
667 def testbasic(self): |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
668 source = [b'foo', b'bar', 1, -10] |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
669 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
670 chunks = [b'\x84', b'\x43', b'foo', b'\x43', b'bar', b'\x01', b'\x29'] |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
671 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
672 self.assertEqual(list(cborutil.streamencode(source)), chunks) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
673 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
674 self.assertEqual(cborutil.decodeall(b''.join(chunks)), [source]) |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
675 |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
676 def testemptyfromiter(self): |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
677 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
678 b''.join(cborutil.streamencodearrayfromiter([])), b'\x9f\xff' |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
679 ) |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
680 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
681 with self.assertRaisesRegex( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
682 cborutil.CBORDecodeError, 'indefinite length uint not allowed' |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
683 ): |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
684 cborutil.decodeall(b'\x9f\xff') |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
685 |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
686 def testfromiter1(self): |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
687 source = [b'foo'] |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
688 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
689 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
690 list(cborutil.streamencodearrayfromiter(source)), |
45942
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
691 [ |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
692 b'\x9f', |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
693 b'\x43', |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
694 b'foo', |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
695 b'\xff', |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
696 ], |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
697 ) |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
698 |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
699 dest = b''.join(cborutil.streamencodearrayfromiter(source)) |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
700 self.assertEqual(cbor.loads(dest), source) |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
701 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
702 with self.assertRaisesRegex( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
703 cborutil.CBORDecodeError, 'indefinite length uint not allowed' |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
704 ): |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
705 cborutil.decodeall(dest) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
706 |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
707 def testtuple(self): |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
708 source = (b'foo', None, 42) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
709 encoded = b''.join(cborutil.streamencode(source)) |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
710 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
711 self.assertEqual(cbor.loads(encoded), list(source)) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
712 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
713 self.assertEqual(cborutil.decodeall(encoded), [list(source)]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
714 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
715 def testpartialdecode(self): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
716 source = list(range(4)) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
717 encoded = b''.join(cborutil.streamencode(source)) |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
718 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
719 cborutil.decodeitem(encoded[0:1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
720 (True, 4, 1, cborutil.SPECIAL_START_ARRAY), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
721 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
722 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
723 cborutil.decodeitem(encoded[0:2]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
724 (True, 4, 1, cborutil.SPECIAL_START_ARRAY), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
725 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
726 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
727 source = list(range(23)) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
728 encoded = b''.join(cborutil.streamencode(source)) |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
729 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
730 cborutil.decodeitem(encoded[0:1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
731 (True, 23, 1, cborutil.SPECIAL_START_ARRAY), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
732 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
733 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
734 cborutil.decodeitem(encoded[0:2]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
735 (True, 23, 1, cborutil.SPECIAL_START_ARRAY), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
736 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
737 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
738 source = list(range(24)) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
739 encoded = b''.join(cborutil.streamencode(source)) |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
740 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
741 cborutil.decodeitem(encoded[0:1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
742 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
743 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
744 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
745 cborutil.decodeitem(encoded[0:2]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
746 (True, 24, 2, cborutil.SPECIAL_START_ARRAY), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
747 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
748 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
749 cborutil.decodeitem(encoded[0:3]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
750 (True, 24, 2, cborutil.SPECIAL_START_ARRAY), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
751 ) |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
752 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
753 source = list(range(256)) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
754 encoded = b''.join(cborutil.streamencode(source)) |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
755 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
756 cborutil.decodeitem(encoded[0:1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
757 (False, None, -2, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
758 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
759 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
760 cborutil.decodeitem(encoded[0:2]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
761 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
762 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
763 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
764 cborutil.decodeitem(encoded[0:3]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
765 (True, 256, 3, cborutil.SPECIAL_START_ARRAY), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
766 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
767 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
768 cborutil.decodeitem(encoded[0:4]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
769 (True, 256, 3, cborutil.SPECIAL_START_ARRAY), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
770 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
771 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
772 def testnested(self): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
773 source = [[], [], [[], [], []]] |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
774 encoded = b''.join(cborutil.streamencode(source)) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
775 self.assertEqual(cborutil.decodeall(encoded), [source]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
776 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
777 source = [True, None, [True, 0, 2], [None], [], [[[]], -87]] |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
778 encoded = b''.join(cborutil.streamencode(source)) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
779 self.assertEqual(cborutil.decodeall(encoded), [source]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
780 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
781 # A set within an array. |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
782 source = [None, {b'foo', b'bar', None, False}, set()] |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
783 encoded = b''.join(cborutil.streamencode(source)) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
784 self.assertEqual(cborutil.decodeall(encoded), [source]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
785 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
786 # A map within an array. |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
787 source = [None, {}, {b'foo': b'bar', True: False}, [{}]] |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
788 encoded = b''.join(cborutil.streamencode(source)) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
789 self.assertEqual(cborutil.decodeall(encoded), [source]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
790 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
791 def testindefinitebytestringvalues(self): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
792 # Single value array whose value is an empty indefinite bytestring. |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
793 encoded = b'\x81\x5f\x40\xff' |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
794 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
795 with self.assertRaisesRegex( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
796 cborutil.CBORDecodeError, |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
797 'indefinite length bytestrings not ' 'allowed as array values', |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
798 ): |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
799 cborutil.decodeall(encoded) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
800 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
801 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
802 class SetTests(TestCase): |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
803 def testempty(self): |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
804 self.assertEqual( |
45942
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
805 list(cborutil.streamencode(set())), |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
806 [ |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
807 b'\xd9\x01\x02', |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
808 b'\x80', |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
809 ], |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
810 ) |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
811 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
812 self.assertEqual(cborutil.decodeall(b'\xd9\x01\x02\x80'), [set()]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
813 |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
814 def testset(self): |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
815 source = {b'foo', None, 42} |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
816 encoded = b''.join(cborutil.streamencode(source)) |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
817 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
818 self.assertEqual(cbor.loads(encoded), source) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
819 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
820 self.assertEqual(cborutil.decodeall(encoded), [source]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
821 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
822 def testinvalidtag(self): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
823 # Must use array to encode sets. |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
824 encoded = b'\xd9\x01\x02\xa0' |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
825 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
826 with self.assertRaisesRegex( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
827 cborutil.CBORDecodeError, |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
828 'expected array after finite set ' 'semantic tag', |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
829 ): |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
830 cborutil.decodeall(encoded) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
831 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
832 def testpartialdecode(self): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
833 # Semantic tag item will be 3 bytes. Set header will be variable |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
834 # depending on length. |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
835 encoded = b''.join(cborutil.streamencode({i for i in range(23)})) |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
836 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
837 cborutil.decodeitem(encoded[0:1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
838 (False, None, -2, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
839 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
840 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
841 cborutil.decodeitem(encoded[0:2]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
842 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
843 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
844 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
845 cborutil.decodeitem(encoded[0:3]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
846 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
847 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
848 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
849 cborutil.decodeitem(encoded[0:4]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
850 (True, 23, 4, cborutil.SPECIAL_START_SET), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
851 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
852 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
853 cborutil.decodeitem(encoded[0:5]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
854 (True, 23, 4, cborutil.SPECIAL_START_SET), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
855 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
856 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
857 encoded = b''.join(cborutil.streamencode({i for i in range(24)})) |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
858 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
859 cborutil.decodeitem(encoded[0:1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
860 (False, None, -2, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
861 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
862 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
863 cborutil.decodeitem(encoded[0:2]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
864 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
865 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
866 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
867 cborutil.decodeitem(encoded[0:3]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
868 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
869 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
870 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
871 cborutil.decodeitem(encoded[0:4]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
872 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
873 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
874 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
875 cborutil.decodeitem(encoded[0:5]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
876 (True, 24, 5, cborutil.SPECIAL_START_SET), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
877 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
878 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
879 cborutil.decodeitem(encoded[0:6]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
880 (True, 24, 5, cborutil.SPECIAL_START_SET), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
881 ) |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
882 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
883 encoded = b''.join(cborutil.streamencode({i for i in range(256)})) |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
884 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
885 cborutil.decodeitem(encoded[0:1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
886 (False, None, -2, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
887 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
888 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
889 cborutil.decodeitem(encoded[0:2]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
890 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
891 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
892 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
893 cborutil.decodeitem(encoded[0:3]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
894 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
895 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
896 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
897 cborutil.decodeitem(encoded[0:4]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
898 (False, None, -2, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
899 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
900 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
901 cborutil.decodeitem(encoded[0:5]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
902 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
903 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
904 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
905 cborutil.decodeitem(encoded[0:6]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
906 (True, 256, 6, cborutil.SPECIAL_START_SET), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
907 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
908 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
909 def testinvalidvalue(self): |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
910 encoded = b''.join( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
911 [ |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
912 b'\xd9\x01\x02', # semantic tag |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
913 b'\x81', # array of size 1 |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
914 b'\x5f\x43foo\xff', # indefinite length bytestring "foo" |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
915 ] |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
916 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
917 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
918 with self.assertRaisesRegex( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
919 cborutil.CBORDecodeError, |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
920 'indefinite length bytestrings not ' 'allowed as set values', |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
921 ): |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
922 cborutil.decodeall(encoded) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
923 |
45942
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
924 encoded = b''.join( |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
925 [ |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
926 b'\xd9\x01\x02', |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
927 b'\x81', |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
928 b'\x80', |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
929 ] |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
930 ) # empty array |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
931 |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
932 with self.assertRaisesRegex( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
933 cborutil.CBORDecodeError, 'collections not allowed as set values' |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
934 ): |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
935 cborutil.decodeall(encoded) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
936 |
45942
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
937 encoded = b''.join( |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
938 [ |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
939 b'\xd9\x01\x02', |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
940 b'\x81', |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
941 b'\xa0', |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
942 ] |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
943 ) # empty map |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
944 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
945 with self.assertRaisesRegex( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
946 cborutil.CBORDecodeError, 'collections not allowed as set values' |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
947 ): |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
948 cborutil.decodeall(encoded) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
949 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
950 encoded = b''.join( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
951 [ |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
952 b'\xd9\x01\x02', |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
953 b'\x81', |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
954 b'\xd9\x01\x02\x81\x01', # set with integer 1 |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
955 ] |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
956 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
957 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
958 with self.assertRaisesRegex( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
959 cborutil.CBORDecodeError, 'collections not allowed as set values' |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
960 ): |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
961 cborutil.decodeall(encoded) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
962 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
963 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
964 class BoolTests(TestCase): |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
965 def testbasic(self): |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
966 self.assertEqual(list(cborutil.streamencode(True)), [b'\xf5']) |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
967 self.assertEqual(list(cborutil.streamencode(False)), [b'\xf4']) |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
968 |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
969 self.assertIs(loadit(cborutil.streamencode(True)), True) |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
970 self.assertIs(loadit(cborutil.streamencode(False)), False) |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
971 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
972 self.assertEqual(cborutil.decodeall(b'\xf4'), [False]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
973 self.assertEqual(cborutil.decodeall(b'\xf5'), [True]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
974 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
975 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
976 cborutil.decodeall(b'\xf4\xf5\xf5\xf4'), [False, True, True, False] |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
977 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
978 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
979 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
980 class NoneTests(TestCase): |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
981 def testbasic(self): |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
982 self.assertEqual(list(cborutil.streamencode(None)), [b'\xf6']) |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
983 |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
984 self.assertIs(loadit(cborutil.streamencode(None)), None) |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
985 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
986 self.assertEqual(cborutil.decodeall(b'\xf6'), [None]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
987 self.assertEqual(cborutil.decodeall(b'\xf6\xf6'), [None, None]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
988 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
989 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
990 class MapTests(TestCase): |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
991 def testempty(self): |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
992 self.assertEqual(list(cborutil.streamencode({})), [b'\xa0']) |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
993 self.assertEqual(loadit(cborutil.streamencode({})), {}) |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
994 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
995 self.assertEqual(cborutil.decodeall(b'\xa0'), [{}]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
996 |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
997 def testemptyindefinite(self): |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
998 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
999 list(cborutil.streamencodemapfromiter([])), [b'\xbf', b'\xff'] |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1000 ) |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1001 |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1002 self.assertEqual(loadit(cborutil.streamencodemapfromiter([])), {}) |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1003 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1004 with self.assertRaisesRegex( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1005 cborutil.CBORDecodeError, 'indefinite length uint not allowed' |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1006 ): |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1007 cborutil.decodeall(b'\xbf\xff') |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1008 |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1009 def testone(self): |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1010 source = {b'foo': b'bar'} |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1011 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1012 list(cborutil.streamencode(source)), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1013 [b'\xa1', b'\x43', b'foo', b'\x43', b'bar'], |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1014 ) |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1015 |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1016 self.assertEqual(loadit(cborutil.streamencode(source)), source) |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1017 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1018 self.assertEqual(cborutil.decodeall(b'\xa1\x43foo\x43bar'), [source]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1019 |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1020 def testmultiple(self): |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1021 source = { |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1022 b'foo': b'bar', |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1023 b'baz': b'value1', |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1024 } |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1025 |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1026 self.assertEqual(loadit(cborutil.streamencode(source)), source) |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1027 |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1028 self.assertEqual( |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1029 loadit(cborutil.streamencodemapfromiter(source.items())), source |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1030 ) |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1031 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1032 encoded = b''.join(cborutil.streamencode(source)) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1033 self.assertEqual(cborutil.decodeall(encoded), [source]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1034 |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1035 def testcomplex(self): |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1036 source = { |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1037 b'key': 1, |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1038 2: -10, |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1039 } |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1040 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1041 self.assertEqual(loadit(cborutil.streamencode(source)), source) |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1042 |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1043 self.assertEqual( |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1044 loadit(cborutil.streamencodemapfromiter(source.items())), source |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1045 ) |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1046 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1047 encoded = b''.join(cborutil.streamencode(source)) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1048 self.assertEqual(cborutil.decodeall(encoded), [source]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1049 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1050 def testnested(self): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1051 source = {b'key1': None, b'key2': {b'sub1': b'sub2'}, b'sub2': {}} |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1052 encoded = b''.join(cborutil.streamencode(source)) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1053 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1054 self.assertEqual(cborutil.decodeall(encoded), [source]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1055 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1056 source = { |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1057 b'key1': [], |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1058 b'key2': [None, False], |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1059 b'key3': {b'foo', b'bar'}, |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1060 b'key4': {}, |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1061 } |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1062 encoded = b''.join(cborutil.streamencode(source)) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1063 self.assertEqual(cborutil.decodeall(encoded), [source]) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1064 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1065 def testillegalkey(self): |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1066 encoded = b''.join( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1067 [ |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1068 # map header + len 1 |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1069 b'\xa1', |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1070 # indefinite length bytestring "foo" in key position |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1071 b'\x5f\x03foo\xff', |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1072 ] |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1073 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1074 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1075 with self.assertRaisesRegex( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1076 cborutil.CBORDecodeError, |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1077 'indefinite length bytestrings not ' 'allowed as map keys', |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1078 ): |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1079 cborutil.decodeall(encoded) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1080 |
45942
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
1081 encoded = b''.join( |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
1082 [ |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
1083 b'\xa1', |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
1084 b'\x80', |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
1085 b'\x43foo', |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
1086 ] |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
1087 ) # empty array |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1088 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1089 with self.assertRaisesRegex( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1090 cborutil.CBORDecodeError, 'collections not supported as map keys' |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1091 ): |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1092 cborutil.decodeall(encoded) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1093 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1094 def testillegalvalue(self): |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1095 encoded = b''.join( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1096 [ |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1097 b'\xa1', # map headers |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1098 b'\x43foo', # key |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1099 b'\x5f\x03bar\xff', # indefinite length value |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1100 ] |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1101 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1102 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1103 with self.assertRaisesRegex( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1104 cborutil.CBORDecodeError, |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1105 'indefinite length bytestrings not ' 'allowed as map values', |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1106 ): |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1107 cborutil.decodeall(encoded) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1108 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1109 def testpartialdecode(self): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1110 source = {b'key1': b'value1'} |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1111 encoded = b''.join(cborutil.streamencode(source)) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1112 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1113 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1114 cborutil.decodeitem(encoded[0:1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1115 (True, 1, 1, cborutil.SPECIAL_START_MAP), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1116 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1117 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1118 cborutil.decodeitem(encoded[0:2]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1119 (True, 1, 1, cborutil.SPECIAL_START_MAP), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1120 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1121 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1122 source = {b'key%d' % i: None for i in range(23)} |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1123 encoded = b''.join(cborutil.streamencode(source)) |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1124 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1125 cborutil.decodeitem(encoded[0:1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1126 (True, 23, 1, cborutil.SPECIAL_START_MAP), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1127 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1128 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1129 source = {b'key%d' % i: None for i in range(24)} |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1130 encoded = b''.join(cborutil.streamencode(source)) |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1131 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1132 cborutil.decodeitem(encoded[0:1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1133 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1134 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1135 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1136 cborutil.decodeitem(encoded[0:2]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1137 (True, 24, 2, cborutil.SPECIAL_START_MAP), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1138 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1139 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1140 cborutil.decodeitem(encoded[0:3]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1141 (True, 24, 2, cborutil.SPECIAL_START_MAP), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1142 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1143 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1144 source = {b'key%d' % i: None for i in range(256)} |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1145 encoded = b''.join(cborutil.streamencode(source)) |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1146 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1147 cborutil.decodeitem(encoded[0:1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1148 (False, None, -2, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1149 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1150 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1151 cborutil.decodeitem(encoded[0:2]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1152 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1153 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1154 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1155 cborutil.decodeitem(encoded[0:3]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1156 (True, 256, 3, cborutil.SPECIAL_START_MAP), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1157 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1158 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1159 cborutil.decodeitem(encoded[0:4]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1160 (True, 256, 3, cborutil.SPECIAL_START_MAP), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1161 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1162 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1163 source = {b'key%d' % i: None for i in range(65536)} |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1164 encoded = b''.join(cborutil.streamencode(source)) |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1165 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1166 cborutil.decodeitem(encoded[0:1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1167 (False, None, -4, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1168 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1169 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1170 cborutil.decodeitem(encoded[0:2]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1171 (False, None, -3, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1172 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1173 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1174 cborutil.decodeitem(encoded[0:3]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1175 (False, None, -2, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1176 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1177 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1178 cborutil.decodeitem(encoded[0:4]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1179 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1180 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1181 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1182 cborutil.decodeitem(encoded[0:5]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1183 (True, 65536, 5, cborutil.SPECIAL_START_MAP), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1184 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1185 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1186 cborutil.decodeitem(encoded[0:6]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1187 (True, 65536, 5, cborutil.SPECIAL_START_MAP), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1188 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1189 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1190 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1191 class SemanticTagTests(TestCase): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1192 def testdecodeforbidden(self): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1193 for i in range(500): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1194 if i == cborutil.SEMANTIC_TAG_FINITE_SET: |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1195 continue |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1196 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1197 tag = cborutil.encodelength(cborutil.MAJOR_TYPE_SEMANTIC, i) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1198 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1199 encoded = tag + cborutil.encodelength(cborutil.MAJOR_TYPE_UINT, 42) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1200 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1201 # Partial decode is incomplete. |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1202 if i < 24: |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1203 pass |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1204 elif i < 256: |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1205 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1206 cborutil.decodeitem(encoded[0:1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1207 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1208 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1209 elif i < 65536: |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1210 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1211 cborutil.decodeitem(encoded[0:1]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1212 (False, None, -2, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1213 ) |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1214 self.assertEqual( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1215 cborutil.decodeitem(encoded[0:2]), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1216 (False, None, -1, cborutil.SPECIAL_NONE), |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1217 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1218 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1219 with self.assertRaisesRegex( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1220 cborutil.CBORDecodeError, r'semantic tag \d+ not allowed' |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1221 ): |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1222 cborutil.decodeitem(encoded) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1223 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1224 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1225 class SpecialTypesTests(TestCase): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1226 def testforbiddentypes(self): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1227 for i in range(256): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1228 if i == cborutil.SUBTYPE_FALSE: |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1229 continue |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1230 elif i == cborutil.SUBTYPE_TRUE: |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1231 continue |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1232 elif i == cborutil.SUBTYPE_NULL: |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1233 continue |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1234 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1235 encoded = cborutil.encodelength(cborutil.MAJOR_TYPE_SPECIAL, i) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1236 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1237 with self.assertRaisesRegex( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1238 cborutil.CBORDecodeError, r'special type \d+ not allowed' |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1239 ): |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1240 cborutil.decodeitem(encoded) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1241 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1242 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1243 class SansIODecoderTests(TestCase): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1244 def testemptyinput(self): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1245 decoder = cborutil.sansiodecoder() |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1246 self.assertEqual(decoder.decode(b''), (False, 0, 0)) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1247 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1248 |
39413
babad5ebaf0a
cborutil: add a buffering decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39412
diff
changeset
|
1249 class BufferingDecoderTests(TestCase): |
babad5ebaf0a
cborutil: add a buffering decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39412
diff
changeset
|
1250 def testsimple(self): |
babad5ebaf0a
cborutil: add a buffering decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39412
diff
changeset
|
1251 source = [ |
babad5ebaf0a
cborutil: add a buffering decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39412
diff
changeset
|
1252 b'foobar', |
babad5ebaf0a
cborutil: add a buffering decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39412
diff
changeset
|
1253 b'x' * 128, |
babad5ebaf0a
cborutil: add a buffering decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39412
diff
changeset
|
1254 {b'foo': b'bar'}, |
babad5ebaf0a
cborutil: add a buffering decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39412
diff
changeset
|
1255 True, |
babad5ebaf0a
cborutil: add a buffering decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39412
diff
changeset
|
1256 False, |
babad5ebaf0a
cborutil: add a buffering decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39412
diff
changeset
|
1257 None, |
43996
6dbb18e1ac8d
tests: avoid using a list comprehension to fill a list with fixed values
Matt Harbison <matt_harbison@yahoo.com>
parents:
43076
diff
changeset
|
1258 [None] * 128, |
39413
babad5ebaf0a
cborutil: add a buffering decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39412
diff
changeset
|
1259 ] |
babad5ebaf0a
cborutil: add a buffering decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39412
diff
changeset
|
1260 |
babad5ebaf0a
cborutil: add a buffering decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39412
diff
changeset
|
1261 encoded = b''.join(cborutil.streamencode(source)) |
babad5ebaf0a
cborutil: add a buffering decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39412
diff
changeset
|
1262 |
babad5ebaf0a
cborutil: add a buffering decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39412
diff
changeset
|
1263 for step in range(1, 32): |
babad5ebaf0a
cborutil: add a buffering decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39412
diff
changeset
|
1264 decoder = cborutil.bufferingdecoder() |
babad5ebaf0a
cborutil: add a buffering decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39412
diff
changeset
|
1265 start = 0 |
babad5ebaf0a
cborutil: add a buffering decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39412
diff
changeset
|
1266 |
babad5ebaf0a
cborutil: add a buffering decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39412
diff
changeset
|
1267 while start < len(encoded): |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1268 decoder.decode(encoded[start : start + step]) |
39413
babad5ebaf0a
cborutil: add a buffering decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39412
diff
changeset
|
1269 start += step |
babad5ebaf0a
cborutil: add a buffering decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39412
diff
changeset
|
1270 |
babad5ebaf0a
cborutil: add a buffering decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39412
diff
changeset
|
1271 self.assertEqual(decoder.getavailable(), [source]) |
babad5ebaf0a
cborutil: add a buffering decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39412
diff
changeset
|
1272 |
40124
b638219a23c3
cborutil: cast bytearray to bytes
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39413
diff
changeset
|
1273 def testbytearray(self): |
b638219a23c3
cborutil: cast bytearray to bytes
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39413
diff
changeset
|
1274 source = b''.join(cborutil.streamencode(b'foobar')) |
b638219a23c3
cborutil: cast bytearray to bytes
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39413
diff
changeset
|
1275 |
b638219a23c3
cborutil: cast bytearray to bytes
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39413
diff
changeset
|
1276 decoder = cborutil.bufferingdecoder() |
b638219a23c3
cborutil: cast bytearray to bytes
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39413
diff
changeset
|
1277 decoder.decode(bytearray(source)) |
b638219a23c3
cborutil: cast bytearray to bytes
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39413
diff
changeset
|
1278 |
b638219a23c3
cborutil: cast bytearray to bytes
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39413
diff
changeset
|
1279 self.assertEqual(decoder.getavailable(), [b'foobar']) |
b638219a23c3
cborutil: cast bytearray to bytes
Gregory Szorc <gregory.szorc@gmail.com>
parents:
39413
diff
changeset
|
1280 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1281 |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1282 class DecodeallTests(TestCase): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1283 def testemptyinput(self): |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1284 self.assertEqual(cborutil.decodeall(b''), []) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1285 |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1286 def testpartialinput(self): |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1287 encoded = b''.join( |
45942
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
1288 [ |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
1289 b'\x82', |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
1290 b'\x01', |
89a2afe31e82
formating: upgrade to black 20.8b1
Augie Fackler <raf@durin42.com>
parents:
43996
diff
changeset
|
1291 ] # array of 2 elements # integer 1 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1292 ) |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1293 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1294 with self.assertRaisesRegex( |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1295 cborutil.CBORDecodeError, 'input data not complete' |
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1296 ): |
39411
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1297 cborutil.decodeall(encoded) |
aeb551a3bb8a
cborutil: implement sans I/O decoder
Gregory Szorc <gregory.szorc@gmail.com>
parents:
37899
diff
changeset
|
1298 |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1299 |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1300 if __name__ == '__main__': |
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1301 import silenttestrunner |
43076
2372284d9457
formatting: blacken the codebase
Augie Fackler <augie@google.com>
parents:
41549
diff
changeset
|
1302 |
37711
65a23cc8e75b
cborutil: implement support for streaming encoding, bytestring decoding
Gregory Szorc <gregory.szorc@gmail.com>
parents:
diff
changeset
|
1303 silenttestrunner.main(__name__) |