git.iko.soy/searx
on engine-data help sign in
searx/tests/unit/test_preferences.py
140 lines 6.3 kB history raw
1from searx.preferences import (EnumStringSetting, MapSetting, MissingArgumentException, SearchLanguageSetting,
2 MultipleChoiceSetting, PluginsSetting, ValidationException)
3from searx.testing import SearxTestCase
6class PluginStub:
8 def __init__(self, plugin_id, default_on):
9 self.id = plugin_id
10 self.default_on = default_on
13class TestSettings(SearxTestCase):
14 # map settings
16 def test_map_setting_invalid_initialization(self):
17 with self.assertRaises(MissingArgumentException):
18 MapSetting(3, wrong_argument={'0': 0})
20 def test_map_setting_invalid_default_value(self):
21 with self.assertRaises(ValidationException):
22 MapSetting(3, map={'dog': 1, 'bat': 2})
24 def test_map_setting_invalid_choice(self):
25 setting = MapSetting(2, map={'dog': 1, 'bat': 2})
26 with self.assertRaises(ValidationException):
27 setting.parse('cat')
29 def test_map_setting_valid_default(self):
30 setting = MapSetting(3, map={'dog': 1, 'bat': 2, 'cat': 3})
31 self.assertEqual(setting.get_value(), 3)
33 def test_map_setting_valid_choice(self):
34 setting = MapSetting(3, map={'dog': 1, 'bat': 2, 'cat': 3})
35 self.assertEqual(setting.get_value(), 3)
36 setting.parse('bat')
37 self.assertEqual(setting.get_value(), 2)
39 # enum settings
40 def test_enum_setting_invalid_initialization(self):
41 with self.assertRaises(MissingArgumentException):
42 EnumStringSetting('cat', wrong_argument=[0, 1, 2])
44 def test_enum_setting_invalid_default_value(self):
45 with self.assertRaises(ValidationException):
46 EnumStringSetting(3, choices=[0, 1, 2])
48 def test_enum_setting_invalid_choice(self):
49 setting = EnumStringSetting(0, choices=[0, 1, 2])
50 with self.assertRaises(ValidationException):
51 setting.parse(3)
53 def test_enum_setting_valid_default(self):
54 setting = EnumStringSetting(3, choices=[1, 2, 3])
55 self.assertEqual(setting.get_value(), 3)
57 def test_enum_setting_valid_choice(self):
58 setting = EnumStringSetting(3, choices=[1, 2, 3])
59 self.assertEqual(setting.get_value(), 3)
60 setting.parse(2)
61 self.assertEqual(setting.get_value(), 2)
63 # multiple choice settings
64 def test_multiple_setting_invalid_initialization(self):
65 with self.assertRaises(MissingArgumentException):
66 MultipleChoiceSetting(['2'], wrong_argument=['0', '1', '2'])
68 def test_multiple_setting_invalid_default_value(self):
69 with self.assertRaises(ValidationException):
70 MultipleChoiceSetting(['3', '4'], choices=['0', '1', '2'])
72 def test_multiple_setting_invalid_choice(self):
73 setting = MultipleChoiceSetting(['1', '2'], choices=['0', '1', '2'])
74 with self.assertRaises(ValidationException):
75 setting.parse('4, 3')
77 def test_multiple_setting_valid_default(self):
78 setting = MultipleChoiceSetting(['3'], choices=['1', '2', '3'])
79 self.assertEqual(setting.get_value(), ['3'])
81 def test_multiple_setting_valid_choice(self):
82 setting = MultipleChoiceSetting(['3'], choices=['1', '2', '3'])
83 self.assertEqual(setting.get_value(), ['3'])
84 setting.parse('2')
85 self.assertEqual(setting.get_value(), ['2'])
87 # search language settings
88 def test_lang_setting_valid_choice(self):
89 setting = SearchLanguageSetting('all', choices=['all', 'de', 'en'])
90 setting.parse('de')
91 self.assertEqual(setting.get_value(), 'de')
93 def test_lang_setting_invalid_choice(self):
94 setting = SearchLanguageSetting('all', choices=['all', 'de', 'en'])
95 setting.parse('xx')
96 self.assertEqual(setting.get_value(), 'all')
98 def test_lang_setting_old_cookie_choice(self):
99 setting = SearchLanguageSetting('all', choices=['all', 'es', 'es-ES'])
100 setting.parse('es_XA')
101 self.assertEqual(setting.get_value(), 'es')
103 def test_lang_setting_old_cookie_format(self):
104 setting = SearchLanguageSetting('all', choices=['all', 'es', 'es-ES'])
105 setting.parse('es_ES')
106 self.assertEqual(setting.get_value(), 'es-ES')
108 # plugins settings
109 def test_plugins_setting_all_default_enabled(self):
110 plugin1 = PluginStub('plugin1', True)
111 plugin2 = PluginStub('plugin2', True)
112 setting = PluginsSetting(['3'], choices=[plugin1, plugin2])
113 self.assertEqual(setting.get_enabled(), set(['plugin1', 'plugin2']))
115 def test_plugins_setting_few_default_enabled(self):
116 plugin1 = PluginStub('plugin1', True)
117 plugin2 = PluginStub('plugin2', False)
118 plugin3 = PluginStub('plugin3', True)
119 setting = PluginsSetting('name', choices=[plugin1, plugin2, plugin3])
120 self.assertEqual(setting.get_enabled(), set(['plugin1', 'plugin3']))
123class TestPreferences(SearxTestCase):
125 def test_encode(self):
126 from searx.preferences import Preferences
127 pref = Preferences(['oscar'], ['general'], {}, [])
128 url_params = 'eJx1VMmO2zAM_Zr6YrTocujJh6JF0QEKzKAz7VVgJNohLIseUU7ivy-VcWy5yyGOTVGP73GLKJNPYjiYgGeT4NB8BS9YOSY' \
129 'TUdifMDYM-vmGY1d5CN0EHTYOK88W_PXNkcDBozOjnzoK0vyi4bWnHs2RU4-zvHr_-RF9a-5Cy3GARByy7X7EkKMoBeMp9CuPQ-SzYMx' \
130 '8Vr9P1qKI-XJ_p1fOkRJWNCgVM0a-zAttmBJbHkaPSZlNts-_jiuBFgUh2mPztkpHHLBhsRArDHvm356eHh5vATS0Mqagr0ZsZO_V8hT' \
131 'B9srt54_v6jewJugqL4Nn_hYSdhxnI-jRpi05GDQCStOT7UGVmJY8ZnltRKyF23SGiLWjqNcygKGkpyeGZIywJfD1gI5AjRTAmBM55Aw' \
132 'Q0Tn626lj7jzWo4e5hnEsIlprX6dTgdBRpyRBFKTDgBF8AasVyT4gvSTEoXRpXWRyG3CYQYld65I_V6lboILTMAlZY65_ejRDcHgp0Tv' \
133 'EPtGAsqTiBf3m76g7pP9B84mwjPvuUtASRDei1nDF2ix_JXW91UJkXrPh6RAhznVmKyQl7dwJdMJ6bz1QOmgzYlrEzHDMcEUuo44AgS1' \
134 'CvkjaOb2Q2AyY5oGDTs_OLXE_c2I5cg9hk3kEJZ0fu4SuktsIA2RhuJwP86AdripThCBeO9uVUejyPGmFSxPrqEYcuWi25zOEXV9tc1m' \
135 '_KP1nafYtdfv6Q9hKfWmGm9A_3G635UwiVndLGdFCiLWkONk0xUxGLGGweGWTa2nZYZ0fS1YKlE3Uuw8fPl52E5U8HJYbC7sbjXUsrnT' \
136 'XHXRbELfO-1fGSqskiGnMK7B0dV3t8Lq08pbdtYpuVdoKWA2Yjuyah_vHp2rZWjo0zXL8Gw8DTj0='
137 pref.parse_encoded_data(url_params)
138 self.assertEqual(
139 vars(pref.key_value_settings['categories']),
140 {'value': ['general'], 'locked': False, 'choices': ['general', 'none']})