summaryrefslogtreecommitdiff
path: root/tests/unit/test_plugin_calculator.py
blob: a26e78e1b3370455ea148c9c44eff3fa97cf65a8 (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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
# SPDX-License-Identifier: AGPL-3.0-or-later
# pylint: disable=missing-module-docstring

from mock import Mock
from parameterized.parameterized import parameterized
from searx import plugins
from tests import SearxTestCase
from .test_utils import random_string

from .test_plugins import get_search_mock


class PluginCalculator(SearxTestCase):  # pylint: disable=missing-class-docstring
    def setUp(self):
        self.store = plugins.PluginStore()
        plugin = plugins.load_and_initialize_plugin('searx.plugins.calculator', False, (None, {}))
        self.store.register(plugin)

    def test_plugin_store_init(self):
        self.assertEqual(1, len(self.store.plugins))

    def test_single_page_number_true(self):
        request = Mock(remote_addr='127.0.0.1')
        search = get_search_mock(query=random_string(10), pageno=2)
        result = self.store.call(self.store.plugins, 'post_search', request, search)
        self.assertTrue(result)
        self.assertNotIn('calculate', search.result_container.answers)

    def test_long_query_true(self):
        request = Mock(remote_addr='127.0.0.1')
        search = get_search_mock(query=random_string(101), pageno=1)
        result = self.store.call(self.store.plugins, 'post_search', request, search)
        self.assertTrue(result)
        self.assertNotIn('calculate', search.result_container.answers)

    def test_alpha_true(self):
        request = Mock(remote_addr='127.0.0.1')
        search = get_search_mock(query=random_string(10), pageno=1)
        result = self.store.call(self.store.plugins, 'post_search', request, search)
        self.assertTrue(result)
        self.assertNotIn('calculate', search.result_container.answers)

    @parameterized.expand(
        [
            "1+1",
            "1-1",
            "1*1",
            "1/1",
            "1**1",
            "1^1",
        ]
    )
    def test_int_operations(self, operation):
        request = Mock(remote_addr='127.0.0.1')
        search = get_search_mock(query=operation, pageno=1)
        result = self.store.call(self.store.plugins, 'post_search', request, search)
        self.assertTrue(result)
        self.assertIn('calculate', search.result_container.answers)

    @parameterized.expand(
        [
            "1/0",
        ]
    )
    def test_invalid_operations(self, operation):
        request = Mock(remote_addr='127.0.0.1')
        search = get_search_mock(query=operation, pageno=1)
        result = self.store.call(self.store.plugins, 'post_search', request, search)
        self.assertTrue(result)
        self.assertNotIn('calculate', search.result_container.answers)