def test_multifile_torrent(self): tdef = TorrentDef() dn = os.path.join(TESTS_DATA_DIR, "contentdir") tdef.add_content(dn, "dirintorrent") fn = os.path.join(TESTS_DATA_DIR, "video.avi") tdef.add_content(fn, os.path.join("dirintorrent", "video.avi")) tdef.set_tracker("http://tribler.org/announce") tdef.finalize() impl = LibtorrentDownloadImpl(self.session, tdef) # Override the add_torrent because it will be called impl.ltmgr = MockObject() impl.ltmgr.add_torrent = lambda _, _dummy2: fake_handler impl.set_selected_files = lambda: None fake_handler = MockObject() fake_handler.is_valid = lambda: True fake_handler.status = lambda: fake_status fake_handler.set_share_mode = lambda _: None fake_handler.resume = lambda: None fake_handler.resolve_countries = lambda _: None fake_status = MockObject() fake_status.share_mode = False # Create a dummy download config impl.dlconfig = DownloadStartupConfig().dlconfig.copy() # Create a dummy pstate pstate = CallbackConfigParser() pstate.add_section("state") test_dict = dict() test_dict["a"] = "b" pstate.set("state", "engineresumedata", test_dict) return impl.network_create_engine_wrapper(pstate)
def test_restart_in_upload_mode(self): tdef = self.create_tdef() def mock_set_upload_mode(handle, mode): handle.status().upload_mode = mode fake_status = MockObject() fake_status.share_mode = False fake_status.upload_mode = False impl = LibtorrentDownloadImpl(self.session, tdef) impl.handle = MockObject() impl.handle.status = lambda: fake_status # impl.handle.upload_mode = False impl.handle.set_priority = lambda _: None impl.handle.set_sequential_download = lambda _: None impl.handle.set_upload_mode = lambda mode, _handle=impl.handle: mock_set_upload_mode(_handle, mode) impl.handle.get_upload_mode = lambda: impl.handle.status().upload_mode impl.handle.is_valid = lambda: True impl.handle.resume = lambda: None # Create a dummy download config impl.dlconfig = DownloadStartupConfig().dlconfig.copy() impl.session.lm.on_download_wrapper_created = lambda _: True impl.set_upload_mode(True) impl.restart() self.assertTrue(impl.get_upload_mode())
def test_multifile_torrent(self): tdef = TorrentDef() tdef.add_content(os.path.join(TESTS_DATA_DIR, "video.avi")) tdef.set_tracker("http://tribler.org/announce") tdef.save() impl = LibtorrentDownloadImpl(self.session, tdef) # Override the add_torrent because it will be called impl.ltmgr = MockObject() impl.ltmgr.add_torrent = lambda _, _dummy2: succeed(fake_handler) impl.set_selected_files = lambda: None fake_handler = MockObject() fake_handler.is_valid = lambda: True fake_handler.status = lambda: fake_status fake_handler.set_share_mode = lambda _: None fake_handler.set_priority = lambda _: None fake_handler.set_sequential_download = lambda _: None fake_handler.resume = lambda: None fake_handler.set_max_connections = lambda _: None fake_status = MockObject() fake_status.share_mode = False # Create a dummy download config impl.dlconfig = DownloadStartupConfig().dlconfig.copy() # Create a dummy pstate pstate = CallbackConfigParser() pstate.add_section("state") test_dict = dict() test_dict["a"] = "b" pstate.set("state", "engineresumedata", test_dict) return impl.network_create_engine_wrapper(pstate)
def test_restart(self): tdef = self.create_tdef() impl = LibtorrentDownloadImpl(self.session, tdef) impl.handle = MockObject() impl.handle.set_priority = lambda _: None impl.handle.set_sequential_download = lambda _: None impl.handle.resume = lambda: None impl.handle.status = lambda: fake_status fake_status = MockObject() fake_status.share_mode = False # Create a dummy download config impl.dlconfig = DownloadStartupConfig().dlconfig.copy() impl.session.lm.on_download_wrapper_created = lambda _: True impl.restart()
def test_restart(self): tdef = self.create_tdef() impl = LibtorrentDownloadImpl(self.session, tdef) impl.handle = MockObject() impl.handle.set_priority = lambda _: None impl.handle.set_sequential_download = lambda _: None impl.handle.resume = lambda: None impl.handle.status = lambda: fake_status fake_status = MockObject() fake_status.share_mode = False # Create a dummy download config impl.dlconfig = DownloadStartupConfig().dlconfig.copy() impl.session.lm.on_download_wrapper_created = lambda _: True impl.restart()