encode_test.py

import unittest

import encode


class TestMain(unittest.TestCase):
    def test_empty_input(self):
        output = encode.main([])
        self.assertEqual(0, len(output["items"]))

    def test_spaces(self):
        output = encode.main(["hello", "alfred"])
        self.assertEqual("aGVsbG8gYWxmcmVk", output["items"][0]["arg"])


class TestEncode(unittest.TestCase):
    def test_base64(self):
        self.assertEqual("aGVsbG8gYWxmcmVk", encode.encode_base64("hello alfred"))

    def test_json(self):
        self.assertEqual(r'"{\"hello\": \"alfred\"}"', encode.encode_json('{"hello": "alfred"}'))

    def test_url(self):
        self.assertEqual("hello%20alfred", encode.encode_url("hello alfred"))

    def test_html(self):
        self.assertEqual("&lt;br/&gt;", encode.encode_html("<br/>"))


if __name__ == "__main__":
    unittest.main()