summaryrefslogtreecommitdiff
path: root/tests/unit/test_results.py
blob: 274b5b37af81a208bd39f769d4912b5913f18761 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
# -*- coding: utf-8 -*-

from searx.results import ResultContainer
from searx.testing import SearxTestCase


def fake_result(url='https://aa.bb/cc?dd=ee#ff',
                title='aaa',
                content='bbb',
                engine='wikipedia', **kwargs):
    result = {'url': url,
              'title': title,
              'content': content,
              'engine': engine}
    result.update(kwargs)
    return result


#  TODO
class ResultContainerTestCase(SearxTestCase):

    def test_empty(self):
        c = ResultContainer()
        self.assertEqual(c.get_ordered_results(), [])

    def test_one_result(self):
        c = ResultContainer()
        c.extend('wikipedia', [fake_result()])
        self.assertEqual(c.results_length(), 1)

    def test_one_suggestion(self):
        c = ResultContainer()
        c.extend('wikipedia', [fake_result(suggestion=True)])
        self.assertEqual(len(c.suggestions), 1)
        self.assertEqual(c.results_length(), 0)

    def test_result_merge(self):
        c = ResultContainer()
        c.extend('wikipedia', [fake_result()])
        c.extend('wikidata', [fake_result(), fake_result(url='https://example.com/')])
        self.assertEqual(c.results_length(), 2)