forked from zaclys/searxng
733b795d53
incident: flask_babel.gettext() does not work in the engine modules. cause: the request() and response() functions of the engine modules run in the processor, whose search() method runs in a thread and in the threads the context of the Flask app does not exist. The context of the Flask app is needed by the gettext() function for the L10n. Solution: copy context of the Flask app into the threads. [1] special case: We cannot equip the search() method of the processors with the decorator [1], because the decorator requires a context (Flask app) that does not yet exist at the time of the initialization of the processors (the initialization of the processors is part of the initialization of the Flask app). [1] https://flask.palletsprojects.com/en/2.3.x/api/#flask.copy_current_request_context Signed-off-by: Markus Heiser <markus.heiser@darmarit.de>
137 lines
4.6 KiB
Python
137 lines
4.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from copy import copy
|
|
|
|
import searx.search
|
|
from searx.search import SearchQuery, EngineRef
|
|
from searx import settings
|
|
from tests import SearxTestCase
|
|
|
|
|
|
SAFESEARCH = 0
|
|
PAGENO = 1
|
|
PUBLIC_ENGINE_NAME = 'general dummy'
|
|
TEST_ENGINES = [
|
|
{
|
|
'name': PUBLIC_ENGINE_NAME,
|
|
'engine': 'dummy',
|
|
'categories': 'general',
|
|
'shortcut': 'gd',
|
|
'timeout': 3.0,
|
|
'tokens': [],
|
|
},
|
|
]
|
|
|
|
|
|
class SearchQueryTestCase(SearxTestCase):
|
|
def test_repr(self):
|
|
s = SearchQuery('test', [EngineRef('bing', 'general')], 'all', 0, 1, '1', 5.0, 'g')
|
|
self.assertEqual(
|
|
repr(s), "SearchQuery('test', [EngineRef('bing', 'general')], 'all', 0, 1, '1', 5.0, 'g')"
|
|
) # noqa
|
|
|
|
def test_eq(self):
|
|
s = SearchQuery('test', [EngineRef('bing', 'general')], 'all', 0, 1, None, None, None)
|
|
t = SearchQuery('test', [EngineRef('google', 'general')], 'all', 0, 1, None, None, None)
|
|
self.assertEqual(s, s)
|
|
self.assertNotEqual(s, t)
|
|
|
|
def test_copy(self):
|
|
s = SearchQuery('test', [EngineRef('bing', 'general')], 'all', 0, 1, None, None, None)
|
|
t = copy(s)
|
|
self.assertEqual(s, t)
|
|
|
|
|
|
class SearchTestCase(SearxTestCase):
|
|
def setUp(self):
|
|
|
|
from searx import webapp # pylint disable=import-outside-toplevel
|
|
|
|
self.app = webapp.app
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
searx.search.initialize(TEST_ENGINES)
|
|
|
|
def test_timeout_simple(self):
|
|
settings['outgoing']['max_request_timeout'] = None
|
|
search_query = SearchQuery(
|
|
'test', [EngineRef(PUBLIC_ENGINE_NAME, 'general')], 'en-US', SAFESEARCH, PAGENO, None, None
|
|
)
|
|
search = searx.search.Search(search_query)
|
|
with self.app.test_request_context('/search'):
|
|
search.search()
|
|
self.assertEqual(search.actual_timeout, 3.0)
|
|
|
|
def test_timeout_query_above_default_nomax(self):
|
|
settings['outgoing']['max_request_timeout'] = None
|
|
search_query = SearchQuery(
|
|
'test', [EngineRef(PUBLIC_ENGINE_NAME, 'general')], 'en-US', SAFESEARCH, PAGENO, None, 5.0
|
|
)
|
|
search = searx.search.Search(search_query)
|
|
with self.app.test_request_context('/search'):
|
|
search.search()
|
|
self.assertEqual(search.actual_timeout, 3.0)
|
|
|
|
def test_timeout_query_below_default_nomax(self):
|
|
settings['outgoing']['max_request_timeout'] = None
|
|
search_query = SearchQuery(
|
|
'test', [EngineRef(PUBLIC_ENGINE_NAME, 'general')], 'en-US', SAFESEARCH, PAGENO, None, 1.0
|
|
)
|
|
search = searx.search.Search(search_query)
|
|
with self.app.test_request_context('/search'):
|
|
search.search()
|
|
self.assertEqual(search.actual_timeout, 1.0)
|
|
|
|
def test_timeout_query_below_max(self):
|
|
settings['outgoing']['max_request_timeout'] = 10.0
|
|
search_query = SearchQuery(
|
|
'test', [EngineRef(PUBLIC_ENGINE_NAME, 'general')], 'en-US', SAFESEARCH, PAGENO, None, 5.0
|
|
)
|
|
search = searx.search.Search(search_query)
|
|
with self.app.test_request_context('/search'):
|
|
search.search()
|
|
self.assertEqual(search.actual_timeout, 5.0)
|
|
|
|
def test_timeout_query_above_max(self):
|
|
settings['outgoing']['max_request_timeout'] = 10.0
|
|
search_query = SearchQuery(
|
|
'test', [EngineRef(PUBLIC_ENGINE_NAME, 'general')], 'en-US', SAFESEARCH, PAGENO, None, 15.0
|
|
)
|
|
search = searx.search.Search(search_query)
|
|
with self.app.test_request_context('/search'):
|
|
search.search()
|
|
self.assertEqual(search.actual_timeout, 10.0)
|
|
|
|
def test_external_bang(self):
|
|
search_query = SearchQuery(
|
|
'yes yes',
|
|
[EngineRef(PUBLIC_ENGINE_NAME, 'general')],
|
|
'en-US',
|
|
SAFESEARCH,
|
|
PAGENO,
|
|
None,
|
|
None,
|
|
external_bang="yt",
|
|
)
|
|
search = searx.search.Search(search_query)
|
|
results = search.search()
|
|
# For checking if the user redirected with the youtube external bang
|
|
self.assertTrue(results.redirect_url is not None)
|
|
|
|
search_query = SearchQuery(
|
|
'youtube never gonna give you up',
|
|
[EngineRef(PUBLIC_ENGINE_NAME, 'general')],
|
|
'en-US',
|
|
SAFESEARCH,
|
|
PAGENO,
|
|
None,
|
|
None,
|
|
)
|
|
|
|
search = searx.search.Search(search_query)
|
|
with self.app.test_request_context('/search'):
|
|
results = search.search()
|
|
# This should not redirect
|
|
self.assertTrue(results.redirect_url is None)
|