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