decode_test.py

import unittest

import decode


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

    def test_invalid(self):
        output = decode.main(["nope"])
        self.assertEqual("-", output["items"][0]["title"])
        self.assertFalse(output["items"][0]["valid"])


class TestDecode(unittest.TestCase):
    def test_base64(self):
        self.assertEqual("hello alfred", decode.decode_base64("aGVsbG8gYWxmcmVk"))

    def test_base64_invalid(self):
        with self.assertRaises(Exception):
            decode.decode_base64("nope")

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

    def test_json_invalid(self):
        with self.assertRaises(Exception):
            decode.decode_json("nope")

    def test_url(self):
        self.assertEqual("hello alfred", decode.decode_url("hello%20alfred"))

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


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