class TestMessageCrypto(unittest.TestCase):
def setUp(self):
self.crypto = MessageCrypto("1111111111111111")
def test_encrypt_generates_new_iv(self):
encrypted_message_0 = self.crypto.encrypt("testtesttesttest")
encrypted_message_1 = self.crypto.encrypt("testtesttesttest")
self.assertNotEqual(encrypted_message_0, encrypted_message_1)
def test_message_padding(self):
encrypted_message_0 = self.crypto.encrypt("test")
self.assertNotEqual(encrypted_message_0, None)
def test_encrypt_decrypt(self):
self.assertEqual(self.crypto.decrypt(self.crypto.encrypt("test")), "test")
def setUp(self):
self.crypto = MessageCrypto("1111111111111111")
