class TestLibtorrentDownloadImplNoSession(TriblerCoreTest):
def setUp(self, annotate=True):
TriblerCoreTest.setUp(self, annotate=annotate)
self.libtorrent_download_impl = LibtorrentDownloadImpl(None, None)
mock_handle = MockObject()
mock_status = MockObject()
mock_status.pieces = [True, False, True, True, False]
torrent_info = MockObject()
file_info = MockObject()
file_info.size = 1234
torrent_info.file_at = lambda _: file_info
map_file_result = MockObject()
map_file_result.piece = 123
torrent_info.map_file = lambda _dummy1, _dummy2, _dummy3: map_file_result
torrent_info.num_pieces = lambda: 5
mock_handle.is_valid = lambda: True
mock_handle.status = lambda: mock_status
mock_handle.get_torrent_info = lambda: torrent_info
mock_handle.set_sequential_download = lambda _: None
mock_handle.set_priority = lambda _: None
mock_handle.prioritize_pieces = lambda _: None
self.libtorrent_download_impl.handle = mock_handle
# Create a fake tdef
self.libtorrent_download_impl.tdef = MockObject()
self.libtorrent_download_impl.tdef.get_name = lambda: "ubuntu.iso"
self.libtorrent_download_impl.tdef.is_multifile_torrent = lambda: False
def tearDown(self, annotate=True):
self.libtorrent_download_impl.cancel_all_pending_tasks()
super(TestLibtorrentDownloadImplNoSession,
self).tearDown(annotate=annotate)
def test_selected_files(self):
"""
Test whether the selected files are set correctly
"""
def mocked_set_file_prios(_):
mocked_set_file_prios.called = True
mocked_set_file_prios.called = False
mocked_file = MockObject()
mocked_file.path = 'my/path'
mock_torrent_info = MockObject()
mock_torrent_info.files = lambda: [mocked_file, mocked_file]
self.libtorrent_download_impl.handle.prioritize_files = mocked_set_file_prios
self.libtorrent_download_impl.handle.get_torrent_info = lambda: mock_torrent_info
self.libtorrent_download_impl.handle.rename_file = lambda *_: None
self.libtorrent_download_impl.get_share_mode = lambda: False
self.libtorrent_download_impl.tdef.get_infohash = lambda: 'a' * 20
self.libtorrent_download_impl.orig_files = ['a', 'b']
self.libtorrent_download_impl.get_save_path = lambda: 'my/path'
self.libtorrent_download_impl.set_selected_files(['a'])
self.assertTrue(mocked_set_file_prios.called)
self.libtorrent_download_impl.get_share_mode = lambda: False
mocked_set_file_prios.called = False
self.assertFalse(mocked_set_file_prios.called)
def test_get_share_mode(self):
"""
Test whether we return the right share mode when requested in the LibtorrentDownloadImpl
"""
self.libtorrent_download_impl.handle.status().share_mode = False
self.assertFalse(self.libtorrent_download_impl.get_share_mode())
self.libtorrent_download_impl.handle.status().share_mode = True
self.assertTrue(self.libtorrent_download_impl.get_share_mode())
def test_set_share_mode(self):
"""
Test whether we set the right share mode in LibtorrentDownloadImpl
"""
def mocked_set_share_mode(val):
self.assertTrue(val)
mocked_set_share_mode.called = True
mocked_set_share_mode.called = False
self.libtorrent_download_impl.handle.set_share_mode = mocked_set_share_mode
self.libtorrent_download_impl.set_share_mode(True)
self.assertTrue(mocked_set_share_mode.called)
def test_set_priority(self):
"""
Test whether setting the priority calls the right methods in LibtorrentDownloadImpl
"""
def mocked_set_priority(prio):
self.assertEqual(prio, 1234)
mocked_set_priority.called = True
mocked_set_priority.called = False
self.libtorrent_download_impl.handle.set_priority = mocked_set_priority
self.libtorrent_download_impl.set_priority(1234)
self.assertTrue(mocked_set_priority.called)
def test_dlconfig_cb_change(self):
"""
Testing whether changing the configuration on runtime calls the right methods in LibtorrentDownloadImpl
"""
def mocked_set_upload_limit(prio):
self.assertEqual(prio, 3 * 1024)
mocked_set_upload_limit.called = True
mocked_set_upload_limit.called = False
self.libtorrent_download_impl.handle.set_upload_limit = mocked_set_upload_limit
def mocked_set_download_limit(prio):
self.assertEqual(prio, 3 * 1024)
mocked_set_download_limit.called = True
mocked_set_download_limit.called = False
self.libtorrent_download_impl.handle.set_download_limit = mocked_set_download_limit
self.libtorrent_download_impl.dlconfig_changed_callback(
'downloadconfig', 'max_upload_rate', 3, 4)
self.assertTrue(mocked_set_upload_limit)
self.libtorrent_download_impl.dlconfig_changed_callback(
'downloadconfig', 'max_download_rate', 3, 4)
self.assertTrue(mocked_set_download_limit)
self.assertFalse(
self.libtorrent_download_impl.dlconfig_changed_callback(
'downloadconfig', 'super_seeder', 3, 4))
def test_add_trackers(self):
"""
Testing whether trackers are added to the libtorrent handler in LibtorrentDownloadImpl
"""
def mocked_add_trackers(tracker_info):
self.assertIsInstance(tracker_info, dict)
self.assertEqual(tracker_info['url'], 'http://google.com')
mocked_add_trackers.called = True
mocked_add_trackers.called = False
self.libtorrent_download_impl.handle.add_tracker = mocked_add_trackers
self.libtorrent_download_impl.add_trackers(['http://google.com'])
self.assertTrue(mocked_add_trackers.called)
def test_process_error_alert(self):
"""
Testing whether error alerts are processed correctly
"""
url = "http://google.com"
mock_alert = MockObject()
mock_alert.msg = None
mock_alert.category = lambda: lt.alert.category_t.error_notification
mock_alert.status_code = 123
mock_alert.url = url
self.libtorrent_download_impl.process_alert(mock_alert,
'tracker_error_alert')
self.assertEqual(self.libtorrent_download_impl.tracker_status[url][1],
'HTTP status code 123')
mock_alert.status_code = 0
self.libtorrent_download_impl.process_alert(mock_alert,
'tracker_error_alert')
self.assertEqual(self.libtorrent_download_impl.tracker_status[url][1],
'Timeout')
def test_tracker_warning_alert(self):
"""
Test whether a tracking warning alert is processed correctly
"""
url = "http://google.com"
mock_alert = MockObject()
mock_alert.category = lambda: lt.alert.category_t.error_notification
mock_alert.url = url
mock_alert.message = lambda: 'test'
self.libtorrent_download_impl.process_alert(mock_alert,
'tracker_warning_alert')
self.assertEqual(self.libtorrent_download_impl.tracker_status[url][1],
'Warning: test')
@deferred(timeout=10)
def test_on_metadata_received_alert(self):
"""
Testing whether the right operations happen when we receive metadata
"""
test_deferred = Deferred()
def mocked_checkpoint():
test_deferred.callback(None)
self.libtorrent_download_impl.handle.trackers = lambda: []
self.libtorrent_download_impl.handle.save_resume_data = lambda: None
torrent_dict = {
'name': 'test',
'piece length': 42,
'pieces': '',
'files': []
}
get_info_from_handle(self.libtorrent_download_impl.handle
).metadata = lambda: lt.bencode(torrent_dict)
self.libtorrent_download_impl.checkpoint = mocked_checkpoint
self.libtorrent_download_impl.session = MockObject()
self.libtorrent_download_impl.session.lm = MockObject()
self.libtorrent_download_impl.session.lm.rtorrent_handler = None
self.libtorrent_download_impl.session.lm.torrent_db = None
self.libtorrent_download_impl.on_metadata_received_alert(None)
return test_deferred
def test_metadata_received_invalid_info(self):
"""
Testing whether the right operations happen when we receive metadata but the torrent info is invalid
"""
def mocked_checkpoint():
raise RuntimeError("This code should not be reached!")
self.libtorrent_download_impl.checkpoint = mocked_checkpoint
self.libtorrent_download_impl.handle.get_torrent_info = lambda: None
self.libtorrent_download_impl.on_metadata_received_alert(None)
def test_torrent_checked_alert(self):
"""
Testing whether the right operations happen after a torrent checked alert is received
"""
def mocked_pause_checkpoint():
mocked_pause_checkpoint.called = True
mocked_pause_checkpoint.called = False
self.libtorrent_download_impl.handle.pause = mocked_pause_checkpoint
self.libtorrent_download_impl.checkpoint = mocked_pause_checkpoint
mock_alert = MockObject()
mock_alert.category = lambda: lt.alert.category_t.error_notification
self.libtorrent_download_impl.pause_after_next_hashcheck = True
self.libtorrent_download_impl.process_alert(mock_alert,
'torrent_checked_alert')
self.assertFalse(
self.libtorrent_download_impl.pause_after_next_hashcheck)
self.assertTrue(mocked_pause_checkpoint.called)
mocked_pause_checkpoint.called = False
self.libtorrent_download_impl.checkpoint_after_next_hashcheck = True
self.libtorrent_download_impl.process_alert(mock_alert,
'torrent_checked_alert')
self.assertFalse(
self.libtorrent_download_impl.checkpoint_after_next_hashcheck)
self.assertTrue(mocked_pause_checkpoint.called)
def test_get_length(self):
"""
Testing whether the right length of the content of the download is returned
"""
self.libtorrent_download_impl.length = 1234
self.assertEqual(self.libtorrent_download_impl.get_length(), 1234)
def test_get_dest_files(self):
"""
Testing whether the right list of files is returned when fetching files from a download
"""
self.libtorrent_download_impl.handle.file_priority = lambda _: 123
mocked_file = MockObject()
mocked_file.path = 'test'
mock_torrent_info = MockObject()
mock_torrent_info.files = lambda: [mocked_file]
self.libtorrent_download_impl.handle.get_torrent_info = lambda: mock_torrent_info
dest_files = self.libtorrent_download_impl.get_dest_files()
self.assertIsInstance(dest_files[0], tuple)
self.assertEqual(dest_files[0][0], 'test')
def test_get_vod_fileindex(self):
"""
Testing whether the right vod file index is returned in LibtorrentDownloadImpl
"""
self.libtorrent_download_impl.vod_index = None
self.assertEqual(self.libtorrent_download_impl.get_vod_fileindex(), -1)
self.libtorrent_download_impl.vod_index = 42
self.assertEqual(self.libtorrent_download_impl.get_vod_fileindex(), 42)
def test_get_vod_filesize(self):
"""
Testing whether the right vod file size is returned in LibtorrentDownloadImpl
"""
mock_file_entry = MockObject()
mock_file_entry.size = 42
mock_torrent_info = MockObject()
mock_torrent_info.file_at = lambda _: mock_file_entry
self.libtorrent_download_impl.handle.get_torrent_info = lambda: mock_torrent_info
self.libtorrent_download_impl.vod_index = None
self.assertEqual(self.libtorrent_download_impl.get_vod_filesize(), 0)
self.libtorrent_download_impl.vod_index = 42
self.assertEqual(self.libtorrent_download_impl.get_vod_filesize(), 42)
def test_get_piece_progress(self):
"""
Testing whether the right piece progress is returned in LibtorrentDownloadImpl
"""
self.assertEqual(
self.libtorrent_download_impl.get_piece_progress(None), 1.0)
self.libtorrent_download_impl.handle.status().pieces = [True, False]
self.assertEqual(
self.libtorrent_download_impl.get_piece_progress([0, 1], True),
0.5)
self.libtorrent_download_impl.handle.status = lambda: None
self.assertEqual(
self.libtorrent_download_impl.get_piece_progress([3, 1]), 0.0)
def test_get_byte_progress(self):
"""
Testing whether the right byte progress is returned in LibtorrentDownloadImpl
"""
self.assertEqual(
self.libtorrent_download_impl.get_byte_progress([(-1, 0, 0)],
False), 1.0)
# Scenario: we have a file with 4 pieces, 250 bytes in each piece.
def map_file(_dummy1, start_byte, _dummy2):
res = MockObject()
res.piece = int(start_byte / 250)
return res
self.libtorrent_download_impl.handle.get_torrent_info(
).num_pieces = lambda: 4
self.libtorrent_download_impl.handle.get_torrent_info(
).map_file = map_file
self.assertEqual(
self.libtorrent_download_impl.get_byte_progress([(0, 10, 270)],
True), 0.5)
def test_setup_exception(self):
"""
Testing whether an exception in the setup method of LibtorrentDownloadImpl is handled correctly
"""
self.libtorrent_download_impl.setup()
self.assertIsInstance(self.libtorrent_download_impl.error, Exception)
def test_tracker_reply_alert(self):
"""
Testing the tracker reply alert in LibtorrentDownloadImpl
"""
mock_alert = MockObject()
mock_alert.url = 'http://google.com'
mock_alert.num_peers = 42
self.libtorrent_download_impl.on_tracker_reply_alert(mock_alert)
self.assertEqual(
self.libtorrent_download_impl.tracker_status['http://google.com'],
[42, 'Working'])
def test_stop(self):
"""
Testing whether the stop method in LibtorrentDownloadImpl invokes the correct method
"""
def mocked_stop_remove(removestate, removecontent):
self.assertFalse(removestate)
self.assertFalse(removecontent)
mocked_stop_remove.called = True
mocked_stop_remove.called = False
self.libtorrent_download_impl.stop_remove = mocked_stop_remove
self.libtorrent_download_impl.stop()
self.assertTrue(mocked_stop_remove.called)
def test_download_finish_alert(self):
"""
Testing whether the right operations are performed when we get a torrent finished alert
"""
status = self.libtorrent_download_impl.handle.status()
status.paused = False
status.state = DLSTATUS_DOWNLOADING
status.progress = 0.9
status.error = None
status.total_wanted = 33
status.download_payload_rate = 928
status.upload_payload_rate = 928
status.all_time_upload = 42
status.all_time_download = 43
status.finished_time = 1234
# Scenario: we have a file with 4 pieces, 250 bytes in each piece.
def map_file(_dummy1, start_byte, _dummy2):
res = MockObject()
res.piece = int(start_byte / 250)
return res
self.libtorrent_download_impl.handle.get_torrent_info(
).num_pieces = lambda: 4
self.libtorrent_download_impl.handle.get_torrent_info(
).map_file = map_file
self.libtorrent_download_impl.handle.piece_priorities = lambda: [
0, 0, 0, 0
]
self.libtorrent_download_impl.set_vod_mode(True)
self.libtorrent_download_impl.set_mode(DLMODE_VOD)
self.libtorrent_download_impl.on_torrent_finished_alert(None)
has_priorities_task = False
for task_name in self.libtorrent_download_impl._pending_tasks.iterkeys(
):
if 'reset_priorities' in task_name:
has_priorities_task = True
self.assertTrue(has_priorities_task)
def test_get_pieces_bitmask(self):
"""
Testing whether a correct pieces bitmask is returned when requested
"""
self.libtorrent_download_impl.handle.status().pieces = [
True, False, True, False, False
]
self.assertEqual(self.libtorrent_download_impl.get_pieces_base64(),
"oA==")
self.libtorrent_download_impl.handle.status().pieces = [True * 16]
self.assertEqual(self.libtorrent_download_impl.get_pieces_base64(),
"gA==")
@deferred(timeout=10)
def test_resume_data_failed(self):
"""
Testing whether the correct operations happen when an error is raised during resume data saving
"""
test_deferred = Deferred()
def on_error(_):
test_deferred.callback(None)
mock_alert = MockObject()
mock_alert.msg = "test error"
self.libtorrent_download_impl.deferreds_resume.append(
Deferred().addErrback(
self.libtorrent_download_impl._on_resume_err).addCallback(
on_error))
self.libtorrent_download_impl.on_save_resume_data_failed_alert(
mock_alert)
return test_deferred
class TestLibtorrentDownloadImplNoSession(TriblerCoreTest):
def setUp(self):
TriblerCoreTest.setUp(self)
self.libtorrent_download_impl = LibtorrentDownloadImpl(None, None)
mock_handle = MockObject()
mock_status = MockObject()
mock_status.pieces = [True, False, True, True, False]
torrent_info = MockObject()
file_info = MockObject()
file_info.size = 1234
torrent_info.file_at = lambda _: file_info
map_file_result = MockObject()
map_file_result.piece = 123
torrent_info.map_file = lambda _dummy1, _dummy2, _dummy3: map_file_result
torrent_info.num_pieces = lambda: 5
mock_handle.is_valid = lambda: True
mock_handle.status = lambda: mock_status
mock_handle.get_torrent_info = lambda: torrent_info
mock_handle.set_sequential_download = lambda _: None
mock_handle.set_priority = lambda _: None
mock_handle.prioritize_pieces = lambda _: None
self.libtorrent_download_impl.handle = mock_handle
# Create a fake tdef
self.libtorrent_download_impl.tdef = MockObject()
self.libtorrent_download_impl.tdef.get_name = lambda: "ubuntu.iso"
self.libtorrent_download_impl.tdef.is_multifile_torrent = lambda: False
def tearDown(self):
self.libtorrent_download_impl.shutdown_task_manager()
super(TestLibtorrentDownloadImplNoSession, self).tearDown()
def test_selected_files(self):
"""
Test whether the selected files are set correctly
"""
def mocked_set_file_prios(_):
mocked_set_file_prios.called = True
mocked_set_file_prios.called = False
mocked_file = MockObject()
mocked_file.path = 'my/path'
mock_torrent_info = MockObject()
mock_torrent_info.files = lambda: [mocked_file, mocked_file]
self.libtorrent_download_impl.handle.prioritize_files = mocked_set_file_prios
self.libtorrent_download_impl.handle.get_torrent_info = lambda: mock_torrent_info
self.libtorrent_download_impl.handle.rename_file = lambda *_: None
self.libtorrent_download_impl.get_share_mode = lambda: False
self.libtorrent_download_impl.tdef.get_infohash = lambda: 'a' * 20
self.libtorrent_download_impl.orig_files = ['a', 'b']
self.libtorrent_download_impl.get_save_path = lambda: 'my/path'
self.libtorrent_download_impl.set_selected_files(['a'])
self.assertTrue(mocked_set_file_prios.called)
self.libtorrent_download_impl.get_share_mode = lambda: False
mocked_set_file_prios.called = False
self.assertFalse(mocked_set_file_prios.called)
def test_get_share_mode(self):
"""
Test whether we return the right share mode when requested in the LibtorrentDownloadImpl
"""
self.libtorrent_download_impl.handle.status().share_mode = False
self.assertFalse(self.libtorrent_download_impl.get_share_mode())
self.libtorrent_download_impl.handle.status().share_mode = True
self.assertTrue(self.libtorrent_download_impl.get_share_mode())
def test_set_share_mode(self):
"""
Test whether we set the right share mode in LibtorrentDownloadImpl
"""
def mocked_set_share_mode(val):
self.assertTrue(val)
mocked_set_share_mode.called = True
mocked_set_share_mode.called = False
self.libtorrent_download_impl.handle.set_share_mode = mocked_set_share_mode
self.libtorrent_download_impl.set_share_mode(True)
self.assertTrue(mocked_set_share_mode.called)
def test_get_num_connected_seeds_peers(self):
"""
Test whether connected peers and seeds are correctly returned
"""
def get_peer_info(seeders, leechers):
peer_info = []
for _ in xrange(seeders):
seeder = MockObject()
seeder.flags = 140347 # some value where seed flag(1024) is true
seeder.seed = 1024
peer_info.append(seeder)
for _ in xrange(leechers):
leecher = MockObject()
leecher.flags = 131242 # some value where seed flag(1024) is false
leecher.seed = 1024
peer_info.append(leecher)
return peer_info
mock_seeders = 15
mock_leechers = 6
self.libtorrent_download_impl.handle.get_peer_info = lambda: get_peer_info(mock_seeders, mock_leechers)
num_seeds, num_peers = self.libtorrent_download_impl.get_num_connected_seeds_peers()
self.assertEqual(num_seeds, mock_seeders, "Expected seeders differ")
self.assertEqual(num_peers, mock_leechers, "Expected peers differ")
def test_set_priority(self):
"""
Test whether setting the priority calls the right methods in LibtorrentDownloadImpl
"""
def mocked_set_priority(prio):
self.assertEqual(prio, 1234)
mocked_set_priority.called = True
mocked_set_priority.called = False
self.libtorrent_download_impl.handle.set_priority = mocked_set_priority
self.libtorrent_download_impl.set_priority(1234)
self.assertTrue(mocked_set_priority.called)
def test_dlconfig_cb_change(self):
"""
Testing whether changing the configuration on runtime calls the right methods in LibtorrentDownloadImpl
"""
def mocked_set_upload_limit(prio):
self.assertEqual(prio, 3 * 1024)
mocked_set_upload_limit.called = True
mocked_set_upload_limit.called = False
self.libtorrent_download_impl.handle.set_upload_limit = mocked_set_upload_limit
def mocked_set_download_limit(prio):
self.assertEqual(prio, 3 * 1024)
mocked_set_download_limit.called = True
mocked_set_download_limit.called = False
self.libtorrent_download_impl.handle.set_download_limit = mocked_set_download_limit
self.libtorrent_download_impl.dlconfig_changed_callback('libtorrent', 'max_upload_rate', 3, 4)
self.assertTrue(mocked_set_upload_limit)
self.libtorrent_download_impl.dlconfig_changed_callback('libtorrent', 'max_download_rate', 3, 4)
self.assertTrue(mocked_set_download_limit)
self.assertFalse(self.libtorrent_download_impl.dlconfig_changed_callback(
'download_defaults', 'super_seeder', 3, 4))
def test_add_trackers(self):
"""
Testing whether trackers are added to the libtorrent handler in LibtorrentDownloadImpl
"""
def mocked_add_trackers(tracker_info):
self.assertIsInstance(tracker_info, dict)
self.assertEqual(tracker_info['url'], 'http://google.com')
mocked_add_trackers.called = True
mocked_add_trackers.called = False
self.libtorrent_download_impl.handle.add_tracker = mocked_add_trackers
self.libtorrent_download_impl.add_trackers(['http://google.com'])
self.assertTrue(mocked_add_trackers.called)
def test_process_error_alert(self):
"""
Testing whether error alerts are processed correctly
"""
url = "http://google.com"
mock_alert = MockObject()
mock_alert.msg = None
mock_alert.category = lambda: lt.alert.category_t.error_notification
mock_alert.status_code = 123
mock_alert.url = url
self.libtorrent_download_impl.process_alert(mock_alert, 'tracker_error_alert')
self.assertEqual(self.libtorrent_download_impl.tracker_status[url][1], 'HTTP status code 123')
mock_alert.status_code = 0
self.libtorrent_download_impl.process_alert(mock_alert, 'tracker_error_alert')
self.assertEqual(self.libtorrent_download_impl.tracker_status[url][1], 'Timeout')
def test_tracker_warning_alert(self):
"""
Test whether a tracking warning alert is processed correctly
"""
url = "http://google.com"
mock_alert = MockObject()
mock_alert.category = lambda: lt.alert.category_t.error_notification
mock_alert.url = url
mock_alert.message = lambda: 'test'
self.libtorrent_download_impl.process_alert(mock_alert, 'tracker_warning_alert')
self.assertEqual(self.libtorrent_download_impl.tracker_status[url][1], 'Warning: test')
@trial_timeout(10)
def test_on_metadata_received_alert(self):
"""
Testing whether the right operations happen when we receive metadata
"""
test_deferred = Deferred()
def mocked_checkpoint():
test_deferred.callback(None)
mocked_file = MockObject()
mocked_file.path = 'test'
self.libtorrent_download_impl.handle.trackers = lambda: []
self.libtorrent_download_impl.handle.save_resume_data = lambda: None
torrent_dict = {'name': 'test', 'piece length': 42, 'pieces': '', 'files': []}
get_info_from_handle(self.libtorrent_download_impl.handle).metadata = lambda: lt.bencode(torrent_dict)
get_info_from_handle(self.libtorrent_download_impl.handle).files = lambda: [mocked_file]
self.libtorrent_download_impl.checkpoint = mocked_checkpoint
self.libtorrent_download_impl.session = MockObject()
self.libtorrent_download_impl.session.lm = MockObject()
self.libtorrent_download_impl.session.lm.rtorrent_handler = None
self.libtorrent_download_impl.session.lm.torrent_db = None
self.libtorrent_download_impl.handle.save_path = lambda: None
self.libtorrent_download_impl.handle.prioritize_files = lambda _: None
self.libtorrent_download_impl.get_save_path = lambda: ''
self.libtorrent_download_impl.get_share_mode = lambda: False
self.libtorrent_download_impl.on_metadata_received_alert(None)
return test_deferred
def test_metadata_received_invalid_info(self):
"""
Testing whether the right operations happen when we receive metadata but the torrent info is invalid
"""
def mocked_checkpoint():
raise RuntimeError("This code should not be reached!")
self.libtorrent_download_impl.checkpoint = mocked_checkpoint
self.libtorrent_download_impl.handle.get_torrent_info = lambda: None
self.libtorrent_download_impl.on_metadata_received_alert(None)
def test_metadata_received_invalid_torrent_with_value_error(self):
"""
Testing whether the right operations happen when we receive metadata but the torrent info is invalid and throws
Value Error
"""
def mocked_checkpoint():
raise RuntimeError("This code should not be reached!")
mocked_file = MockObject()
mocked_file.path = 'test'
# The line below should trigger Value Error
self.libtorrent_download_impl.handle.trackers = lambda: [{'url': 'no-DHT'}]
torrent_dict = {'name': 'test', 'piece length': 42, 'pieces': '', 'files': []}
get_info_from_handle(self.libtorrent_download_impl.handle).metadata = lambda: lt.bencode(torrent_dict)
get_info_from_handle(self.libtorrent_download_impl.handle).files = lambda: [mocked_file]
self.libtorrent_download_impl.checkpoint = mocked_checkpoint
self.libtorrent_download_impl.on_metadata_received_alert(None)
def test_torrent_checked_alert(self):
"""
Testing whether the right operations happen after a torrent checked alert is received
"""
def mocked_pause_checkpoint():
mocked_pause_checkpoint.called = True
mocked_pause_checkpoint.called = False
self.libtorrent_download_impl.handle.pause = mocked_pause_checkpoint
self.libtorrent_download_impl.checkpoint = mocked_pause_checkpoint
mock_alert = MockObject()
mock_alert.category = lambda: lt.alert.category_t.error_notification
self.libtorrent_download_impl.pause_after_next_hashcheck = True
self.libtorrent_download_impl.process_alert(mock_alert, 'torrent_checked_alert')
self.assertFalse(self.libtorrent_download_impl.pause_after_next_hashcheck)
self.assertTrue(mocked_pause_checkpoint.called)
mocked_pause_checkpoint.called = False
self.libtorrent_download_impl.checkpoint_after_next_hashcheck = True
self.libtorrent_download_impl.process_alert(mock_alert, 'torrent_checked_alert')
self.assertFalse(self.libtorrent_download_impl.checkpoint_after_next_hashcheck)
self.assertTrue(mocked_pause_checkpoint.called)
def test_get_dest_files(self):
"""
Testing whether the right list of files is returned when fetching files from a download
"""
self.libtorrent_download_impl.handle.file_priority = lambda _: 123
mocked_file = MockObject()
mocked_file.path = 'test'
mock_torrent_info = MockObject()
mock_torrent_info.files = lambda: [mocked_file]
self.libtorrent_download_impl.handle.get_torrent_info = lambda: mock_torrent_info
dest_files = self.libtorrent_download_impl.get_dest_files()
self.assertIsInstance(dest_files[0], tuple)
self.assertEqual(dest_files[0][0], 'test')
def test_get_vod_fileindex(self):
"""
Testing whether the right vod file index is returned in LibtorrentDownloadImpl
"""
self.libtorrent_download_impl.vod_index = None
self.assertEqual(self.libtorrent_download_impl.get_vod_fileindex(), -1)
self.libtorrent_download_impl.vod_index = 42
self.assertEqual(self.libtorrent_download_impl.get_vod_fileindex(), 42)
def test_get_vod_filesize(self):
"""
Testing whether the right vod file size is returned in LibtorrentDownloadImpl
"""
mock_file_entry = MockObject()
mock_file_entry.size = 42
mock_torrent_info = MockObject()
mock_torrent_info.file_at = lambda _: mock_file_entry
self.libtorrent_download_impl.handle.get_torrent_info = lambda: mock_torrent_info
self.libtorrent_download_impl.vod_index = None
self.assertEqual(self.libtorrent_download_impl.get_vod_filesize(), 0)
self.libtorrent_download_impl.vod_index = 42
self.assertEqual(self.libtorrent_download_impl.get_vod_filesize(), 42)
def test_get_piece_progress(self):
"""
Testing whether the right piece progress is returned in LibtorrentDownloadImpl
"""
self.assertEqual(self.libtorrent_download_impl.get_piece_progress(None), 1.0)
self.libtorrent_download_impl.handle.status().pieces = [True, False]
self.assertEqual(self.libtorrent_download_impl.get_piece_progress([0, 1], True), 0.5)
self.libtorrent_download_impl.handle.status = lambda: None
self.assertEqual(self.libtorrent_download_impl.get_piece_progress([3, 1]), 0.0)
def test_get_byte_progress(self):
"""
Testing whether the right byte progress is returned in LibtorrentDownloadImpl
"""
self.assertEqual(self.libtorrent_download_impl.get_byte_progress([(-1, 0, 0)], False), 1.0)
# Scenario: we have a file with 4 pieces, 250 bytes in each piece.
def map_file(_dummy1, start_byte, _dummy2):
res = MockObject()
res.piece = int(start_byte / 250)
return res
self.libtorrent_download_impl.handle.get_torrent_info().num_pieces = lambda: 4
self.libtorrent_download_impl.handle.get_torrent_info().map_file = map_file
self.assertEqual(self.libtorrent_download_impl.get_byte_progress([(0, 10, 270)], True), 0.5)
def test_setup_exception(self):
"""
Testing whether an exception in the setup method of LibtorrentDownloadImpl is handled correctly
"""
self.libtorrent_download_impl.setup()
self.assertIsInstance(self.libtorrent_download_impl.error, Exception)
def test_tracker_reply_alert(self):
"""
Testing the tracker reply alert in LibtorrentDownloadImpl
"""
mock_alert = MockObject()
mock_alert.url = 'http://google.com'
mock_alert.num_peers = 42
self.libtorrent_download_impl.on_tracker_reply_alert(mock_alert)
self.assertEqual(self.libtorrent_download_impl.tracker_status['http://google.com'], [42, 'Working'])
def test_stop(self):
"""
Testing whether the stop method in LibtorrentDownloadImpl invokes the correct method
"""
def mocked_stop_remove(removestate, removecontent):
self.assertFalse(removestate)
self.assertFalse(removecontent)
mocked_stop_remove.called = True
mocked_stop_remove.called = False
self.libtorrent_download_impl.stop_remove = mocked_stop_remove
self.libtorrent_download_impl.stop()
self.assertTrue(mocked_stop_remove.called)
def test_download_finish_alert(self):
"""
Testing whether the right operations are performed when we get a torrent finished alert
"""
status = self.libtorrent_download_impl.handle.status()
status.paused = False
status.state = DLSTATUS_DOWNLOADING
status.progress = 0.9
status.error = None
status.total_wanted = 33
status.download_payload_rate = 928
status.upload_payload_rate = 928
status.all_time_upload = 42
status.all_time_download = 43
status.finished_time = 1234
# Scenario: we have a file with 4 pieces, 250 bytes in each piece.
def map_file(_dummy1, start_byte, _dummy2):
res = MockObject()
res.piece = int(start_byte / 250)
return res
self.libtorrent_download_impl.handle.get_torrent_info().num_pieces = lambda: 4
self.libtorrent_download_impl.handle.get_torrent_info().map_file = map_file
self.libtorrent_download_impl.handle.piece_priorities = lambda: [0, 0, 0, 0]
self.libtorrent_download_impl.set_vod_mode(True)
self.libtorrent_download_impl.set_mode(DLMODE_VOD)
self.libtorrent_download_impl.on_torrent_finished_alert(None)
has_priorities_task = False
for task_name in self.libtorrent_download_impl._pending_tasks.iterkeys():
if 'reset_priorities' in task_name:
has_priorities_task = True
self.assertTrue(has_priorities_task)
def test_get_pieces_bitmask(self):
"""
Testing whether a correct pieces bitmask is returned when requested
"""
self.libtorrent_download_impl.handle.status().pieces = [True, False, True, False, False]
self.assertEqual(self.libtorrent_download_impl.get_pieces_base64(), "oA==")
self.libtorrent_download_impl.handle.status().pieces = [True * 16]
self.assertEqual(self.libtorrent_download_impl.get_pieces_base64(), "gA==")
@trial_timeout(10)
def test_resume_data_failed(self):
"""
Testing whether the correct operations happen when an error is raised during resume data saving
"""
test_deferred = Deferred()
def on_error(_):
test_deferred.callback(None)
mock_alert = MockObject()
mock_alert.msg = "test error"
self.libtorrent_download_impl.deferreds_resume.append(Deferred().addErrback(
self.libtorrent_download_impl._on_resume_err).addCallback(on_error))
self.libtorrent_download_impl.on_save_resume_data_failed_alert(mock_alert)
return test_deferred
