logo

searx

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

test_nyaa.py (5085B)

    

  1. from collections import defaultdict
  2. import mock
  3. from searx.engines import nyaa
  4. from searx.testing import SearxTestCase
  7. class TestNyaaEngine(SearxTestCase):
  9.     def test_request(self):
  10.         query = 'test_query'
  11.         dic = defaultdict(dict)
  12.         dic['pageno'] = 1
  13.         params = nyaa.request(query, dic)
  14.         self.assertTrue('url' in params)
  15.         self.assertTrue(query in params['url'])
  16.         self.assertTrue('nyaa.si' in params['url'])
  18.     def test_response(self):
  19.         resp = mock.Mock(text='<html></html>')
  20.         self.assertEqual(nyaa.response(resp), [])
  22.         html = """
  23.         <table class="table table-bordered table-hover table-striped torrent-list">
  24.         <thead>
  25.         <tr>
  26.         <th class="hdr-category text-center" style="width:80px;">
  27.         <div>Category</div>
  28.         </th>
  29.         <th class="hdr-name" style="width:auto;">
  30.         <div>Name</div>
  31.         </th>
  32.         <th class="hdr-comments sorting text-center" title="Comments" style="width:50px;">
  33.         <a href="/?f=0&amp;c=0_0&amp;q=Death+Parade&amp;s=comments&amp;o=desc"></a>
  34.         <i class="fa fa-comments-o"></i>
  35.         </th>
  36.         <th class="hdr-link text-center" style="width:70px;">
  37.         <div>Link</div>
  38.         </th>
  39.         <th class="hdr-size sorting text-center" style="width:100px;">
  40.         <a href="/?f=0&amp;c=0_0&amp;q=Death+Parade&amp;s=size&amp;o=desc"></a>
  41.         <div>Size</div>
  42.         </th>
  43.         <th class="hdr-date sorting_desc text-center" title="In local time" style="width:140px;">
  44.         <a href="/?f=0&amp;c=0_0&amp;q=Death+Parade&amp;s=id&amp;o=asc"></a>
  45.         <div>Date</div>
  46.         </th>
  47.         <th class="hdr-seeders sorting text-center" title="Seeders" style="width:50px;">
  48.         <a href="/?f=0&amp;c=0_0&amp;q=Death+Parade&amp;s=seeders&amp;o=desc"></a>
  49.         <i class="fa fa-arrow-up" aria-hidden="true"></i>
  50.         </th>
  51.         <th class="hdr-leechers sorting text-center" title="Leechers" style="width:50px;">
  52.         <a href="/?f=0&amp;c=0_0&amp;q=Death+Parade&amp;s=leechers&amp;o=desc"></a>
  53.         <i class="fa fa-arrow-down" aria-hidden="true"></i>
  54.         </th>
  55.         <th class="hdr-downloads sorting text-center" title="Completed downloads" style="width:50px;">
  56.         <a href="/?f=0&amp;c=0_0&amp;q=Death+Parade&amp;s=downloads&amp;o=desc"></a>
  57.         <i class="fa fa-check" aria-hidden="true"></i>
  58.         </th>
  59.         </tr>
  60.         </thead>
  61.         <tbody>
  62.         <tr class="default">
  63.         <td style="padding:0 4px;">
  64.         <a href="/?c=1_2" title="Anime - English-translated">
  65.         <img src="/static/img/icons/nyaa/1_2.png" alt="Anime - English-translated">
  66.         </a>
  67.         </td>
  68.         <td colspan="2">
  69.         <a href="/view/1" title="Sample title 1">Sample title 1</a>
  70.         </td>
  71.         <td class="text-center" style="white-space: nowrap;">
  72.         <a href="/download/1.torrent"><i class="fa fa-fw fa-download"></i></a>
  73.         <a href="magnet:?xt=urn:btih:2"><i class="fa fa-fw fa-magnet"></i></a>
  74.         </td>
  75.         <td class="text-center">723.7 MiB</td>
  76.         <td class="text-center" data-timestamp="1503307456" title="1 week 3
  77.         days 9 hours 44 minutes 39 seconds ago">2017-08-21 11:24</td>
  78.         <td class="text-center" style="color: green;">1</td>
  79.         <td class="text-center" style="color: red;">3</td>
  80.         <td class="text-center">12</td>
  81.         </tr>
  82.         <tr class="default">
  83.         <td style="padding:0 4px;">
  84.         <a href="/?c=1_2" title="Anime - English-translated">
  85.         <img src="/static/img/icons/nyaa/1_2.png" alt="Anime - English-translated">
  86.         </a>
  87.         </td>
  88.         <td colspan="2">
  89.         <a href="/view/2" title="Sample title 2">Sample title 2</a>
  90.         </td>
  91.         <td class="text-center" style="white-space: nowrap;">
  92.         <a href="magnet:?xt=urn:btih:2"><i class="fa fa-fw fa-magnet"></i></a>
  93.         </td>
  94.         <td class="text-center">8.2 GiB</td>
  95.         <td class="text-center" data-timestamp="1491608400" title="4 months 3
  96.         weeks 4 days 19 hours 28 minutes 55 seconds ago">2017-04-08 01:40</td>
  97.         <td class="text-center" style="color: green;">10</td>
  98.         <td class="text-center" style="color: red;">1</td>
  99.         <td class="text-center">206</td>
  100.         </tr>
  101.         </tbody>
  102.         </table>
  103.         """
  105.         resp = mock.Mock(text=html)
  106.         results = nyaa.response(resp)
  108.         self.assertEqual(type(results), list)
  109.         self.assertEqual(len(results), 2)
  111.         r = results[0]
  112.         self.assertTrue(r['url'].find('1') >= 0)
  113.         self.assertTrue(r['torrentfile'].find('1.torrent') >= 0)
  114.         self.assertTrue(r['content'].find('Anime - English-translated') >= 0)
  115.         self.assertTrue(r['content'].find('Downloaded 12 times.') >= 0)
  117.         self.assertEqual(r['title'], 'Sample title 1')
  118.         self.assertEqual(r['seed'], 1)
  119.         self.assertEqual(r['leech'], 3)
  120.         self.assertEqual(r['filesize'], 723700000)
  122.         r = results[1]
  123.         self.assertTrue(r['url'].find('2') >= 0)
  124.         self.assertTrue(r['magnetlink'].find('magnet:') >= 0)