REDROOM

Navigation

Breadcrumbs

Create

Preview: b64_test.py Size: 2.20 KB

/proc/thread-self/root/lib/python3.6/site-packages/josepy/b64_test.py

"""Tests for josepy.b64."""
import unittest


# https://en.wikipedia.org/wiki/Base64#Examples
B64_PADDING_EXAMPLES = {
    b'any carnal pleasure.': (b'YW55IGNhcm5hbCBwbGVhc3VyZS4', b'='),
    b'any carnal pleasure': (b'YW55IGNhcm5hbCBwbGVhc3VyZQ', b'=='),
    b'any carnal pleasur': (b'YW55IGNhcm5hbCBwbGVhc3Vy', b''),
    b'any carnal pleasu': (b'YW55IGNhcm5hbCBwbGVhc3U', b'='),
    b'any carnal pleas': (b'YW55IGNhcm5hbCBwbGVhcw', b'=='),
}


B64_URL_UNSAFE_EXAMPLES = {
    bytes((251, 239)): b'--8',
    bytes((255,)) * 2: b'__8',
}


class B64EncodeTest(unittest.TestCase):
    """Tests for josepy.b64.b64encode."""

    @classmethod
    def _call(cls, data):
        from josepy.b64 import b64encode
        return b64encode(data)

    def test_empty(self):
        self.assertEqual(self._call(b''), b'')

    def test_unsafe_url(self):
        for text, b64 in B64_URL_UNSAFE_EXAMPLES.items():
            self.assertEqual(self._call(text), b64)

    def test_different_paddings(self):
        for text, (b64, _) in B64_PADDING_EXAMPLES.items():
            self.assertEqual(self._call(text), b64)

    def test_unicode_fails_with_type_error(self):
        self.assertRaises(TypeError, self._call, u'some unicode')


class B64DecodeTest(unittest.TestCase):
    """Tests for josepy.b64.b64decode."""

    @classmethod
    def _call(cls, data):
        from josepy.b64 import b64decode
        return b64decode(data)

    def test_unsafe_url(self):
        for text, b64 in B64_URL_UNSAFE_EXAMPLES.items():
            self.assertEqual(self._call(b64), text)

    def test_input_without_padding(self):
        for text, (b64, _) in B64_PADDING_EXAMPLES.items():
            self.assertEqual(self._call(b64), text)

    def test_input_with_padding(self):
        for text, (b64, pad) in B64_PADDING_EXAMPLES.items():
            self.assertEqual(self._call(b64 + pad), text)

    def test_unicode_with_ascii(self):
        self.assertEqual(self._call(u'YQ'), b'a')

    def test_non_ascii_unicode_fails(self):
        self.assertRaises(ValueError, self._call, u'\u0105')

    def test_type_error_no_unicode_or_bytes(self):
        self.assertRaises(TypeError, self._call, object())


if __name__ == '__main__':
    unittest.main()  # pragma: no cover

Name	Size	Perms	Modified	Actions
testdata DIR	-	drwxr-xr-x	2025-04-13 08:21:05	Edit Download recursive recursive
__pycache__ DIR	-	drwxr-xr-x	2025-04-13 08:21:05	Edit Download recursive recursive
b64.py	1.40 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
b64_test.py	2.20 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
errors.py	762 B	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
errors_test.py	463 B	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
interfaces.py	7.51 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
interfaces_test.py	3.51 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
json_util.py	15.35 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
json_util_test.py	13.83 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
jwa.py	8.17 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
jwa_test.py	8.80 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
jwk.py	13.12 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
jwk_test.py	12.69 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
jws.py	13.84 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
jws_test.py	8.32 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
magic_typing.py	529 B	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
magic_typing_test.py	1.05 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
test_util.py	2.57 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
util.py	8.37 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
util_test.py	8.15 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive
__init__.py	1.66 KB	lrw-r--r--	2021-09-09 21:33:18	Edit Download recursive recursive

Directory Contents