forked from Ponysearch/Ponysearch
67 lines
2.2 KiB
Python
67 lines
2.2 KiB
Python
|
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)
|