bemaniutils/bemani/tests/test_AES.py

# vim: set fileencoding=utf-8
import unittest

from bemani.common import AESCipher


class TestAESCipher(unittest.TestCase):
    def test_pad(self) -> None:
        aes = AESCipher("a wonderful key")
        self.assertEqual(aes._pad(""), "0.--------------")
        self.assertEqual(aes._unpad(aes._pad("")), "")
        self.assertEqual(aes._pad("1337"), "4.1337----------")
        self.assertEqual(aes._unpad(aes._pad("1337")), "1337")
        self.assertEqual(aes._pad("aaaaaaaaaaaaaaaa"), "16.aaaaaaaaaaaaaaaa-------------")
        self.assertEqual(aes._unpad(aes._pad("aaaaaaaaaaaaaaaa")), "aaaaaaaaaaaaaaaa")
        self.assertEqual(aes._pad("aaaaaaaaaaaaa"), "13.aaaaaaaaaaaaa")
        self.assertEqual(aes._unpad(aes._pad("aaaaaaaaaaaaa")), "aaaaaaaaaaaaa")

    def test_crypto(self) -> None:
        aes = AESCipher("a wonderful key")
        ciphertext = aes.encrypt("testing")
        plaintext = aes.decrypt(ciphertext)
        self.assertEqual(plaintext, "testing")
        self.assertNotEqual(ciphertext, plaintext)