mirror of
https://github.com/parchlinuxB/Gitee.git
synced 2025-02-23 02:15:43 -05:00
36a1ef1239
In [1] and [2] we discussed the need of a Result.results property and how we can
avoid unclear code. This patch implements a class for the reslut-lists of
engines::
searx.result_types.EngineResults
A simple example for the usage in engine development::
from searx.result_types import EngineResults
...
def response(resp) -> EngineResults:
res = EngineResults()
...
res.add( res.types.Answer(answer="lorem ipsum ..", url="https://example.org") )
...
return res
[1] https://github.com/searxng/searxng/pull/4183#pullrequestreview-257400034
[2] https://github.com/searxng/searxng/pull/4183#issuecomment-2614301580
Signed-off-by: Markus Heiser <markus.heiser@darmarit.de>
98 lines
3.4 KiB
Python
98 lines
3.4 KiB
Python
# SPDX-License-Identifier: AGPL-3.0-or-later
|
|
# pylint: disable=missing-module-docstring,disable=missing-class-docstring,invalid-name
|
|
|
|
from parameterized.parameterized import parameterized
|
|
|
|
import searx.plugins
|
|
import searx.preferences
|
|
|
|
from searx.extended_types import sxng_request
|
|
from searx.plugins._core import _default, ModulePlugin
|
|
from searx.result_types import Answer
|
|
from searx.utils import load_module
|
|
|
|
from tests import SearxTestCase
|
|
from .test_utils import random_string
|
|
from .test_plugins import do_post_search
|
|
|
|
|
|
class PluginCalculator(SearxTestCase):
|
|
|
|
def setUp(self):
|
|
super().setUp()
|
|
|
|
f = _default / "calculator.py"
|
|
mod = load_module(f.name, str(f.parent))
|
|
engines = {}
|
|
|
|
self.storage = searx.plugins.PluginStorage()
|
|
self.storage.register(ModulePlugin(mod))
|
|
self.storage.init(self.app)
|
|
self.pref = searx.preferences.Preferences(["simple"], ["general"], engines, self.storage)
|
|
self.pref.parse_dict({"locale": "en"})
|
|
|
|
def test_plugin_store_init(self):
|
|
self.assertEqual(1, len(self.storage))
|
|
|
|
def test_pageno_1_2(self):
|
|
with self.app.test_request_context():
|
|
sxng_request.preferences = self.pref
|
|
query = "1+1"
|
|
answer = Answer(answer=f"{query} = {eval(query)}") # pylint: disable=eval-used
|
|
|
|
search = do_post_search(query, self.storage, pageno=1)
|
|
self.assertIn(answer, search.result_container.answers)
|
|
|
|
search = do_post_search(query, self.storage, pageno=2)
|
|
self.assertEqual(list(search.result_container.answers), [])
|
|
|
|
def test_long_query_ignored(self):
|
|
with self.app.test_request_context():
|
|
sxng_request.preferences = self.pref
|
|
query = f"1+1 {random_string(101)}"
|
|
search = do_post_search(query, self.storage)
|
|
self.assertEqual(list(search.result_container.answers), [])
|
|
|
|
@parameterized.expand(
|
|
[
|
|
("1+1", "2", "en"),
|
|
("1-1", "0", "en"),
|
|
("1*1", "1", "en"),
|
|
("1/1", "1", "en"),
|
|
("1**1", "1", "en"),
|
|
("1^1", "1", "en"),
|
|
("1,000.0+1,000.0", "2,000", "en"),
|
|
("1.0+1.0", "2", "en"),
|
|
("1.0-1.0", "0", "en"),
|
|
("1.0*1.0", "1", "en"),
|
|
("1.0/1.0", "1", "en"),
|
|
("1.0**1.0", "1", "en"),
|
|
("1.0^1.0", "1", "en"),
|
|
("1.000,0+1.000,0", "2.000", "de"),
|
|
("1,0+1,0", "2", "de"),
|
|
("1,0-1,0", "0", "de"),
|
|
("1,0*1,0", "1", "de"),
|
|
("1,0/1,0", "1", "de"),
|
|
("1,0**1,0", "1", "de"),
|
|
("1,0^1,0", "1", "de"),
|
|
]
|
|
)
|
|
def test_localized_query(self, query: str, res: str, lang: str):
|
|
with self.app.test_request_context():
|
|
self.pref.parse_dict({"locale": lang})
|
|
sxng_request.preferences = self.pref
|
|
answer = Answer(answer=f"{query} = {res}")
|
|
|
|
search = do_post_search(query, self.storage)
|
|
self.assertIn(answer, search.result_container.answers)
|
|
|
|
@parameterized.expand(
|
|
[
|
|
"1/0",
|
|
]
|
|
)
|
|
def test_invalid_operations(self, query):
|
|
with self.app.test_request_context():
|
|
sxng_request.preferences = self.pref
|
|
search = do_post_search(query, self.storage)
|
|
self.assertEqual(list(search.result_container.answers), [])
|