logo

searx

My custom branche(s) on searx, a meta-search engine git clone https://hacktivis.me/git/searx.git

test_plugins.py (3136B)

    

  1. # -*- coding: utf-8 -*-
  3. from searx.testing import SearxTestCase
  4. from searx import plugins
  5. from mock import Mock
  8. def get_search_mock(query, **kwargs):
  9.     return Mock(search_query=Mock(query=query, **kwargs),
  10.                 result_container=Mock(answers=set()))
  13. class PluginStoreTest(SearxTestCase):
  15.     def test_PluginStore_init(self):
  16.         store = plugins.PluginStore()
  17.         self.assertTrue(isinstance(store.plugins, list) and len(store.plugins) == 0)
  19.     def test_PluginStore_register(self):
  20.         store = plugins.PluginStore()
  21.         testplugin = plugins.Plugin()
  22.         store.register(testplugin)
  24.         self.assertTrue(len(store.plugins) == 1)
  26.     def test_PluginStore_call(self):
  27.         store = plugins.PluginStore()
  28.         testplugin = plugins.Plugin()
  29.         store.register(testplugin)
  30.         setattr(testplugin, 'asdf', Mock())
  31.         request = Mock()
  32.         store.call([], 'asdf', request, Mock())
  34.         self.assertFalse(testplugin.asdf.called)
  36.         store.call([testplugin], 'asdf', request, Mock())
  37.         self.assertTrue(testplugin.asdf.called)
  40. class SelfIPTest(SearxTestCase):
  42.     def test_PluginStore_init(self):
  43.         store = plugins.PluginStore()
  44.         store.register(plugins.self_info)
  46.         self.assertTrue(len(store.plugins) == 1)
  48.         # IP test
  49.         request = Mock(remote_addr='127.0.0.1')
  50.         request.headers.getlist.return_value = []
  51.         search = get_search_mock(query=b'ip', pageno=1)
  52.         store.call(store.plugins, 'post_search', request, search)
  53.         self.assertTrue('127.0.0.1' in search.result_container.answers)
  55.         search = get_search_mock(query=b'ip', pageno=2)
  56.         store.call(store.plugins, 'post_search', request, search)
  57.         self.assertFalse('127.0.0.1' in search.result_container.answers)
  59.         # User agent test
  60.         request = Mock(user_agent='Mock')
  61.         request.headers.getlist.return_value = []
  63.         search = get_search_mock(query=b'user-agent', pageno=1)
  64.         store.call(store.plugins, 'post_search', request, search)
  65.         self.assertTrue('Mock' in search.result_container.answers)
  67.         search = get_search_mock(query=b'user-agent', pageno=2)
  68.         store.call(store.plugins, 'post_search', request, search)
  69.         self.assertFalse('Mock' in search.result_container.answers)
  71.         search = get_search_mock(query=b'user-agent', pageno=1)
  72.         store.call(store.plugins, 'post_search', request, search)
  73.         self.assertTrue('Mock' in search.result_container.answers)
  75.         search = get_search_mock(query=b'user-agent', pageno=2)
  76.         store.call(store.plugins, 'post_search', request, search)
  77.         self.assertFalse('Mock' in search.result_container.answers)
  79.         search = get_search_mock(query=b'What is my User-Agent?', pageno=1)
  80.         store.call(store.plugins, 'post_search', request, search)
  81.         self.assertTrue('Mock' in search.result_container.answers)
  83.         search = get_search_mock(query=b'What is my User-Agent?', pageno=2)
  84.         store.call(store.plugins, 'post_search', request, search)
  85.         self.assertFalse('Mock' in search.result_container.answers)