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
  8. class TestPiratebayEngine(SearxTestCase):
  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'])
  21.         dicto['category'] = 'music'
  22.         params = piratebay.request(query, dicto)
  23.         self.assertIn('100', params['url'])
  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, '[]')
  31.         response = mock.Mock(text='<html></html>')
  32.         self.assertEqual(piratebay.response(response), [])
  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')
  109.         self.assertEqual(results[1]['torrentfile'], None)
  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')
  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)