@@ -5,6 +5,12 @@ from searx import plugins
55from mock import Mock
66
77
8+def get_search_mock(query, **kwargs):
9+ return {'search': Mock(query=query,
10+ result_container=Mock(answers=set()),
11+ **kwargs)}
12+
13+
814class PluginStoreTest(SearxTestCase):
915
1016 def test_PluginStore_init(self):
@@ -46,23 +52,23 @@ class SelfIPTest(SearxTestCase):
4652 request = Mock(user_plugins=store.plugins,
4753 remote_addr='127.0.0.1')
4854 request.headers.getlist.return_value = []
49− ctx = {'search': Mock(answers=set(),
50− query='ip')}
55+ ctx = get_search_mock(query='ip')
5156 store.call('post_search', request, ctx)
52− self.assertTrue('127.0.0.1' in ctx['search'].answers)
57+ self.assertTrue('127.0.0.1' in ctx['search'].result_container.answers)
5358
5459 # User agent test
5560 request = Mock(user_plugins=store.plugins,
5661 user_agent='Mock')
5762 request.headers.getlist.return_value = []
58− ctx = {'search': Mock(answers=set(),
59− query='user-agent')}
63+
64+ ctx = get_search_mock(query='user-agent')
6065 store.call('post_search', request, ctx)
61− self.assertTrue('Mock' in ctx['search'].answers)
62− ctx = {'search': Mock(answers=set(),
63− query='user agent')}
66+ self.assertTrue('Mock' in ctx['search'].result_container.answers)
67+
68+ ctx = get_search_mock(query='user-agent')
6469 store.call('post_search', request, ctx)
65− self.assertTrue('Mock' in ctx['search'].answers)
66− ctx = {'search': Mock(answers=set(),
67− query='What is my User-Agent?')}
70+ self.assertTrue('Mock' in ctx['search'].result_container.answers)
71+
72+ ctx = get_search_mock(query='What is my User-Agent?')
6873 store.call('post_search', request, ctx)
74+ self.assertTrue('Mock' in ctx['search'].result_container.answers)