logo

searx

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

test_digbt.py (2481B)

    

  1. from collections import defaultdict

  2. import mock

  3. from searx.engines import digbt

  4. from searx.testing import SearxTestCase

  5. 

  6. 

  7. class TestDigBTEngine(SearxTestCase):

  8. 

  9.     def test_request(self):

  10.         query = 'test_query'

  11.         dicto = defaultdict(dict)

  12.         dicto['pageno'] = 0

  13.         params = digbt.request(query, dicto)

  14.         self.assertIn('url', params)

  15.         self.assertIn(query, params['url'])

  16.         self.assertIn('digbt.org', params['url'])

  17. 

  18.     def test_response(self):

  19.         self.assertRaises(AttributeError, digbt.response, None)

  20.         self.assertRaises(AttributeError, digbt.response, [])

  21.         self.assertRaises(AttributeError, digbt.response, '')

  22.         self.assertRaises(AttributeError, digbt.response, '[]')

  23. 

  24.         response = mock.Mock(text='<html></html>')

  25.         self.assertEqual(digbt.response(response), [])

  26. 

  27.         html = """

  28.         <table class="table">

  29.             <tr><td class="x-item">

  30.             <div>

  31.                 <a title="The Big Bang Theory" class="title" href="/The-Big-Bang-Theory-d2.html">

  32.                     The Big <span class="highlight">Bang</span> Theory

  33.                 </a>

  34.                 <span class="ctime"><span style="color:red;">4 hours ago</span></span>

  35.             </div>

  36.             <div class="files">

  37.                 <ul>

  38.                     <li>The Big Bang Theory  2.9 GB</li>

  39.                     <li>....</li>

  40.                 </ul>

  41.             </div>

  42.             <div class="tail">

  43.                 Files: 1 Size: 2.9 GB  Downloads: 1 Updated: <span style="color:red;">4 hours ago</span>

  44.                 &nbsp; &nbsp;

  45.                 <a class="title" href="magnet:?xt=urn:btih:a&amp;dn=The+Big+Bang+Theory">

  46.                     <span class="glyphicon glyphicon-magnet"></span> magnet-link

  47.                 </a>

  48.                 &nbsp; &nbsp;

  49.             </div>

  50.             </td></tr>

  51.         </table>

  52.         """

  53.         response = mock.Mock(text=html.encode('utf-8'))

  54.         results = digbt.response(response)

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

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

  57.         self.assertEqual(results[0]['title'], 'The Big Bang Theory')

  58.         self.assertEqual(results[0]['url'], 'https://digbt.org/The-Big-Bang-Theory-d2.html')

  59.         self.assertEqual(results[0]['content'], 'The Big Bang Theory 2.9 GB ....')

  60.         self.assertEqual(results[0]['filesize'], 3113851289)

  61.         self.assertEqual(results[0]['magnetlink'], 'magnet:?xt=urn:btih:a&dn=The+Big+Bang+Theory')