astroport-iptubes/yggcrawl/tests/test_torrent.py

33 lines
850 B
Python

import os
import unittest
import requests
from ..yggtorrentscraper import YggTorrentScraper
class TestTorrent(unittest.TestCase):
scraper = YggTorrentScraper(requests.session())
def test_str(self):
torrent_url = self.scraper.most_completed()[0]
torrent = self.scraper.extract_details(torrent_url)
torrent.__str__(files=True, comments=True)
def test_str_logged(self):
yggtorrent_identifiant = os.environ.get("YGGTORRENT_IDENTIFIANT")
yggtorrent_password = os.environ.get("YGGTORRENT_PASSWORD")
self.scraper.login(yggtorrent_identifiant, yggtorrent_password)
torrent_url = self.scraper.most_completed()[0]
torrent = self.scraper.extract_details(torrent_url)
torrent.__str__(files=True, comments=True)
def tearDown(self):
self.scraper.logout()