logo

searx

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

test_archlinux.py (3629B)

    

  1. from collections import defaultdict

  2. import mock

  3. from searx.engines import archlinux

  4. from searx.testing import SearxTestCase

  5. 

  6. domains = {

  7.     'en': 'https://wiki.archlinux.org',

  8.     'de': 'https://wiki.archlinux.de',

  9.     'fr': 'https://wiki.archlinux.fr',

  10.     'ja': 'https://wiki.archlinuxjp.org',

  11.     'ro': 'http://wiki.archlinux.ro',

  12.     'tr': 'http://archtr.org/wiki'

  13. }

  14. 

  15. 

  16. class TestArchLinuxEngine(SearxTestCase):

  17. 

  18.     def test_request(self):

  19.         query = 'test_query'

  20.         dic = defaultdict(dict)

  21.         dic['pageno'] = 1

  22.         dic['language'] = 'en-US'

  23.         params = archlinux.request(query, dic)

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

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

  26.         self.assertTrue('wiki.archlinux.org' in params['url'])

  27. 

  28.         for lang, name in archlinux.main_langs:

  29.             dic['language'] = lang

  30.             params = archlinux.request(query, dic)

  31.             self.assertTrue(name in params['url'])

  32. 

  33.         for lang, domain in domains.items():

  34.             dic['language'] = lang

  35.             params = archlinux.request(query, dic)

  36.             self.assertTrue(domain in params['url'])

  37. 

  38.     def test_response(self):

  39.         response = mock.Mock(text='<html></html>',

  40.                              search_params={'language': 'en_US'})

  41.         self.assertEqual(archlinux.response(response), [])

  42. 

  43.         html = """

  44.         <ul class="mw-search-results">

  45.           <li>

  46.           <div class="mw-search-result-heading">

  47.             <a href="/index.php/ATI" title="ATI">ATI</a>

  48.           </div>

  49.           <div class="searchresult">

  50.             Lorem ipsum dolor sit amet

  51.           </div>

  52.           <div class="mw-search-result-data">

  53.             30 KB (4,630 words) - 19:04, 17 March 2016</div>

  54.           </li>

  55.           <li>

  56.           <div class="mw-search-result-heading">

  57.             <a href="/index.php/Frequently_asked_questions" title="Frequently asked questions">

  58.               Frequently asked questions

  59.             </a>

  60.           </div>

  61.           <div class="searchresult">

  62.             CPUs with AMDs instruction set "AMD64"

  63.           </div>

  64.           <div class="mw-search-result-data">

  65.             17 KB (2,722 words) - 20:13, 21 March 2016

  66.           </div>

  67.           </li>

  68.           <li>

  69.           <div class="mw-search-result-heading">

  70.             <a href="/index.php/CPU_frequency_scaling" title="CPU frequency scaling">CPU frequency scaling</a>

  71.           </div>

  72.           <div class="searchresult">

  73.             ondemand for AMD and older Intel CPU

  74.           </div>

  75.           <div class="mw-search-result-data">

  76.             15 KB (2,319 words) - 23:46, 16 March 2016

  77.           </div>

  78.           </li>

  79.         </ul>

  80.         """

  81. 

  82.         expected = [

  83.             {

  84.                 'title': 'ATI',

  85.                 'url': 'https://wiki.archlinux.org/index.php/ATI'

  86.             },

  87.             {

  88.                 'title': 'Frequently asked questions',

  89.                 'url': 'https://wiki.archlinux.org/index.php/Frequently_asked_questions'

  90.             },

  91.             {

  92.                 'title': 'CPU frequency scaling',

  93.                 'url': 'https://wiki.archlinux.org/index.php/CPU_frequency_scaling'

  94.             }

  95.         ]

  96. 

  97.         response = mock.Mock(text=html)

  98.         response.search_params = {

  99.             'language': 'en_US'

  100.         }

  101.         results = archlinux.response(response)

  102. 

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

  104.         self.assertEqual(len(results), len(expected))

  105. 

  106.         i = 0

  107.         for exp in expected:

  108.             res = results[i]

  109.             i += 1

  110.             for key, value in exp.items():

  111.                 self.assertEqual(res[key], value)