# -*- coding: utf-8 -*- from collections import defaultdict import mock from searx.engines import piratebay from searx.testing import SearxTestCase class TestPiratebayEngine(SearxTestCase): def test_request(self): query = 'test_query' dicto = defaultdict(dict) dicto['pageno'] = 1 dicto['category'] = 'Toto' params = piratebay.request(query, dicto) self.assertIn('url', params) self.assertIn(query, params['url']) self.assertIn('piratebay.se', params['url']) self.assertIn('0', params['url']) dicto['category'] = 'music' params = piratebay.request(query, dicto) self.assertIn('100', params['url']) def test_response(self): self.assertRaises(AttributeError, piratebay.response, None) self.assertRaises(AttributeError, piratebay.response, []) self.assertRaises(AttributeError, piratebay.response, '') self.assertRaises(AttributeError, piratebay.response, '[]') response = mock.Mock(text='<html></html>') self.assertEqual(piratebay.response(response), []) html = """ <table id="searchResult"> <tr> </tr> <tr> <td class="vertTh"> <center> <a href="#" title="More from this category">Anime</a><br/> (<a href="#" title="More from this category">Anime</a>) </center> </td> <td> <div class="detName"> <a href="/this.is.the.link" class="detLink" title="Title"> This is the title </a> </div> <a href="magnet:?xt=urn:btih:MAGNETLINK" title="Download this torrent using magnet"> <img src="/static/img/icon-magnet.gif" alt="Magnet link"/> </a> <a href="http://torcache.net/torrent/TORRENTFILE.torrent" title="Download this torrent"> <img src="/static/img/dl.gif" class="dl" alt="Download"/> </a> <a href="/user/HorribleSubs"> <img src="/static/img/vip.gif" alt="VIP" title="VIP" style="width:11px;" border='0'/> </a> <img src="/static/img/11x11p.png"/> <font class="detDesc"> This is the content <span>and should be</span> OK </font> </td> <td align="right">13</td> <td align="right">334</td> </tr> <tr> <td class="vertTh"> <center> <a href="#" title="More from this category">Anime</a><br/> (<a href="#" title="More from this category">Anime</a>) </center> </td> <td> <div class="detName"> <a href="/this.is.the.link" class="detLink" title="Title"> This is the title </a> </div> <a href="magnet:?xt=urn:btih:MAGNETLINK" title="Download this torrent using magnet"> <img src="/static/img/icon-magnet.gif" alt="Magnet link"/> </a> <a href="/user/HorribleSubs"> <img src="/static/img/vip.gif" alt="VIP" title="VIP" style="width:11px;" border='0'/> </a> <img src="/static/img/11x11p.png"/> <font class="detDesc"> This is the content <span>and should be</span> OK </font> </td> <td align="right">13</td> <td align="right">334</td> </tr> </table> """ response = mock.Mock(text=html) results = piratebay.response(response) self.assertEqual(type(results), list) self.assertEqual(len(results), 2) self.assertEqual(results[0]['title'], 'This is the title') self.assertEqual(results[0]['url'], 'https://thepiratebay.se/this.is.the.link') self.assertEqual(results[0]['content'], 'This is the content and should be OK') self.assertEqual(results[0]['seed'], 13) self.assertEqual(results[0]['leech'], 334) self.assertEqual(results[0]['magnetlink'], 'magnet:?xt=urn:btih:MAGNETLINK') self.assertEqual(results[0]['torrentfile'], 'http://torcache.net/torrent/TORRENTFILE.torrent') self.assertEqual(results[1]['torrentfile'], None) html = """ <table id="searchResult"> <tr> </tr> <tr> <td class="vertTh"> <center> <a href="#" title="More from this category">Anime</a><br/> (<a href="#" title="More from this category">Anime</a>) </center> </td> <td> <div class="detName"> <a href="/this.is.the.link" class="detLink" title="Title"> This is the title </a> </div> <a href="magnet:?xt=urn:btih:MAGNETLINK" title="Download this torrent using magnet"> <img src="/static/img/icon-magnet.gif" alt="Magnet link"/> </a> <a href="http://torcache.net/torrent/TORRENTFILE.torrent" title="Download this torrent"> <img src="/static/img/dl.gif" class="dl" alt="Download"/> </a> <a href="/user/HorribleSubs"> <img src="/static/img/vip.gif" alt="VIP" title="VIP" style="width:11px;" border='0'/> </a> <img src="/static/img/11x11p.png"/> <font class="detDesc"> This is the content <span>and should be</span> OK </font> </td> <td align="right">s</td> <td align="right">d</td> </tr> </table> """ response = mock.Mock(text=html) results = piratebay.response(response) self.assertEqual(type(results), list) self.assertEqual(len(results), 1) self.assertEqual(results[0]['title'], 'This is the title') self.assertEqual(results[0]['url'], 'https://thepiratebay.se/this.is.the.link') self.assertEqual(results[0]['content'], 'This is the content and should be OK') self.assertEqual(results[0]['seed'], 0) self.assertEqual(results[0]['leech'], 0) self.assertEqual(results[0]['magnetlink'], 'magnet:?xt=urn:btih:MAGNETLINK') self.assertEqual(results[0]['torrentfile'], 'http://torcache.net/torrent/TORRENTFILE.torrent') html = """ <table id="searchResult"> </table> """ response = mock.Mock(text=html) results = piratebay.response(response) self.assertEqual(type(results), list) self.assertEqual(len(results), 0)