| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111 |
- from collections import defaultdict
- import mock
- from searx.engines import archlinux
- from searx.testing import SearxTestCase
-
- domains = {
- 'en': 'https://wiki.archlinux.org',
- 'de': 'https://wiki.archlinux.de',
- 'fr': 'https://wiki.archlinux.fr',
- 'ja': 'https://wiki.archlinuxjp.org',
- 'ro': 'http://wiki.archlinux.ro',
- 'tr': 'http://archtr.org/wiki'
- }
-
-
- class TestArchLinuxEngine(SearxTestCase):
-
- def test_request(self):
- query = 'test_query'
- dic = defaultdict(dict)
- dic['pageno'] = 1
- dic['language'] = 'en-US'
- params = archlinux.request(query, dic)
- self.assertTrue('url' in params)
- self.assertTrue(query in params['url'])
- self.assertTrue('wiki.archlinux.org' in params['url'])
-
- for lang, name in archlinux.main_langs:
- dic['language'] = lang
- params = archlinux.request(query, dic)
- self.assertTrue(name in params['url'])
-
- for lang, domain in domains.items():
- dic['language'] = lang
- params = archlinux.request(query, dic)
- self.assertTrue(domain in params['url'])
-
- def test_response(self):
- response = mock.Mock(text='<html></html>',
- search_params={'language': 'en_US'})
- self.assertEqual(archlinux.response(response), [])
-
- html = """
- <ul class="mw-search-results">
- <li>
- <div class="mw-search-result-heading">
- <a href="/index.php/ATI" title="ATI">ATI</a>
- </div>
- <div class="searchresult">
- Lorem ipsum dolor sit amet
- </div>
- <div class="mw-search-result-data">
- 30 KB (4,630 words) - 19:04, 17 March 2016</div>
- </li>
- <li>
- <div class="mw-search-result-heading">
- <a href="/index.php/Frequently_asked_questions" title="Frequently asked questions">
- Frequently asked questions
- </a>
- </div>
- <div class="searchresult">
- CPUs with AMDs instruction set "AMD64"
- </div>
- <div class="mw-search-result-data">
- 17 KB (2,722 words) - 20:13, 21 March 2016
- </div>
- </li>
- <li>
- <div class="mw-search-result-heading">
- <a href="/index.php/CPU_frequency_scaling" title="CPU frequency scaling">CPU frequency scaling</a>
- </div>
- <div class="searchresult">
- ondemand for AMD and older Intel CPU
- </div>
- <div class="mw-search-result-data">
- 15 KB (2,319 words) - 23:46, 16 March 2016
- </div>
- </li>
- </ul>
- """
-
- expected = [
- {
- 'title': 'ATI',
- 'url': 'https://wiki.archlinux.org/index.php/ATI'
- },
- {
- 'title': 'Frequently asked questions',
- 'url': 'https://wiki.archlinux.org/index.php/Frequently_asked_questions'
- },
- {
- 'title': 'CPU frequency scaling',
- 'url': 'https://wiki.archlinux.org/index.php/CPU_frequency_scaling'
- }
- ]
-
- response = mock.Mock(text=html)
- response.search_params = {
- 'language': 'en_US'
- }
- results = archlinux.response(response)
-
- self.assertEqual(type(results), list)
- self.assertEqual(len(results), len(expected))
-
- i = 0
- for exp in expected:
- res = results[i]
- i += 1
- for key, value in exp.items():
- self.assertEqual(res[key], value)
|
