|
|
@@ -0,0 +1,137 @@
|
|
|
+# -*- 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.cr', 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>
|
|
|
+ </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.cr/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')
|
|
|
+
|
|
|
+ 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.cr/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)
|
Cqoicebordel