def test_password(self):
password = '******'
key_file = AESCrypto.create_keys(password)
try:
AESCrypto(key_file, password)
assert True
except WrongPassword:
assert False
def test_wrong_password(self):
password = '******'
key_file = AESCrypto.create_keys(password)
wrong_password = "******"
try:
AESCrypto(key_file, wrong_password)
assert False
except WrongPassword:
assert True
def test_encryption_input_error(self):
text = "Spr scrt mssg, dnt tll nyn."
password = '******'
key_file = AESCrypto.create_keys(password)
aes_crypto = AESCrypto(key_file, password)
try:
aes_crypto.encrypt(text)
assert False
except TypeError:
assert True
def test_encrypt_decrypt(self):
text = b"A very very secret message."
password = '******'
id = 1
key_file = AESCrypto.create_keys(password)
aes_crypto = AESCrypto(key_file, password)
token = aes_crypto.encrypt(text, id)
assert text != token
data_id, data = aes_crypto.decrypt(token)
assert id == data_id
assert text == data
def test_decryption_error(self):
text = b"Spr scrt mssg, dnt tll nyn."
password = '******'
key_file = AESCrypto.create_keys(password)
aes_crypto = AESCrypto(key_file, password)
ciphertext = aes_crypto.encrypt(text, 5)
try:
aes_crypto.decrypt(ciphertext[10:50])
assert False
except InvalidToken:
assert True
def verify_pw(pw):
key_file = KeyFile.load_from_file()
return AESCrypto(key_file, pw)
def create_keys(pw):
key_file = AESCrypto.create_keys(pw)
key_file.save_to_file()
