summaryrefslogtreecommitdiff
path: root/tests/unit/engines/test_nyaa.py
blob: db412e1cca0683005a476d7ab7fcf5b6828d22af (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
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)