Skip to content
Closed
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
more base64 test cases for unpadded input
  • Loading branch information
guillp authored May 24, 2018
commit df1102c42a5ddd06da85d0e65b2cd13f08de173f
31 changes: 31 additions & 0 deletions Lib/test/test_base64.py
Original file line number Diff line number Diff line change
Expand Up @@ -248,6 +248,7 @@ def test_b64decode(self):
self.assertEqual(base64.b64decode(b'abc', padded=False), b'i\xb7')
self.assertEqual(base64.b64decode('abc', padded=False), b'i\xb7')
self.assertEqual(base64.urlsafe_b64decode('abcd_-', padded=False), b'i\xb7\x1d\xff')
self.assertEquals(base64.b64decode(b'ab=', padded=False), b'i')

def test_b64decode_padding_error(self):
self.assertRaises(binascii.Error, base64.b64decode, b'abc')
Expand Down Expand Up @@ -283,6 +284,36 @@ def test_b64decode_invalid_chars(self):
res = b'\xFB\xEF\xBE\xFF\xFF\xFF'
self.assertEqual(base64.b64decode(b'++[[//]]', b'[]'), res)
self.assertEqual(base64.urlsafe_b64decode(b'++--//__'), res)

def test_b64decode_unpadded_invalid_chars(self):
# issue 1466065: Test some invalid characters.
tests = ((b'%3d', b'\xdd'),
(b'$3d', b'\xdd'),
(b'[', b''),
(b'YW]3', b'am'),
(b'3{d', b'\xdd'),
(b'3d}', b'\xdd'),
(b'@@', b''),
(b'!', b''),
(b'YWJj\nYWI', b'abcab'))
funcs = (
base64.b64decode,
base64.urlsafe_b64decode,
)
for bstr, res in tests:
for func in funcs:
with self.subTest(bstr=bstr, func=func):
self.assertEqual(func(bstr, padded=False), res)
self.assertEqual(func(bstr.decode('ascii'), padded=False), res)
with self.assertRaises(binascii.Error):
base64.b64decode(bstr, validate=True, padded=False)
with self.assertRaises(binascii.Error):
base64.b64decode(bstr.decode('ascii'), validate=True, padded=False)

# Normal alphabet characters not discarded when alternative given
res = b'\xFB\xEF\xBE\xFF\xFF\xFF'
self.assertEqual(base64.b64decode(b'++[[//]]', b'[]'), res)
self.assertEqual(base64.urlsafe_b64decode(b'++--//__'), res)

def test_b32encode(self):
eq = self.assertEqual
Expand Down