def testStreamEncryptor(self):
crypt = EncryptStream(self.password)
c = self.bigdata
mid = len(c)//2
crypt.write(c[:mid])
crypt.write(c[mid:])
crypt.close()
enc = crypt.read()
dec = decrypt_string(self.password, enc)
self.assertEqual(c, dec)
def testStreamers(self):
crypt = EncryptStream(self.password)
clear = ""
for dummy in xrange(64):
data = os.urandom(1024)
crypt.write(data)
clear += data
crypt.close()
crypt2 = DecryptStream(self.password)
crypt2.write(crypt.read())
crypt2.close()
self.assertEqual(clear, crypt2.read())
