logo

searx

My custom branche(s) on searx, a meta-search engine git clone https://hacktivis.me/git/searx.git

test_piratebay.py (7312B)

    

  1. # -*- coding: utf-8 -*-

  2. from collections import defaultdict

  3. import mock

  4. from searx.engines import piratebay

  5. from searx.testing import SearxTestCase

  6. 

  7. 

  8. class TestPiratebayEngine(SearxTestCase):

  9. 

  10.     def test_request(self):

  11.         query = 'test_query'

  12.         dicto = defaultdict(dict)

  13.         dicto['pageno'] = 1

  14.         dicto['category'] = 'Toto'

  15.         params = piratebay.request(query, dicto)

  16.         self.assertIn('url', params)

  17.         self.assertIn(query, params['url'])

  18.         self.assertIn('piratebay.org', params['url'])

  19.         self.assertIn('0', params['url'])

  20. 

  21.         dicto['category'] = 'music'

  22.         params = piratebay.request(query, dicto)

  23.         self.assertIn('100', params['url'])

  24. 

  25.     def test_response(self):

  26.         self.assertRaises(AttributeError, piratebay.response, None)

  27.         self.assertRaises(AttributeError, piratebay.response, [])

  28.         self.assertRaises(AttributeError, piratebay.response, '')

  29.         self.assertRaises(AttributeError, piratebay.response, '[]')

  30. 

  31.         response = mock.Mock(text='<html></html>')

  32.         self.assertEqual(piratebay.response(response), [])

  33. 

  34.         html = """

  35.         <table id="searchResult">

  36.             <tr>

  37.             </tr>

  38.             <tr>

  39.                 <td class="vertTh">

  40.                     <center>

  41.                         <a href="#" title="More from this category">Anime</a><br/>

  42.                         (<a href="#" title="More from this category">Anime</a>)

  43.                     </center>

  44.                 </td>

  45.                 <td>

  46.                     <div class="detName">

  47.                         <a href="/this.is.the.link" class="detLink" title="Title">

  48.                             This is the title

  49.                         </a>

  50.                     </div>

  51.                     <a href="magnet:?xt=urn:btih:MAGNETLINK" title="Download this torrent using magnet">

  52.                         <img src="/static/img/icon-magnet.gif" alt="Magnet link"/>

  53.                     </a>

  54.                     <a href="http://torcache.net/torrent/TORRENTFILE.torrent" title="Download this torrent">

  55.                         <img src="/static/img/dl.gif" class="dl" alt="Download"/>

  56.                     </a>

  57.                     <a href="/user/HorribleSubs">

  58.                         <img src="/static/img/vip.gif" alt="VIP" title="VIP" style="width:11px;" border='0'/>

  59.                     </a>

  60.                     <img src="/static/img/11x11p.png"/>

  61.                     <font class="detDesc">

  62.                         This is the content <span>and should be</span> OK

  63.                     </font>

  64.                 </td>

  65.                 <td align="right">13</td>

  66.                 <td align="right">334</td>

  67.             </tr>

  68.             <tr>

  69.                 <td class="vertTh">

  70.                     <center>

  71.                         <a href="#" title="More from this category">Anime</a><br/>

  72.                         (<a href="#" title="More from this category">Anime</a>)

  73.                     </center>

  74.                 </td>

  75.                 <td>

  76.                     <div class="detName">

  77.                         <a href="/this.is.the.link" class="detLink" title="Title">

  78.                             This is the title

  79.                         </a>

  80.                     </div>

  81.                     <a href="magnet:?xt=urn:btih:MAGNETLINK" title="Download this torrent using magnet">

  82.                         <img src="/static/img/icon-magnet.gif" alt="Magnet link"/>

  83.                     </a>

  84.                     <a href="/user/HorribleSubs">

  85.                         <img src="/static/img/vip.gif" alt="VIP" title="VIP" style="width:11px;" border='0'/>

  86.                     </a>

  87.                     <img src="/static/img/11x11p.png"/>

  88.                     <font class="detDesc">

  89.                         This is the content <span>and should be</span> OK

  90.                     </font>

  91.                 </td>

  92.                 <td align="right">13</td>

  93.                 <td align="right">334</td>

  94.             </tr>

  95.         </table>

  96.         """

  97.         response = mock.Mock(text=html)

  98.         results = piratebay.response(response)

  99.         self.assertEqual(type(results), list)

  100.         self.assertEqual(len(results), 2)

  101.         self.assertEqual(results[0]['title'], 'This is the title')

  102.         self.assertEqual(results[0]['url'], 'https://thepiratebay.org/this.is.the.link')

  103.         self.assertEqual(results[0]['content'], 'This is the content and should be OK')

  104.         self.assertEqual(results[0]['seed'], 13)

  105.         self.assertEqual(results[0]['leech'], 334)

  106.         self.assertEqual(results[0]['magnetlink'], 'magnet:?xt=urn:btih:MAGNETLINK')

  107.         self.assertEqual(results[0]['torrentfile'], 'http://torcache.net/torrent/TORRENTFILE.torrent')

  108. 

  109.         self.assertEqual(results[1]['torrentfile'], None)

  110. 

  111.         html = """

  112.         <table id="searchResult">

  113.             <tr>

  114.             </tr>

  115.             <tr>

  116.                 <td class="vertTh">

  117.                     <center>

  118.                         <a href="#" title="More from this category">Anime</a><br/>

  119.                         (<a href="#" title="More from this category">Anime</a>)

  120.                     </center>

  121.                 </td>

  122.                 <td>

  123.                     <div class="detName">

  124.                         <a href="/this.is.the.link" class="detLink" title="Title">

  125.                             This is the title

  126.                         </a>

  127.                     </div>

  128.                     <a href="magnet:?xt=urn:btih:MAGNETLINK" title="Download this torrent using magnet">

  129.                         <img src="/static/img/icon-magnet.gif" alt="Magnet link"/>

  130.                     </a>

  131.                     <a href="http://torcache.net/torrent/TORRENTFILE.torrent" title="Download this torrent">

  132.                         <img src="/static/img/dl.gif" class="dl" alt="Download"/>

  133.                     </a>

  134.                     <a href="/user/HorribleSubs">

  135.                         <img src="/static/img/vip.gif" alt="VIP" title="VIP" style="width:11px;" border='0'/>

  136.                     </a>

  137.                     <img src="/static/img/11x11p.png"/>

  138.                     <font class="detDesc">

  139.                         This is the content <span>and should be</span> OK

  140.                     </font>

  141.                 </td>

  142.                 <td align="right">s</td>

  143.                 <td align="right">d</td>

  144.             </tr>

  145.         </table>

  146.         """

  147.         response = mock.Mock(text=html)

  148.         results = piratebay.response(response)

  149.         self.assertEqual(type(results), list)

  150.         self.assertEqual(len(results), 1)

  151.         self.assertEqual(results[0]['title'], 'This is the title')

  152.         self.assertEqual(results[0]['url'], 'https://thepiratebay.org/this.is.the.link')

  153.         self.assertEqual(results[0]['content'], 'This is the content and should be OK')

  154.         self.assertEqual(results[0]['seed'], 0)

  155.         self.assertEqual(results[0]['leech'], 0)

  156.         self.assertEqual(results[0]['magnetlink'], 'magnet:?xt=urn:btih:MAGNETLINK')

  157.         self.assertEqual(results[0]['torrentfile'], 'http://torcache.net/torrent/TORRENTFILE.torrent')

  158. 

  159.         html = """

  160.         <table id="searchResult">

  161.         </table>

  162.         """

  163.         response = mock.Mock(text=html)

  164.         results = piratebay.response(response)

  165.         self.assertEqual(type(results), list)

  166.         self.assertEqual(len(results), 0)