from community_qa_bot.aes import AESCipher


class TestAESCipher:
    def test_encrypt_decrypt(self):
        cipher = AESCipher(b'0123456789abcdef0123456789abcdef')
        input_text = 'mysecret123456'
        encrypted = cipher.encrypt(input_text)
        assert encrypted != input_text
        assert cipher.decrypt(encrypted) == input_text