def test_crypt_iterator_content_hash(self): original_data = StringIO.StringIO("test data to be iterated") iterator = FileCrypt('12345678901234561234567890123456', '1234567890123456') enc_iter = iterator.encrypt(original_data) for enc_data in enc_iter: pass original_data.seek(0) content_hash = hashlib.md5() content_hash.update(original_data.read()) content_hash = content_hash.hexdigest() self.assertEquals(content_hash, enc_iter.get_content_hash())
def test_crypt_iterator_content_hash(self): original_data = io.StringIO("test data to be iterated") iterator = FileCrypt('12345678901234561234567890123456', '1234567890123456') enc_iter = iterator.encrypt(original_data) for enc_data in enc_iter: pass original_data.seek(0) content_hash = hashlib.md5() content_hash.update(original_data.read()) content_hash = content_hash.hexdigest() self.assertEquals(content_hash, enc_iter.get_content_hash())
def test_crypt_iterator(self): original_data = StringIO.StringIO("test data to be iterated") enc_iterator = FileCrypt('Invalidkey', '1234567890123456') self.assertRaises(ECCryptException, enc_iterator.encrypt, original_data, 1) enc_iterator = FileCrypt('12345678901234561234567890123456', 'InvalidIV') self.assertRaises(ECCryptException, enc_iterator.encrypt, original_data, 1) enc_iterator = FileCrypt('Invalidkey', 'InvalidIV') self.assertRaises(ECCryptException, enc_iterator.encrypt, original_data, 1) enc_iterator = FileCrypt('12345678901234561234567890123456', '1234567890123456') data_enc_out = StringIO.StringIO() encrypt = enc_iterator.encrypt(original_data, 1) for data in encrypt: data_enc_out.write(data) data_enc_out.seek(0) self.assertNotEqual(original_data.read(), data_enc_out.read()) data_enc_out.seek(0) data_dec_out = StringIO.StringIO() dec_iterator = FileCrypt('Invalidkey', '1234567890123456') self.assertRaises(ECCryptException, dec_iterator.decrypt, data_enc_out, 1) dec_iterator = FileCrypt('12345678901234561234567890123456', 'InvalidIV') self.assertRaises(ECCryptException, dec_iterator.decrypt, data_enc_out, 1) dec_iterator = FileCrypt('Invalidkey', 'InvalidIV') self.assertRaises(ECCryptException, dec_iterator.decrypt, data_enc_out, 1) dec_iterator = FileCrypt('12345678901234561234567890123456', '1234567890123456') decrypt = dec_iterator.decrypt(data_enc_out, 1) for data in decrypt: data_dec_out.write(data) original_data.seek(0), data_dec_out.seek(0) self.assertEquals(original_data.read(), data_dec_out.read())
def test_crypt_iterator(self): original_data = io.StringIO("test data to be iterated") enc_iterator = FileCrypt('Invalidkey', '1234567890123456') self.assertRaises(ECCryptException, enc_iterator.encrypt, original_data, 1) enc_iterator = FileCrypt('12345678901234561234567890123456', 'InvalidIV') self.assertRaises(ECCryptException, enc_iterator.encrypt, original_data, 1) enc_iterator = FileCrypt('Invalidkey', 'InvalidIV') self.assertRaises(ECCryptException, enc_iterator.encrypt, original_data, 1) enc_iterator = FileCrypt('12345678901234561234567890123456', '1234567890123456') data_enc_out = io.StringIO() encrypt = enc_iterator.encrypt(original_data, 1) for data in encrypt: data_enc_out.write(data) data_enc_out.seek(0) self.assertNotEqual(original_data.read(), data_enc_out.read()) data_enc_out.seek(0) data_dec_out = io.StringIO() dec_iterator = FileCrypt('Invalidkey', '1234567890123456') self.assertRaises(ECCryptException, dec_iterator.decrypt, data_enc_out, 1) dec_iterator = FileCrypt('12345678901234561234567890123456', 'InvalidIV') self.assertRaises(ECCryptException, dec_iterator.decrypt, data_enc_out, 1) dec_iterator = FileCrypt('Invalidkey', 'InvalidIV') self.assertRaises(ECCryptException, dec_iterator.decrypt, data_enc_out, 1) dec_iterator = FileCrypt('12345678901234561234567890123456', '1234567890123456') decrypt = dec_iterator.decrypt(data_enc_out, 1) for data in decrypt: data_dec_out.write(data) original_data.seek(0), data_dec_out.seek(0) self.assertEquals(original_data.read(), data_dec_out.read())