Browse Source

Added unit test

rinpatch 7 years ago
parent
commit
dcc9fdb47f
2 changed files with 68 additions and 1 deletions
  1. 1 1
      searx/engines/acgsou.py
  2. 67 0
      tests/unit/engines/test_acgsou.py

+ 1 - 1
searx/engines/acgsou.py

@@ -63,7 +63,7 @@ def response(resp):
             filesize = get_torrent_size(filesize, filesize_multiplier)
         except:
             pass
-        # I didn't add download/seed/leech count since as I figured out they are generated randowmly everytime
+        # I didn't add download/seed/leech count since as I figured out they are generated randomly everytime
         content = 'Category: "{category}".'
         content = content.format(category=category)
 

+ 67 - 0
tests/unit/engines/test_acgsou.py

@@ -0,0 +1,67 @@
+from collections import defaultdict
+import mock
+from searx.engines import acgsou
+from searx.testing import SearxTestCase
+
+
+class TestAcgsouEngine(SearxTestCase):
+
+    def test_request(self):
+        query = 'test_query'
+        dic = defaultdict(dict)
+        dic['pageno'] = 1
+        params = acgsou.request(query, dic)
+        self.assertTrue('url' in params)
+        self.assertTrue(query in params['url'])
+        self.assertTrue('acgsou.com' in params['url'])
+
+    def test_response(self):
+        resp = mock.Mock(text='<html></html>')
+        self.assertEqual(acgsou.response(resp), [])
+
+        html = """
+<table id="listTable" class="list_style table_fixed">
+  <thead class="tcat">
+      <tr>
+      tablehead
+      </tr>
+  </thead>
+  <tbody class="tbody" id="data_list">
+              <tr class="alt1 ">
+        <td nowrap="nowrap">date</td>
+        <td><a href="category.html">testcategory</a></td>
+        <td style="text-align:left;">
+            <a href="show-torrentid.html" target="_blank">torrentname</a>
+        </td>
+        <td>1MB</td>
+        <td nowrap="nowrap">
+            <span class="bts_1">
+            29
+            </span>
+        </td>
+        <td nowrap="nowrap">
+            <span class="btl_1">
+            211
+        </span>
+        </td>
+        <td nowrap="nowrap">
+        <span class="btc_">
+            168
+        </span>
+        </td>
+        <td><a href="random.html">user</a></td>
+      </tr>
+</table>
+        """
+
+        resp = mock.Mock(text=html)
+        results = acgsou.response(resp)
+
+        self.assertEqual(type(results), list)
+        self.assertEqual(len(results), 1)
+
+        r = results[0]
+        self.assertEqual(r['url'], 'https://www.acgsou.com/show-torrentid.html')
+        self.assertEqual(r['content'], 'Category: "testcategory".')
+        self.assertEqual(r['title'], 'torrentname')
+        self.assertEqual(r['filesize'], 1048576)