| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566 |
- from collections import defaultdict
- import mock
- from searx.engines import nyaa
- from searx.testing import SearxTestCase
- class TestNyaaEngine(SearxTestCase):
- def test_request(self):
- query = 'test_query'
- dic = defaultdict(dict)
- dic['pageno'] = 1
- params = nyaa.request(query, dic)
- self.assertTrue('url' in params)
- self.assertTrue(query in params['url'])
- self.assertTrue('nyaa.se' in params['url'])
- def test_response(self):
- resp = mock.Mock(text='<html></html>')
- self.assertEqual(nyaa.response(resp), [])
- html = """
- <table class="tlist">
- <tbody>
- <tr class="trusted tlistrow">
- <td class="tlisticon">
- <a href="//www.nyaa.se" title="English-translated Anime">
- <img src="//files.nyaa.se" alt="English-translated Anime">
- </a>
- </td>
- <td class="tlistname">
- <a href="//www.nyaa.se/?page3">
- Sample torrent title
- </a>
- </td>
- <td class="tlistdownload">
- <a href="//www.nyaa.se/?page_dl" title="Download">
- <img src="//files.nyaa.se/www-dl.png" alt="DL">
- </a>
- </td>
- <td class="tlistsize">10 MiB</td>
- <td class="tlistsn">1</td>
- <td class="tlistln">3</td>
- <td class="tlistdn">666</td>
- <td class="tlistmn">0</td>
- </tr>
- </tbody>
- </table>
- """
- resp = mock.Mock(text=html)
- results = nyaa.response(resp)
- self.assertEqual(type(results), list)
- self.assertEqual(len(results), 1)
- r = results[0]
- self.assertTrue(r['url'].find('www.nyaa.se/?page3') >= 0)
- self.assertTrue(r['torrentfile'].find('www.nyaa.se/?page_dl') >= 0)
- self.assertTrue(r['content'].find('English-translated Anime') >= 0)
- self.assertTrue(r['content'].find('Downloaded 666 times.') >= 0)
- self.assertEqual(r['title'], 'Sample torrent title')
- self.assertEqual(r['seed'], 1)
- self.assertEqual(r['leech'], 3)
- self.assertEqual(r['filesize'], 10 * 1024 * 1024)
|
