61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
import os
|
|
import unittest
|
|
|
|
import requests
|
|
|
|
from ..yggtorrentscraper import YggTorrentScraper
|
|
|
|
|
|
class TestExtractDetails(unittest.TestCase):
|
|
scraper = YggTorrentScraper(requests.session())
|
|
|
|
def test_extract_details(self):
|
|
torrent = self.scraper.extract_details(
|
|
"https://www2.yggtorrent.pe/torrent/filmvideo/serie-tv/440445-game-of-thrones-s08e02-multi-1080p-amzn-web-dl-dd5-1-x264-ark01"
|
|
)
|
|
|
|
self.assertTrue(torrent.name is not None)
|
|
self.assertTrue(torrent.uploaded_datetime is not None)
|
|
self.assertTrue(torrent.size is not None)
|
|
self.assertTrue(torrent.uploader is not None)
|
|
|
|
self.assertTrue(len(torrent.keywords) > 0)
|
|
|
|
self.assertTrue(torrent.completed > -1)
|
|
self.assertTrue(torrent.seeders > -1)
|
|
self.assertTrue(torrent.leechers > -1)
|
|
|
|
self.assertTrue(torrent.url is None)
|
|
|
|
self.assertTrue(len(torrent.files) > 0)
|
|
self.assertTrue(len(torrent.comments) > 0)
|
|
|
|
def test_extract_details_logged(self):
|
|
yggtorrent_identifiant = os.environ.get("YGGTORRENT_IDENTIFIANT")
|
|
yggtorrent_password = os.environ.get("YGGTORRENT_PASSWORD")
|
|
|
|
self.scraper.login(yggtorrent_identifiant, yggtorrent_password)
|
|
|
|
torrent = self.scraper.extract_details(
|
|
"https://www2.yggtorrent.pe/torrent/filmvideo/serie-tv/440445-game-of-thrones-s08e02-multi-1080p-amzn-web-dl-dd5-1-x264-ark01"
|
|
)
|
|
|
|
self.assertTrue(torrent.name is not None)
|
|
self.assertTrue(torrent.uploaded_datetime is not None)
|
|
self.assertTrue(torrent.size is not None)
|
|
self.assertTrue(torrent.uploader is not None)
|
|
|
|
self.assertTrue(len(torrent.keywords) > 0)
|
|
|
|
self.assertTrue(torrent.completed > -1)
|
|
self.assertTrue(torrent.seeders > -1)
|
|
self.assertTrue(torrent.leechers > -1)
|
|
|
|
self.assertTrue(torrent.url is not None)
|
|
|
|
self.assertTrue(len(torrent.files) > 0)
|
|
self.assertTrue(len(torrent.comments) > 0)
|
|
|
|
def tearDown(self):
|
|
self.scraper.logout()
|