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

generate 3 files, check that streamed is the largest, add method to list files

Browse source
This commit is contained in:
Victor Shyba 2022-09-29 17:29:26 -03:00
parent f650e8f07e
commit e862c99f6c
2 changed files with 17 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
lbry/torrent/session.py
View file

@ -4,7 +4,7 @@ import os
import logging import logging
import random import random
from tempfile import mkdtemp from tempfile import mkdtemp
from typing import Optional, Tuple from typing import Optional, Tuple, Dict
import libtorrent import libtorrent
@ -154,7 +154,7 @@ class TorrentSession:
self.tasks = [] self.tasks = []
self.wait_start = True self.wait_start = True
async def add_fake_torrent(self, file_count=1): async def add_fake_torrent(self, file_count=3):
tmpdir = mkdtemp() tmpdir = mkdtemp()
info = _create_fake_torrent(tmpdir, file_count=file_count) info = _create_fake_torrent(tmpdir, file_count=file_count)
flags = libtorrent.add_torrent_params_flags_t.flag_seed_mode flags = libtorrent.add_torrent_params_flags_t.flag_seed_mode
@ -259,20 +259,30 @@ class TorrentSession:
handle = self._handles[btih] handle = self._handles[btih]
return handle.stream_range_as_completed(handle.largest_file_index, start, end) return handle.stream_range_as_completed(handle.largest_file_index, start, end)
def get_files(self, btih) -> Dict:
handle = self._handles[btih]
return {
handle.torrent_file.file_path(file_num): handle.torrent_file.file_size(file_num)
for file_num in range(handle.torrent_file.num_files())
if '.pad' not in handle.torrent_file.file_path(file_num)
}
def get_magnet_uri(btih): def get_magnet_uri(btih):
return f"magnet:?xt=urn:btih:{btih}" return f"magnet:?xt=urn:btih:{btih}"
def _create_fake_torrent(tmpdir, file_count=1): def _create_fake_torrent(tmpdir, file_count=3, largest_index=1):
# layout: subdir/tmp{0..file_count-1} 40k files. v1+v2. automatic piece size. # layout: subdir/tmp{0..file_count-1} files. v1+v2. automatic piece size.
# largest_index: which file index {0 ... file_count} will be the largest file
file_storage = libtorrent.file_storage() file_storage = libtorrent.file_storage()
subfolder = os.path.join(tmpdir, "subdir") subfolder = os.path.join(tmpdir, "subdir")
os.mkdir(subfolder) os.mkdir(subfolder)
for file_number in range(file_count): for file_number in range(file_count):
file_name = f"tmp{file_number}" file_name = f"tmp{file_number}"
with open(os.path.join(subfolder, file_name), 'wb') as myfile: with open(os.path.join(subfolder, file_name), 'wb') as myfile:
size = myfile.write(bytes([random.randint(0, 255) for _ in range(40)]) * 1024) size = myfile.write(
bytes([random.randint(0, 255) for _ in range(10 - abs(file_number - largest_index))]) * 1024)
file_storage.add_file(os.path.join("subdir", file_name), size) file_storage.add_file(os.path.join("subdir", file_name), size)
t = libtorrent.create_torrent(file_storage, 0, 0) t = libtorrent.create_torrent(file_storage, 0, 0)
libtorrent.set_piece_hashes(t, tmpdir) libtorrent.set_piece_hashes(t, tmpdir)

2
tests/integration/datanetwork/test_file_commands.py
View file

@ -71,6 +71,8 @@ class FileCommands(CommandTestCase):
self.assertEqual(content_length, len(streamed_bytes)) self.assertEqual(content_length, len(streamed_bytes))
self.assertEqual(f"bytes 0-{expected_size - 1}/{expected_size}", content_range) self.assertEqual(f"bytes 0-{expected_size - 1}/{expected_size}", content_range)
self.assertEqual(len(streamed_bytes), max(self.seeder_session.get_files(btih).values()))
@skipIf(TorrentSession is None, "libtorrent not installed") @skipIf(TorrentSession is None, "libtorrent not installed")
async def test_download_torrent(self): async def test_download_torrent(self):
tx, btih = await self.initialize_torrent() tx, btih = await self.initialize_torrent()