LBRYCommunity/lbry-sdk
1
0
Fork 1
Code Issues Pull requests Projects Releases 282 Packages Wiki Activity

fix test

Browse source
This commit is contained in:
Kay Kurokawa 2017-09-29 14:43:29 -04:00
parent 96d8cb17d9
commit 51d4668763
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
lbrynet/tests/unit/cryptstream/test_cryptblob.py
View file

@ -9,6 +9,7 @@ from Crypto import Random
from Crypto.Cipher import AES from Crypto.Cipher import AES
import random import random
import string import string
import StringIO
class MocBlob(object): class MocBlob(object):
def __init__(self): def __init__(self):
@ -19,6 +20,9 @@ class MocBlob(object):
write_func(data) write_func(data)
return defer.succeed(True) return defer.succeed(True)
def open_for_reading(self):
return StringIO.StringIO(self.data)
def write(self, data): def write(self, data):
self.data += data self.data += data
@ -64,7 +68,7 @@ class TestCryptBlob(unittest.TestCase):
# decrypt string # decrypt string
decryptor = CryptBlob.StreamBlobDecryptor(blob, key, iv, size_of_data) decryptor = CryptBlob.StreamBlobDecryptor(blob, key, iv, size_of_data)
decryptor.decrypt(write_func) yield decryptor.decrypt(write_func)
self.assertEqual(self.data_buf, string_to_encrypt) self.assertEqual(self.data_buf, string_to_encrypt)
@defer.inlineCallbacks @defer.inlineCallbacks