diff options
Diffstat (limited to 'tests/unit/test_webadapter.py')
| -rw-r--r-- | tests/unit/test_webadapter.py | 46 |
1 files changed, 16 insertions, 30 deletions
diff --git a/tests/unit/test_webadapter.py b/tests/unit/test_webadapter.py index 493020859..4519600a9 100644 --- a/tests/unit/test_webadapter.py +++ b/tests/unit/test_webadapter.py @@ -1,52 +1,38 @@ # SPDX-License-Identifier: AGPL-3.0-or-later -# pylint: disable=missing-module-docstring +# pylint: disable=missing-module-docstring,disable=missing-class-docstring,invalid-name -from searx.preferences import Preferences -from searx.engines import engines +import searx.plugins -import searx.search +from searx.engines import engines +from searx.preferences import Preferences from searx.search import EngineRef from searx.webadapter import validate_engineref_list + from tests import SearxTestCase +PRIVATE_ENGINE_NAME = "dummy private engine" # from the ./settings/test_settings.yml +SEARCHQUERY = [EngineRef(PRIVATE_ENGINE_NAME, "general")] + + +class ValidateQueryCase(SearxTestCase): -PRIVATE_ENGINE_NAME = 'general private offline' -TEST_ENGINES = [ - { - 'name': PRIVATE_ENGINE_NAME, - 'engine': 'dummy-offline', - 'categories': 'general', - 'shortcut': 'do', - 'timeout': 3.0, - 'engine_type': 'offline', - 'tokens': ['my-token'], - }, -] -SEARCHQUERY = [EngineRef(PRIVATE_ENGINE_NAME, 'general')] - - -class ValidateQueryCase(SearxTestCase): # pylint: disable=missing-class-docstring - @classmethod - def setUpClass(cls): - searx.search.initialize(TEST_ENGINES) - - def test_query_private_engine_without_token(self): # pylint:disable=invalid-name - preferences = Preferences(['simple'], ['general'], engines, []) + def test_without_token(self): + preferences = Preferences(['simple'], ['general'], engines, searx.plugins.STORAGE) valid, unknown, invalid_token = validate_engineref_list(SEARCHQUERY, preferences) self.assertEqual(len(valid), 0) self.assertEqual(len(unknown), 0) self.assertEqual(len(invalid_token), 1) - def test_query_private_engine_with_incorrect_token(self): # pylint:disable=invalid-name - preferences_with_tokens = Preferences(['simple'], ['general'], engines, []) + def test_with_incorrect_token(self): + preferences_with_tokens = Preferences(['simple'], ['general'], engines, searx.plugins.STORAGE) preferences_with_tokens.parse_dict({'tokens': 'bad-token'}) valid, unknown, invalid_token = validate_engineref_list(SEARCHQUERY, preferences_with_tokens) self.assertEqual(len(valid), 0) self.assertEqual(len(unknown), 0) self.assertEqual(len(invalid_token), 1) - def test_query_private_engine_with_correct_token(self): # pylint:disable=invalid-name - preferences_with_tokens = Preferences(['simple'], ['general'], engines, []) + def test_with_correct_token(self): + preferences_with_tokens = Preferences(['simple'], ['general'], engines, searx.plugins.STORAGE) preferences_with_tokens.parse_dict({'tokens': 'my-token'}) valid, unknown, invalid_token = validate_engineref_list(SEARCHQUERY, preferences_with_tokens) self.assertEqual(len(valid), 1) |