logo

searx

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

test_results.py (1225B)

    

  1. # -*- coding: utf-8 -*-

  2. 

  3. from searx.results import ResultContainer

  4. from searx.testing import SearxTestCase

  5. 

  6. 

  7. def fake_result(url='https://aa.bb/cc?dd=ee#ff',

  8.                 title='aaa',

  9.                 content='bbb',

  10.                 engine='wikipedia', **kwargs):

  11.     result = {'url': url,

  12.               'title': title,

  13.               'content': content,

  14.               'engine': engine}

  15.     result.update(kwargs)

  16.     return result

  17. 

  18. 

  19. #  TODO

  20. class ResultContainerTestCase(SearxTestCase):

  21. 

  22.     def test_empty(self):

  23.         c = ResultContainer()

  24.         self.assertEqual(c.get_ordered_results(), [])

  25. 

  26.     def test_one_result(self):

  27.         c = ResultContainer()

  28.         c.extend('wikipedia', [fake_result()])

  29.         self.assertEqual(c.results_length(), 1)

  30. 

  31.     def test_one_suggestion(self):

  32.         c = ResultContainer()

  33.         c.extend('wikipedia', [fake_result(suggestion=True)])

  34.         self.assertEqual(len(c.suggestions), 1)

  35.         self.assertEqual(c.results_length(), 0)

  36. 

  37.     def test_result_merge(self):

  38.         c = ResultContainer()

  39.         c.extend('wikipedia', [fake_result()])

  40.         c.extend('wikidata', [fake_result(), fake_result(url='https://example.com/')])

  41.         self.assertEqual(c.results_length(), 2)