import base64
import unittest

import encoding


class EncodingTests(unittest.TestCase):
    def test_str_roundtrip(self):
        original = "foo\nbar\0"
        encoded = base64.b64encode(original.encode())
        self.assertEqual(encoding.decode_str(encoded), original)

    def test_bytes_roundtrip(self):
        original = b"foo\nbar\0"
        encoded = base64.b64encode(original)
        self.assertEqual(encoding.decode_bytes(encoded), original)


unittest.main()