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
  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. }
  16. class TestArchLinuxEngine(SearxTestCase):
  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'])
  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'])
  33.         for lang, domain in domains.items():
  34.             dic['language'] = lang
  35.             params = archlinux.request(query, dic)
  36.             self.assertTrue(domain in params['url'])
  38.     def test_response(self):
  39.         response = mock.Mock(text='<html></html>',
  40.                              search_params={'language': 'en_US'})
  41.         self.assertEqual(archlinux.response(response), [])
  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.         """
  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.         ]
  97.         response = mock.Mock(text=html)
  98.         response.search_params = {
  99.             'language': 'en_US'
  100.         }
  101.         results = archlinux.response(response)
  103.         self.assertEqual(type(results), list)
  104.         self.assertEqual(len(results), len(expected))
  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)