lbry-sdk/tests/functional/test_streamify.py

108 lines
4.1 KiB
Python
Raw Normal View History

2016-10-20 18:23:39 +02:00
import os
import shutil
2018-02-12 20:19:39 +01:00
import tempfile
from hashlib import md5
2016-10-20 18:23:39 +02:00
from twisted.trial.unittest import TestCase
from twisted.internet import defer, threads
2016-10-20 18:23:39 +02:00
from lbrynet.core.StreamDescriptor import StreamDescriptorIdentifier
from lbrynet.core.BlobManager import DiskBlobManager
2018-02-12 20:19:39 +01:00
from lbrynet.core.StreamDescriptor import get_sd_info
from lbrynet.core.PeerManager import PeerManager
2017-09-29 12:44:22 +02:00
from lbrynet.core.RateLimiter import DummyRateLimiter
from lbrynet.database.storage import SQLiteStorage
from lbrynet.core.PaymentRateManager import OnlyFreePaymentsManager
from lbrynet.file_manager.EncryptedFileCreator import create_lbry_file
from lbrynet.file_manager.EncryptedFileManager import EncryptedFileManager
from tests import mocks
2016-10-20 18:23:39 +02:00
FakeNode = mocks.Node
FakeWallet = mocks.Wallet
FakePeerFinder = mocks.PeerFinder
FakeAnnouncer = mocks.Announcer
GenFile = mocks.GenFile
test_create_stream_sd_file = mocks.create_stream_sd_file
DummyBlobAvailabilityTracker = mocks.BlobAvailabilityTracker
class TestStreamify(TestCase):
2018-02-12 20:19:39 +01:00
maxDiff = 5000
2018-07-24 18:45:32 +02:00
2016-10-20 18:23:39 +02:00
def setUp(self):
2017-01-17 04:23:20 +01:00
mocks.mock_conf_settings(self)
2016-10-20 18:23:39 +02:00
self.session = None
self.lbry_file_manager = None
self.is_generous = True
2018-02-12 20:19:39 +01:00
self.db_dir = tempfile.mkdtemp()
self.blob_dir = os.path.join(self.db_dir, "blobfiles")
os.mkdir(self.blob_dir)
2018-07-24 18:45:32 +02:00
self.dht_node = FakeNode()
self.wallet = FakeWallet()
self.peer_manager = PeerManager()
self.peer_finder = FakePeerFinder(5553, self.peer_manager, 2)
self.rate_limiter = DummyRateLimiter()
self.sd_identifier = StreamDescriptorIdentifier()
self.storage = SQLiteStorage(self.db_dir)
self.blob_manager = DiskBlobManager(self.blob_dir, self.storage, self.dht_node._dataStore)
self.prm = OnlyFreePaymentsManager()
2018-08-03 15:36:03 +02:00
self.lbry_file_manager = EncryptedFileManager(
self.peer_finder, self.rate_limiter, self.blob_manager, self.wallet, self.prm, self.storage,
self.sd_identifier
)
d = self.storage.setup()
d.addCallback(lambda _: self.lbry_file_manager.setup())
return d
2018-02-12 20:19:39 +01:00
@defer.inlineCallbacks
def tearDown(self):
lbry_files = self.lbry_file_manager.lbry_files
for lbry_file in lbry_files:
yield self.lbry_file_manager.delete_lbry_file(lbry_file)
yield self.lbry_file_manager.stop()
yield self.storage.stop()
yield threads.deferToThread(shutil.rmtree, self.db_dir)
2018-02-12 20:19:39 +01:00
if os.path.exists("test_file"):
os.remove("test_file")
2016-10-20 18:23:39 +02:00
def test_create_stream(self):
def verify_equal(sd_info):
self.assertEqual(sd_info, test_create_stream_sd_file)
def verify_stream_descriptor_file(stream_hash):
d = get_sd_info(self.storage, stream_hash, True)
2016-10-20 18:23:39 +02:00
d.addCallback(verify_equal)
return d
def iv_generator():
iv = 0
while 1:
iv += 1
yield b"%016d" % iv
2016-10-20 18:23:39 +02:00
def create_stream():
2018-08-10 06:55:28 +02:00
test_file = GenFile(5209343, bytes((i + 3) for i in range(0, 64, 6)))
2018-08-03 15:36:03 +02:00
d = create_lbry_file(
self.blob_manager, self.storage, self.prm, self.lbry_file_manager, "test_file", test_file,
key=b'0123456701234567', iv_generator=iv_generator()
2018-08-03 15:36:03 +02:00
)
2018-02-12 20:19:39 +01:00
d.addCallback(lambda lbry_file: lbry_file.stream_hash)
2016-10-20 18:23:39 +02:00
return d
d = create_stream()
2016-10-20 18:23:39 +02:00
d.addCallback(verify_stream_descriptor_file)
return d
@defer.inlineCallbacks
2016-10-20 18:23:39 +02:00
def test_create_and_combine_stream(self):
2018-08-10 06:55:28 +02:00
test_file = GenFile(53209343, bytes((i + 5) for i in range(0, 64, 6)))
lbry_file = yield create_lbry_file(self.blob_manager, self.storage, self.prm, self.lbry_file_manager,
"test_file", test_file)
sd_hash = yield self.storage.get_sd_blob_hash_for_stream(lbry_file.stream_hash)
self.assertTrue(lbry_file.sd_hash, sd_hash)
yield lbry_file.start()
f = open('test_file', 'rb')
hashsum = md5()
hashsum.update(f.read())
self.assertEqual(hashsum.hexdigest(), "68959747edc73df45e45db6379dd7b3b")