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())