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

  5. 

  6. 

  7. class TestNyaaEngine(SearxTestCase):

  8. 

  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'])

  17. 

  18.     def test_response(self):

  19.         resp = mock.Mock(text='<html></html>')

  20.         self.assertEqual(nyaa.response(resp), [])

  21. 

  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.         """

  104. 

  105.         resp = mock.Mock(text=html)

  106.         results = nyaa.response(resp)

  107. 

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

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

  110. 

  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)

  116. 

  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)

  121. 

  122.         r = results[1]

  123.         self.assertTrue(r['url'].find('2') >= 0)

  124.         self.assertTrue(r['magnetlink'].find('magnet:') >= 0)