Gitiles
Code Review Sign In
gerrit-public.fairphone.software / platform / external / python / cpython3 / 31d8896ee21c28ee55ae4ba2764e11fba88c9fd4 / . / Lib / test / test_base64.py
blob: a7fa2b98904b35449be574a6e9bdb38c7a5aee56 [file] [log] [blame]
Walter Dörwald21d3a322003-05-01 17:45:56 +0000[diff] [blame]1import unittest
2from test import test_support
3import base64
Raymond Hettinger2ae87532002-05-18 00:25:10 +0000[diff] [blame]4
Raymond Hettinger2ae87532002-05-18 00:25:10 +0000[diff] [blame]5
Barry Warsaw4f019d32004-01-04 01:13:02 +0000[diff] [blame]6
7class LegacyBase64TestCase(unittest.TestCase):
Guido van Rossumcb682582002-08-22 19:18:56 +0000[diff] [blame]8 def test_encodestring(self):
Barry Warsaw4f019d32004-01-04 01:13:02 +0000[diff] [blame]9 eq = self.assertEqual
10 eq(base64.encodestring("www.python.org"), "d3d3LnB5dGhvbi5vcmc=\n")
11 eq(base64.encodestring("a"), "YQ==\n")
12 eq(base64.encodestring("ab"), "YWI=\n")
13 eq(base64.encodestring("abc"), "YWJj\n")
14 eq(base64.encodestring(""), "")
15 eq(base64.encodestring("abcdefghijklmnopqrstuvwxyz"
16 "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
17 "0123456789!@#0^&*();:<>,. []{}"),
18 "YWJjZGVmZ2hpamtsbW5vcHFyc3R1dnd4eXpBQkNE"
19 "RUZHSElKS0xNTk9QUVJTVFVWV1hZWjAxMjM0\nNT"
20 "Y3ODkhQCMwXiYqKCk7Ojw+LC4gW117fQ==\n")
Guido van Rossumcb682582002-08-22 19:18:56 +0000[diff] [blame]21
22 def test_decodestring(self):
Barry Warsaw4f019d32004-01-04 01:13:02 +0000[diff] [blame]23 eq = self.assertEqual
24 eq(base64.decodestring("d3d3LnB5dGhvbi5vcmc=\n"), "www.python.org")
25 eq(base64.decodestring("YQ==\n"), "a")
26 eq(base64.decodestring("YWI=\n"), "ab")
27 eq(base64.decodestring("YWJj\n"), "abc")
28 eq(base64.decodestring("YWJjZGVmZ2hpamtsbW5vcHFyc3R1dnd4eXpBQkNE"
29 "RUZHSElKS0xNTk9QUVJTVFVWV1hZWjAxMjM0\nNT"
30 "Y3ODkhQCMwXiYqKCk7Ojw+LC4gW117fQ==\n"),
31 "abcdefghijklmnopqrstuvwxyz"
32 "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
33 "0123456789!@#0^&*();:<>,. []{}")
34 eq(base64.decodestring(''), '')
35
36 def test_encode(self):
37 eq = self.assertEqual
38 from cStringIO import StringIO
39 infp = StringIO('abcdefghijklmnopqrstuvwxyz'
40 'ABCDEFGHIJKLMNOPQRSTUVWXYZ'
41 '0123456789!@#0^&*();:<>,. []{}')
42 outfp = StringIO()
43 base64.encode(infp, outfp)
44 eq(outfp.getvalue(),
45 'YWJjZGVmZ2hpamtsbW5vcHFyc3R1dnd4eXpBQkNE'
46 'RUZHSElKS0xNTk9QUVJTVFVWV1hZWjAxMjM0\nNT'
47 'Y3ODkhQCMwXiYqKCk7Ojw+LC4gW117fQ==\n')
48
49 def test_decode(self):
50 from cStringIO import StringIO
51 infp = StringIO('d3d3LnB5dGhvbi5vcmc=')
52 outfp = StringIO()
53 base64.decode(infp, outfp)
54 self.assertEqual(outfp.getvalue(), 'www.python.org')
55
56
57
58class BaseXYTestCase(unittest.TestCase):
59 def test_b64encode(self):
60 eq = self.assertEqual
61 # Test default alphabet
62 eq(base64.b64encode("www.python.org"), "d3d3LnB5dGhvbi5vcmc=")
63 eq(base64.b64encode("a"), "YQ==")
64 eq(base64.b64encode("ab"), "YWI=")
65 eq(base64.b64encode("abc"), "YWJj")
66 eq(base64.b64encode(""), "")
67 eq(base64.b64encode("abcdefghijklmnopqrstuvwxyz"
68 "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
69 "0123456789!@#0^&*();:<>,. []{}"),
70 "YWJjZGVmZ2hpamtsbW5vcHFyc3R1dnd4eXpBQkNE"
71 "RUZHSElKS0xNTk9QUVJTVFVWV1hZWjAxMjM0NT"
72 "Y3ODkhQCMwXiYqKCk7Ojw+LC4gW117fQ==")
73 # Test with arbitrary alternative characters
74 eq(base64.b64encode('\xd3V\xbeo\xf7\x1d', altchars='*$'), '01a*b$cd')
75 # Test standard alphabet
76 eq(base64.standard_b64encode("www.python.org"), "d3d3LnB5dGhvbi5vcmc=")
77 eq(base64.standard_b64encode("a"), "YQ==")
78 eq(base64.standard_b64encode("ab"), "YWI=")
79 eq(base64.standard_b64encode("abc"), "YWJj")
80 eq(base64.standard_b64encode(""), "")
81 eq(base64.standard_b64encode("abcdefghijklmnopqrstuvwxyz"
82 "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
83 "0123456789!@#0^&*();:<>,. []{}"),
84 "YWJjZGVmZ2hpamtsbW5vcHFyc3R1dnd4eXpBQkNE"
85 "RUZHSElKS0xNTk9QUVJTVFVWV1hZWjAxMjM0NT"
86 "Y3ODkhQCMwXiYqKCk7Ojw+LC4gW117fQ==")
87 # Test with 'URL safe' alternative characters
88 eq(base64.urlsafe_b64encode('\xd3V\xbeo\xf7\x1d'), '01a-b_cd')
89
90 def test_b64decode(self):
91 eq = self.assertEqual
92 eq(base64.b64decode("d3d3LnB5dGhvbi5vcmc="), "www.python.org")
93 eq(base64.b64decode("YQ=="), "a")
94 eq(base64.b64decode("YWI="), "ab")
95 eq(base64.b64decode("YWJj"), "abc")
96 eq(base64.b64decode("YWJjZGVmZ2hpamtsbW5vcHFyc3R1dnd4eXpBQkNE"
97 "RUZHSElKS0xNTk9QUVJTVFVWV1hZWjAxMjM0\nNT"
98 "Y3ODkhQCMwXiYqKCk7Ojw+LC4gW117fQ=="),
99 "abcdefghijklmnopqrstuvwxyz"
100 "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
101 "0123456789!@#0^&*();:<>,. []{}")
102 eq(base64.b64decode(''), '')
103 # Test with arbitrary alternative characters
104 eq(base64.b64decode('01a*b$cd', altchars='*$'), '\xd3V\xbeo\xf7\x1d')
105 # Test standard alphabet
106 eq(base64.standard_b64decode("d3d3LnB5dGhvbi5vcmc="), "www.python.org")
107 eq(base64.standard_b64decode("YQ=="), "a")
108 eq(base64.standard_b64decode("YWI="), "ab")
109 eq(base64.standard_b64decode("YWJj"), "abc")
110 eq(base64.standard_b64decode(""), "")
111 eq(base64.standard_b64decode("YWJjZGVmZ2hpamtsbW5vcHFyc3R1dnd4eXpBQkNE"
112 "RUZHSElKS0xNTk9QUVJTVFVWV1hZWjAxMjM0NT"
113 "Y3ODkhQCMwXiYqKCk7Ojw+LC4gW117fQ=="),
114 "abcdefghijklmnopqrstuvwxyz"
115 "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
116 "0123456789!@#0^&*();:<>,. []{}")
117 # Test with 'URL safe' alternative characters
118 eq(base64.urlsafe_b64decode('01a-b_cd'), '\xd3V\xbeo\xf7\x1d')
119
120 def test_b64decode_error(self):
121 self.assertRaises(TypeError, base64.b64decode, 'abc')
122
123 def test_b32encode(self):
124 eq = self.assertEqual
125 eq(base64.b32encode(''), '')
126 eq(base64.b32encode('a'), 'ME======')
127 eq(base64.b32encode('ab'), 'MFRA====')
128 eq(base64.b32encode('abc'), 'MFRGG===')
129 eq(base64.b32encode('abcd'), 'MFRGGZA=')
130 eq(base64.b32encode('abcde'), 'MFRGGZDF')
131
132 def test_b32decode(self):
133 eq = self.assertEqual
134 eq(base64.b32decode(''), '')
135 eq(base64.b32decode('ME======'), 'a')
136 eq(base64.b32decode('MFRA===='), 'ab')
137 eq(base64.b32decode('MFRGG==='), 'abc')
138 eq(base64.b32decode('MFRGGZA='), 'abcd')
139 eq(base64.b32decode('MFRGGZDF'), 'abcde')
140
141 def test_b32decode_casefold(self):
142 eq = self.assertEqual
143 eq(base64.b32decode('', True), '')
144 eq(base64.b32decode('ME======', True), 'a')
145 eq(base64.b32decode('MFRA====', True), 'ab')
146 eq(base64.b32decode('MFRGG===', True), 'abc')
147 eq(base64.b32decode('MFRGGZA=', True), 'abcd')
148 eq(base64.b32decode('MFRGGZDF', True), 'abcde')
149 # Lower cases
150 eq(base64.b32decode('me======', True), 'a')
151 eq(base64.b32decode('mfra====', True), 'ab')
152 eq(base64.b32decode('mfrgg===', True), 'abc')
153 eq(base64.b32decode('mfrggza=', True), 'abcd')
154 eq(base64.b32decode('mfrggzdf', True), 'abcde')
155 # Expected exceptions
156 self.assertRaises(TypeError, base64.b32decode, 'me======')
157 # Mapping zero and one
158 eq(base64.b32decode('MLO23456'), 'b\xdd\xad\xf3\xbe')
159 eq(base64.b32decode('M1023456', map01='L'), 'b\xdd\xad\xf3\xbe')
160 eq(base64.b32decode('M1023456', map01='I'), 'b\x1d\xad\xf3\xbe')
161
162 def test_b32decode_error(self):
163 self.assertRaises(TypeError, base64.b32decode, 'abc')
164 self.assertRaises(TypeError, base64.b32decode, 'ABCDEF==')
165
166 def test_b16encode(self):
167 eq = self.assertEqual
168 eq(base64.b16encode('\x01\x02\xab\xcd\xef'), '0102ABCDEF')
169
170 def test_b16decode(self):
171 eq = self.assertEqual
172 eq(base64.b16decode('0102ABCDEF'), '\x01\x02\xab\xcd\xef')
173 # Lower case is not allowed without a flag
174 self.assertRaises(TypeError, base64.b16decode, '0102abcdef')
175 # Case fold
176 eq(base64.b16decode('0102abcdef', True), '\x01\x02\xab\xcd\xef')
177
178
179
180def suite():
181 suite = unittest.TestSuite()
182 suite.addTest(unittest.makeSuite(LegacyBase64TestCase))
183 suite.addTest(unittest.makeSuite(BaseXYTestCase))
184 return suite
185
Tim Peters8ac14952002-05-23 15:15:30 +0000[diff] [blame]186
Raymond Hettinger2ae87532002-05-18 00:25:10 +0000[diff] [blame]187def test_main():
Barry Warsaw4f019d32004-01-04 01:13:02 +0000[diff] [blame]188 test_support.run_suite(suite())
Raymond Hettinger2ae87532002-05-18 00:25:10 +0000[diff] [blame]189
Barry Warsaw4f019d32004-01-04 01:13:02 +0000[diff] [blame]190
191if __name__ == '__main__':
192 unittest.main(defaultTest='suite')