def test_add_torrent_ret_false(self): handler = TorrentHandler(self.log) torrent_url = "http://url.com/file.torrent" cookies_dict = {} global test_component test_component.download_success = False handler.download_torrent_file = test_component.download_torrent_file torrent_info = {"link": torrent_url, "site_cookies_dict": cookies_dict} torrent_download = handler.add_torrent(torrent_info) self.assertFalse(torrent_download.success) # Set by download_torrent_file self.assertEquals(torrent_download.torrent_url, torrent_url) self.assertEquals(torrent_download.cookies_dict, cookies_dict) test_component.download_success = True
def test_add_torrent_ret_false(self): handler = TorrentHandler(self.log) torrent_url = "http://url.com/file.torrent" cookies = {} global test_component test_component.download_success = False handler.download_torrent_file = test_component.download_torrent_file torrent_info = {"link": torrent_url, "site_cookies_dict": cookies} torrent_download = handler.add_torrent(torrent_info) self.assertFalse(torrent_download.success) # Set by download_torrent_file self.assertEquals(torrent_download.torrent_url, torrent_url) self.assertEquals(torrent_download.cookies, cookies) test_component.download_success = True
def test_get_torrent(self): handler = TorrentHandler(self.log) handler.download_torrent_file = test_component.download_torrent_file filename = yarss2.util.common.get_resource( "FreeBSD-9.0-RELEASE-amd64-dvd1.torrent", path="tests/data/") test_component.use_filedump = read_file(filename) torrent_info = { "link": "http://url.com/file.torrent", "site_cookies_dict": { "cookiekey": "cookievalue" }, "user_agent": "test" } download = handler.get_torrent(torrent_info) self.assertEquals(download.headers, {'User-Agent': 'test'}) self.assertEquals(download.cookies, {'cookiekey': 'cookievalue'}) self.assertFalse(download.is_magnet)
def test_add_torrents(self): handler = TorrentHandler(self.log) from yarss2.rssfeed_handling import RSSFeedHandler self.rssfeedhandler = RSSFeedHandler(self.log) # Override method download_torrent_file handler.download_torrent_file = test_component.download_torrent_file filename = yarss2.util.common.get_resource("FreeBSD-9.0-RELEASE-amd64-dvd1.torrent", path="tests/data/") test_component.use_filedump = read_file(filename) config = get_test_config_dict() # 0 is the rssfeed key match_result = self.rssfeedhandler.fetch_feed_torrents(config, "0") matching_torrents = match_result["matching_torrents"] saved_subscriptions = [] def save_subscription_func(subscription_data): saved_subscriptions.append(subscription_data) handler.add_torrents(save_subscription_func, matching_torrents, self.config.get_config()) self.assertEquals(len(saved_subscriptions), 1) handler.use_filedump = None
def test_add_torrents(self): handler = TorrentHandler(self.log) from yarss2.rssfeed_handling import RSSFeedHandler self.rssfeedhandler = RSSFeedHandler(self.log) # Override method download_torrent_file handler.download_torrent_file = test_component.download_torrent_file filename = yarss2.util.common.get_resource( "FreeBSD-9.0-RELEASE-amd64-dvd1.torrent", path="tests/data/") test_component.use_filedump = read_file(filename) config = get_test_config_dict() # 0 is the rssfeed key match_result = self.rssfeedhandler.fetch_feed_torrents(config, "0") matching_torrents = match_result["matching_torrents"] saved_subscriptions = [] def save_subscription_func(subscription_data): saved_subscriptions.append(subscription_data) handler.add_torrents(save_subscription_func, matching_torrents, self.config.get_config()) self.assertEquals(len(saved_subscriptions), 3) handler.use_filedump = None