diff options
Diffstat (limited to 'tests/unit/engines/test_piratebay.py')
| -rw-r--r-- | tests/unit/engines/test_piratebay.py | 166 |
1 files changed, 0 insertions, 166 deletions
diff --git a/tests/unit/engines/test_piratebay.py b/tests/unit/engines/test_piratebay.py deleted file mode 100644 index 89a78e796..000000000 --- a/tests/unit/engines/test_piratebay.py +++ /dev/null @@ -1,166 +0,0 @@ -# -*- 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.org', 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.org/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.org/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) |