def cipher_string_test():
key = ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(10))
lock = LockerCipher(key)
rand = ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(2048))
enc = lock.encrypt(rand)
dec = lock.decrypt(enc)
assert enc != dec
assert dec == rand
def cipher_test():
key = ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(10))
lock = LockerCipher(key)
lock.encrypt_file(tmp_file, "%s.enc" % tmp_file)
lock.decrypt_file("%s.enc" % tmp_file, "%s.dec" % tmp_file)
md5_in = hashlib.md5(open(tmp_file).read()).hexdigest()
md5_dec = hashlib.md5(open("%s.dec" % tmp_file).read()).hexdigest()
assert md5_in == md5_dec
