git.iko.soy/searx
on 00a7041 help sign in
searx/tests/unit/test_preferences.py
124 lines 5.1 kB history raw
1from searx.preferences import (EnumStringSetting, MapSetting, MissingArgumentException, SearchLanguageSetting,
2 MultipleChoiceSetting, PluginsSetting, ValidationException)
3from searx.testing import SearxTestCase
6class PluginStub(object):
8 def __init__(self, id, default_on):
9 self.id = 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 setting = MapSetting(3, wrong_argument={'0': 0})
20 def test_map_setting_invalid_default_value(self):
21 with self.assertRaises(ValidationException):
22 setting = 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.assertEquals(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.assertEquals(setting.get_value(), 3)
36 setting.parse('bat')
37 self.assertEquals(setting.get_value(), 2)
39 def test_enum_setting_invalid_initialization(self):
40 with self.assertRaises(MissingArgumentException):
41 setting = EnumStringSetting('cat', wrong_argument=[0, 1, 2])
43 # enum settings
44 def test_enum_setting_invalid_initialization(self):
45 with self.assertRaises(MissingArgumentException):
46 setting = EnumStringSetting('cat', wrong_argument=[0, 1, 2])
48 def test_enum_setting_invalid_default_value(self):
49 with self.assertRaises(ValidationException):
50 setting = EnumStringSetting(3, choices=[0, 1, 2])
52 def test_enum_setting_invalid_choice(self):
53 setting = EnumStringSetting(0, choices=[0, 1, 2])
54 with self.assertRaises(ValidationException):
55 setting.parse(3)
57 def test_enum_setting_valid_default(self):
58 setting = EnumStringSetting(3, choices=[1, 2, 3])
59 self.assertEquals(setting.get_value(), 3)
61 def test_enum_setting_valid_choice(self):
62 setting = EnumStringSetting(3, choices=[1, 2, 3])
63 self.assertEquals(setting.get_value(), 3)
64 setting.parse(2)
65 self.assertEquals(setting.get_value(), 2)
67 # multiple choice settings
68 def test_multiple_setting_invalid_initialization(self):
69 with self.assertRaises(MissingArgumentException):
70 setting = MultipleChoiceSetting(['2'], wrong_argument=['0', '1', '2'])
72 def test_multiple_setting_invalid_default_value(self):
73 with self.assertRaises(ValidationException):
74 setting = MultipleChoiceSetting(['3', '4'], choices=['0', '1', '2'])
76 def test_multiple_setting_invalid_choice(self):
77 setting = MultipleChoiceSetting(['1', '2'], choices=['0', '1', '2'])
78 with self.assertRaises(ValidationException):
79 setting.parse('4, 3')
81 def test_multiple_setting_valid_default(self):
82 setting = MultipleChoiceSetting(['3'], choices=['1', '2', '3'])
83 self.assertEquals(setting.get_value(), ['3'])
85 def test_multiple_setting_valid_choice(self):
86 setting = MultipleChoiceSetting(['3'], choices=['1', '2', '3'])
87 self.assertEquals(setting.get_value(), ['3'])
88 setting.parse('2')
89 self.assertEquals(setting.get_value(), ['2'])
91 # search language settings
92 def test_lang_setting_valid_choice(self):
93 setting = SearchLanguageSetting('all', choices=['all', 'de', 'en'])
94 setting.parse('de')
95 self.assertEquals(setting.get_value(), 'de')
97 def test_lang_setting_invalid_choice(self):
98 setting = SearchLanguageSetting('all', choices=['all', 'de', 'en'])
99 setting.parse('xx')
100 self.assertEquals(setting.get_value(), 'all')
102 def test_lang_setting_old_cookie_choice(self):
103 setting = SearchLanguageSetting('all', choices=['all', 'es', 'es-ES'])
104 setting.parse('es_XA')
105 self.assertEquals(setting.get_value(), 'es')
107 def test_lang_setting_old_cookie_format(self):
108 setting = SearchLanguageSetting('all', choices=['all', 'es', 'es-ES'])
109 setting.parse('es_ES')
110 self.assertEquals(setting.get_value(), 'es-ES')
112 # plugins settings
113 def test_plugins_setting_all_default_enabled(self):
114 plugin1 = PluginStub('plugin1', True)
115 plugin2 = PluginStub('plugin2', True)
116 setting = PluginsSetting(['3'], choices=[plugin1, plugin2])
117 self.assertEquals(setting.get_enabled(), set(['plugin1', 'plugin2']))
119 def test_plugins_setting_few_default_enabled(self):
120 plugin1 = PluginStub('plugin1', True)
121 plugin2 = PluginStub('plugin2', False)
122 plugin3 = PluginStub('plugin3', True)
123 setting = PluginsSetting('name', choices=[plugin1, plugin2, plugin3])
124 self.assertEquals(setting.get_enabled(), set(['plugin1', 'plugin3']))