Gecko [ 37julay.space ]

Name	Size	Permission	Date
__pycache__	[ DIR ]	drwxr-xr-x	2024-01-22 04:29
testdata	[ DIR ]	drwxr-xr-x	2024-01-22 04:29
__init__.py	1.66 KB	-rw-r--r--	2021-09-09 21:33
b64.py	1.4 KB	-rw-r--r--	2021-09-09 21:33
b64_test.py	2.2 KB	-rw-r--r--	2021-09-09 21:33
errors.py	762 B	-rw-r--r--	2021-09-09 21:33
errors_test.py	463 B	-rw-r--r--	2021-09-09 21:33
interfaces.py	7.51 KB	-rw-r--r--	2021-09-09 21:33
interfaces_test.py	3.51 KB	-rw-r--r--	2021-09-09 21:33
json_util.py	15.35 KB	-rw-r--r--	2021-09-09 21:33
json_util_test.py	13.83 KB	-rw-r--r--	2021-09-09 21:33
jwa.py	8.17 KB	-rw-r--r--	2021-09-09 21:33
jwa_test.py	8.8 KB	-rw-r--r--	2021-09-09 21:33
jwk.py	13.12 KB	-rw-r--r--	2021-09-09 21:33
jwk_test.py	12.69 KB	-rw-r--r--	2021-09-09 21:33
jws.py	13.84 KB	-rw-r--r--	2021-09-09 21:33
jws_test.py	8.32 KB	-rw-r--r--	2021-09-09 21:33
magic_typing.py	529 B	-rw-r--r--	2021-09-09 21:33
magic_typing_test.py	1.05 KB	-rw-r--r--	2021-09-09 21:33
test_util.py	2.57 KB	-rw-r--r--	2021-09-09 21:33
util.py	8.37 KB	-rw-r--r--	2021-09-09 21:33
util_test.py	8.15 KB	-rw-r--r--	2021-09-09 21:33

Rename

"""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