diff options
author | Andrej Shadura <andrew.shadura@collabora.co.uk> | 2022-05-17 11:43:37 +0200 |
---|---|---|
committer | Andrej Shadura <andrew.shadura@collabora.co.uk> | 2022-05-17 11:43:37 +0200 |
commit | 5f97349d6a81a0e59e837093290cb43aee708270 (patch) | |
tree | b0b4e2f63deaafb3d06e0cdbf2f0c2bf82d75721 /tests/test_unpaddedbase64.py | |
parent | 35f29cba23fc56f87acaff1741ec8d921db0c59f (diff) |
New upstream version 2.1.0
Diffstat (limited to 'tests/test_unpaddedbase64.py')
-rw-r--r-- | tests/test_unpaddedbase64.py | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/tests/test_unpaddedbase64.py b/tests/test_unpaddedbase64.py new file mode 100644 index 0000000..90d21a0 --- /dev/null +++ b/tests/test_unpaddedbase64.py @@ -0,0 +1,41 @@ +# Copyright 2015 OpenMarket Ltd +# Copyright 2021 The Matrix.org Foundation C.I.C. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +from unpaddedbase64 import encode_base64, decode_base64 +import unittest + + +class TestUnpaddedBase64(unittest.TestCase): + def test_encode(self) -> None: + self.assertEqual(encode_base64(b""), "") + self.assertEqual(encode_base64(b"\x00"), "AA") + self.assertEqual(encode_base64(b"\x00\x00"), "AAA") + self.assertEqual(encode_base64(b"\x00\x00\x00"), "AAAA") + + def test_decode(self) -> None: + self.assertEqual(decode_base64(""), b"") + self.assertEqual(decode_base64("AA"), b"\x00") + self.assertEqual(decode_base64("AAA"), b"\x00\x00") + self.assertEqual(decode_base64("AAAA"), b"\x00\x00\x00") + with self.assertRaises(Exception): + decode_base64("A") + + def test_encode_urlunsafe_chars(self) -> None: + self.assertEqual(encode_base64(b"\xff\xe6\x9a"), "/+aa") + self.assertEqual(encode_base64(b"\xff\xe6\x9a", True), "_-aa") + + def test_decode_urlunsafe_chars(self) -> None: + self.assertEqual(decode_base64("/+aa"), b"\xff\xe6\x9a") + self.assertEqual(decode_base64("_-aa"), b"\xff\xe6\x9a") |