Loading tests/unit/test_plugins.py +9 −10 Original line number Original line Diff line number Diff line Loading @@ -6,8 +6,7 @@ from mock import Mock def get_search_mock(query, **kwargs): def get_search_mock(query, **kwargs): return Mock(search_query=Mock(query=query, **kwargs), return Mock(query=query, answers=set(), **kwargs) result_container=Mock(answers=set())) class PluginStoreTest(SearxTestCase): class PluginStoreTest(SearxTestCase): Loading Loading @@ -50,11 +49,11 @@ class SelfIPTest(SearxTestCase): request.headers.getlist.return_value = [] request.headers.getlist.return_value = [] search = get_search_mock(query=b'ip', pageno=1) search = get_search_mock(query=b'ip', pageno=1) store.call(store.plugins, 'post_search', request, search) store.call(store.plugins, 'post_search', request, search) self.assertTrue('127.0.0.1' in search.result_container.answers) self.assertTrue('127.0.0.1' in search.answers) search = get_search_mock(query=b'ip', pageno=2) search = get_search_mock(query=b'ip', pageno=2) store.call(store.plugins, 'post_search', request, search) store.call(store.plugins, 'post_search', request, search) self.assertFalse('127.0.0.1' in search.result_container.answers) self.assertFalse('127.0.0.1' in search.answers) # User agent test # User agent test request = Mock(user_agent='Mock') request = Mock(user_agent='Mock') Loading @@ -62,24 +61,24 @@ class SelfIPTest(SearxTestCase): search = get_search_mock(query=b'user-agent', pageno=1) search = get_search_mock(query=b'user-agent', pageno=1) store.call(store.plugins, 'post_search', request, search) store.call(store.plugins, 'post_search', request, search) self.assertTrue('Mock' in search.result_container.answers) self.assertTrue('Mock' in search.answers) search = get_search_mock(query=b'user-agent', pageno=2) search = get_search_mock(query=b'user-agent', pageno=2) store.call(store.plugins, 'post_search', request, search) store.call(store.plugins, 'post_search', request, search) self.assertFalse('Mock' in search.result_container.answers) self.assertFalse('Mock' in search.answers) search = get_search_mock(query=b'user-agent', pageno=1) search = get_search_mock(query=b'user-agent', pageno=1) store.call(store.plugins, 'post_search', request, search) store.call(store.plugins, 'post_search', request, search) self.assertTrue('Mock' in search.result_container.answers) self.assertTrue('Mock' in search.answers) search = get_search_mock(query=b'user-agent', pageno=2) search = get_search_mock(query=b'user-agent', pageno=2) store.call(store.plugins, 'post_search', request, search) store.call(store.plugins, 'post_search', request, search) self.assertFalse('Mock' in search.result_container.answers) self.assertFalse('Mock' in search.answers) search = get_search_mock(query=b'What is my User-Agent?', pageno=1) search = get_search_mock(query=b'What is my User-Agent?', pageno=1) store.call(store.plugins, 'post_search', request, search) store.call(store.plugins, 'post_search', request, search) self.assertTrue('Mock' in search.result_container.answers) self.assertTrue('Mock' in search.answers) search = get_search_mock(query=b'What is my User-Agent?', pageno=2) search = get_search_mock(query=b'What is my User-Agent?', pageno=2) store.call(store.plugins, 'post_search', request, search) store.call(store.plugins, 'post_search', request, search) self.assertFalse('Mock' in search.result_container.answers) self.assertFalse('Mock' in search.answers) Loading
tests/unit/test_plugins.py +9 −10 Original line number Original line Diff line number Diff line Loading @@ -6,8 +6,7 @@ from mock import Mock def get_search_mock(query, **kwargs): def get_search_mock(query, **kwargs): return Mock(search_query=Mock(query=query, **kwargs), return Mock(query=query, answers=set(), **kwargs) result_container=Mock(answers=set())) class PluginStoreTest(SearxTestCase): class PluginStoreTest(SearxTestCase): Loading Loading @@ -50,11 +49,11 @@ class SelfIPTest(SearxTestCase): request.headers.getlist.return_value = [] request.headers.getlist.return_value = [] search = get_search_mock(query=b'ip', pageno=1) search = get_search_mock(query=b'ip', pageno=1) store.call(store.plugins, 'post_search', request, search) store.call(store.plugins, 'post_search', request, search) self.assertTrue('127.0.0.1' in search.result_container.answers) self.assertTrue('127.0.0.1' in search.answers) search = get_search_mock(query=b'ip', pageno=2) search = get_search_mock(query=b'ip', pageno=2) store.call(store.plugins, 'post_search', request, search) store.call(store.plugins, 'post_search', request, search) self.assertFalse('127.0.0.1' in search.result_container.answers) self.assertFalse('127.0.0.1' in search.answers) # User agent test # User agent test request = Mock(user_agent='Mock') request = Mock(user_agent='Mock') Loading @@ -62,24 +61,24 @@ class SelfIPTest(SearxTestCase): search = get_search_mock(query=b'user-agent', pageno=1) search = get_search_mock(query=b'user-agent', pageno=1) store.call(store.plugins, 'post_search', request, search) store.call(store.plugins, 'post_search', request, search) self.assertTrue('Mock' in search.result_container.answers) self.assertTrue('Mock' in search.answers) search = get_search_mock(query=b'user-agent', pageno=2) search = get_search_mock(query=b'user-agent', pageno=2) store.call(store.plugins, 'post_search', request, search) store.call(store.plugins, 'post_search', request, search) self.assertFalse('Mock' in search.result_container.answers) self.assertFalse('Mock' in search.answers) search = get_search_mock(query=b'user-agent', pageno=1) search = get_search_mock(query=b'user-agent', pageno=1) store.call(store.plugins, 'post_search', request, search) store.call(store.plugins, 'post_search', request, search) self.assertTrue('Mock' in search.result_container.answers) self.assertTrue('Mock' in search.answers) search = get_search_mock(query=b'user-agent', pageno=2) search = get_search_mock(query=b'user-agent', pageno=2) store.call(store.plugins, 'post_search', request, search) store.call(store.plugins, 'post_search', request, search) self.assertFalse('Mock' in search.result_container.answers) self.assertFalse('Mock' in search.answers) search = get_search_mock(query=b'What is my User-Agent?', pageno=1) search = get_search_mock(query=b'What is my User-Agent?', pageno=1) store.call(store.plugins, 'post_search', request, search) store.call(store.plugins, 'post_search', request, search) self.assertTrue('Mock' in search.result_container.answers) self.assertTrue('Mock' in search.answers) search = get_search_mock(query=b'What is my User-Agent?', pageno=2) search = get_search_mock(query=b'What is my User-Agent?', pageno=2) store.call(store.plugins, 'post_search', request, search) store.call(store.plugins, 'post_search', request, search) self.assertFalse('Mock' in search.result_container.answers) self.assertFalse('Mock' in search.answers)
