tests/unit/test_webadapter.py · searx

on engine-data help sign in

searx/tests/unit/test_webadapter.py

54 lines 2.0 kB history raw

1# -*- coding: utf-8 -*-

3from searx.testing import SearxTestCase

4from searx.preferences import Preferences

5from searx.engines import engines

7import searx.search

8from searx.search import EngineRef

9from searx.webadapter import validate_engineref_list

12PRIVATE_ENGINE_NAME = 'general private offline'

13TEST_ENGINES = [

14 {

15 'name': PRIVATE_ENGINE_NAME,

16 'engine': 'dummy-offline',

17 'categories': 'general',

18 'shortcut': 'do',

19 'timeout': 3.0,

20 'engine_type': 'offline',

21 'tokens': ['my-token'],

22 },

24SEARCHQUERY = [EngineRef(PRIVATE_ENGINE_NAME, 'general')]

27class ValidateQueryCase(SearxTestCase):

29 @classmethod

30 def setUpClass(cls):

31 searx.search.initialize(TEST_ENGINES)

33 def test_query_private_engine_without_token(self):

34 preferences = Preferences(['oscar'], ['general'], engines, [])

35 valid, unknown, invalid_token = validate_engineref_list(SEARCHQUERY, preferences)

36 self.assertEqual(len(valid), 0)

37 self.assertEqual(len(unknown), 0)

38 self.assertEqual(len(invalid_token), 1)

40 def test_query_private_engine_with_incorrect_token(self):

41 preferences_with_tokens = Preferences(['oscar'], ['general'], engines, [])

42 preferences_with_tokens.parse_dict({'tokens': 'bad-token'})

43 valid, unknown, invalid_token = validate_engineref_list(SEARCHQUERY, preferences_with_tokens)

44 self.assertEqual(len(valid), 0)

45 self.assertEqual(len(unknown), 0)

46 self.assertEqual(len(invalid_token), 1)

48 def test_query_private_engine_with_correct_token(self):

49 preferences_with_tokens = Preferences(['oscar'], ['general'], engines, [])

50 preferences_with_tokens.parse_dict({'tokens': 'my-token'})

51 valid, unknown, invalid_token = validate_engineref_list(SEARCHQUERY, preferences_with_tokens)

52 self.assertEqual(len(valid), 1)

53 self.assertEqual(len(unknown), 0)

54 self.assertEqual(len(invalid_token), 0)