logo

searx

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

test_acgsou.py (2468B)

    

  1. # coding=utf-8

  2. from collections import defaultdict

  3. import mock

  4. from searx.engines import acgsou

  5. from searx.testing import SearxTestCase

  6. 

  7. 

  8. class TestAcgsouEngine(SearxTestCase):

  9. 

  10.     def test_request(self):

  11.         query = 'test_query'

  12.         dic = defaultdict(dict)

  13.         dic['pageno'] = 1

  14.         params = acgsou.request(query, dic)

  15.         self.assertTrue('url' in params)

  16.         self.assertTrue(query in params['url'])

  17.         self.assertTrue('acgsou.com' in params['url'])

  18. 

  19.     def test_response(self):

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

  21.         self.assertEqual(acgsou.response(resp), [])

  22. 

  23.         html = u"""

  24.         <html>

  25. <table id="listTable" class="list_style table_fixed">

  26.   <thead class="tcat">

  27.       <tr>

  28.         <th axis="string" class="l1 tableHeaderOver">test</th>

  29.         <th axis="string" class="l2 tableHeaderOver">test</th>

  30.         <th axis="string" class="l3 tableHeaderOver">test</th>

  31.         <th axis="size" class="l4 tableHeaderOver">test</th>

  32.         <th axis="number" class="l5 tableHeaderOver">test</th>

  33.         <th axis="number" class="l6 tableHeaderOver">test</th>

  34.         <th axis="number" class="l7 tableHeaderOver">test</th>

  35.         <th axis="string" class="l8 tableHeaderOver">test</th>

  36.       </tr>

  37.   </thead>

  38.   <tbody class="tbody" id="data_list">

  39.  <tr class="alt1 ">

  40.         <td nowrap="nowrap">date</td>

  41.         <td><a href="category.html">testcategory テスト</a></td>

  42.         <td style="text-align:left;">

  43.             <a href="show-torrentid.html" target="_blank">torrentname テスト</a>

  44.         </td>

  45.         <td>1MB</td>

  46.         <td nowrap="nowrap">

  47.             <span class="bts_1">

  48.             29

  49.             </span>

  50.         </td>

  51.         <td nowrap="nowrap">

  52.             <span class="btl_1">

  53.             211

  54.         </span>

  55.         </td>

  56.         <td nowrap="nowrap">

  57.         <span class="btc_">

  58.             168

  59.         </span>

  60.         </td>

  61.         <td><a href="random.html">user</a></td>

  62.       </tr>

  63.       </tbody>

  64. </table>

  65. </html>

  66.         """

  67. 

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

  69.         results = acgsou.response(resp)

  70. 

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

  72.         self.assertEqual(len(results), 1)

  73. 

  74.         r = results[0]

  75.         self.assertEqual(r['url'], 'http://www.acgsou.com/show-torrentid.html')

  76.         self.assertEqual(r['content'], u'Category: "testcategory テスト".')

  77.         self.assertEqual(r['title'], u'torrentname テスト')

  78.         self.assertEqual(r['filesize'], 1048576)