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 -*-

  2. 

  3. from searx.testing import SearxTestCase

  4. from searx import plugins

  5. from mock import Mock

  6. 

  7. 

  8. def get_search_mock(query, **kwargs):

  9.     return Mock(search_query=Mock(query=query, **kwargs),

  10.                 result_container=Mock(answers=set()))

  11. 

  12. 

  13. class PluginStoreTest(SearxTestCase):

  14. 

  15.     def test_PluginStore_init(self):

  16.         store = plugins.PluginStore()

  17.         self.assertTrue(isinstance(store.plugins, list) and len(store.plugins) == 0)

  18. 

  19.     def test_PluginStore_register(self):

  20.         store = plugins.PluginStore()

  21.         testplugin = plugins.Plugin()

  22.         store.register(testplugin)

  23. 

  24.         self.assertTrue(len(store.plugins) == 1)

  25. 

  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())

  33. 

  34.         self.assertFalse(testplugin.asdf.called)

  35. 

  36.         store.call([testplugin], 'asdf', request, Mock())

  37.         self.assertTrue(testplugin.asdf.called)

  38. 

  39. 

  40. class SelfIPTest(SearxTestCase):

  41. 

  42.     def test_PluginStore_init(self):

  43.         store = plugins.PluginStore()

  44.         store.register(plugins.self_info)

  45. 

  46.         self.assertTrue(len(store.plugins) == 1)

  47. 

  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)

  54. 

  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)

  58. 

  59.         # User agent test

  60.         request = Mock(user_agent='Mock')

  61.         request.headers.getlist.return_value = []

  62. 

  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)

  66. 

  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)

  70. 

  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)

  74. 

  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)

  78. 

  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)

  82. 

  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)