@@ -5,6 +5,7 @@ from searx.preferences import Preferences
5from searx.engines import engines
5from searx.engines import engines
7import searx.search
7import searx.search
8+from searx.search import SearchQuery
10SAFESEARCH = 0
11SAFESEARCH = 0
@@ -40,53 +41,53 @@ class SearchTestCase(SearxTestCase):
41 def test_timeout_simple(self):
42 def test_timeout_simple(self):
42 searx.search.max_request_timeout = None
43 searx.search.max_request_timeout = None
43− search_query = searx.query.SearchQuery('test', [{'category': 'general', 'name': PUBLIC_ENGINE_NAME}],
44+ search_query = SearchQuery('test', [{'category': 'general', 'name': PUBLIC_ENGINE_NAME}],
44− ['general'], 'en-US', SAFESEARCH, PAGENO, None, None,
45+ ['general'], 'en-US', SAFESEARCH, PAGENO, None, None,
45− preferences=Preferences(['oscar'], ['general'], engines, []))
46+ preferences=Preferences(['oscar'], ['general'], engines, []))
46 search = searx.search.Search(search_query)
47 search = searx.search.Search(search_query)
47 search.search()
48 search.search()
48 self.assertEqual(search.actual_timeout, 3.0)
49 self.assertEqual(search.actual_timeout, 3.0)
50 def test_timeout_query_above_default_nomax(self):
51 def test_timeout_query_above_default_nomax(self):
51 searx.search.max_request_timeout = None
52 searx.search.max_request_timeout = None
52− search_query = searx.query.SearchQuery('test', [{'category': 'general', 'name': PUBLIC_ENGINE_NAME}],
53+ search_query = SearchQuery('test', [{'category': 'general', 'name': PUBLIC_ENGINE_NAME}],
53− ['general'], 'en-US', SAFESEARCH, PAGENO, None, 5.0,
54+ ['general'], 'en-US', SAFESEARCH, PAGENO, None, 5.0,
54− preferences=Preferences(['oscar'], ['general'], engines, []))
55+ preferences=Preferences(['oscar'], ['general'], engines, []))
55 search = searx.search.Search(search_query)
56 search = searx.search.Search(search_query)
56 search.search()
57 search.search()
57 self.assertEqual(search.actual_timeout, 3.0)
58 self.assertEqual(search.actual_timeout, 3.0)
59 def test_timeout_query_below_default_nomax(self):
60 def test_timeout_query_below_default_nomax(self):
60 searx.search.max_request_timeout = None
61 searx.search.max_request_timeout = None
61− search_query = searx.query.SearchQuery('test', [{'category': 'general', 'name': PUBLIC_ENGINE_NAME}],
62+ search_query = SearchQuery('test', [{'category': 'general', 'name': PUBLIC_ENGINE_NAME}],
62− ['general'], 'en-US', SAFESEARCH, PAGENO, None, 1.0,
63+ ['general'], 'en-US', SAFESEARCH, PAGENO, None, 1.0,
63− preferences=Preferences(['oscar'], ['general'], engines, []))
64+ preferences=Preferences(['oscar'], ['general'], engines, []))
64 search = searx.search.Search(search_query)
65 search = searx.search.Search(search_query)
65 search.search()
66 search.search()
66 self.assertEqual(search.actual_timeout, 1.0)
67 self.assertEqual(search.actual_timeout, 1.0)
68 def test_timeout_query_below_max(self):
69 def test_timeout_query_below_max(self):
69 searx.search.max_request_timeout = 10.0
70 searx.search.max_request_timeout = 10.0
70− search_query = searx.query.SearchQuery('test', [{'category': 'general', 'name': PUBLIC_ENGINE_NAME}],
71+ search_query = SearchQuery('test', [{'category': 'general', 'name': PUBLIC_ENGINE_NAME}],
71− ['general'], 'en-US', SAFESEARCH, PAGENO, None, 5.0,
72+ ['general'], 'en-US', SAFESEARCH, PAGENO, None, 5.0,
72− preferences=Preferences(['oscar'], ['general'], engines, []))
73+ preferences=Preferences(['oscar'], ['general'], engines, []))
73 search = searx.search.Search(search_query)
74 search = searx.search.Search(search_query)
74 search.search()
75 search.search()
75 self.assertEqual(search.actual_timeout, 5.0)
76 self.assertEqual(search.actual_timeout, 5.0)
77 def test_timeout_query_above_max(self):
78 def test_timeout_query_above_max(self):
78 searx.search.max_request_timeout = 10.0
79 searx.search.max_request_timeout = 10.0
79− search_query = searx.query.SearchQuery('test', [{'category': 'general', 'name': PUBLIC_ENGINE_NAME}],
80+ search_query = SearchQuery('test', [{'category': 'general', 'name': PUBLIC_ENGINE_NAME}],
80− ['general'], 'en-US', SAFESEARCH, PAGENO, None, 15.0,
81+ ['general'], 'en-US', SAFESEARCH, PAGENO, None, 15.0,
81− preferences=Preferences(['oscar'], ['general'], engines, []))
82+ preferences=Preferences(['oscar'], ['general'], engines, []))
82 search = searx.search.Search(search_query)
83 search = searx.search.Search(search_query)
83 search.search()
84 search.search()
84 self.assertEqual(search.actual_timeout, 10.0)
85 self.assertEqual(search.actual_timeout, 10.0)
86 def test_query_private_engine_without_token(self):
87 def test_query_private_engine_without_token(self):
87− search_query = searx.query.SearchQuery('test', [{'category': 'general', 'name': PRIVATE_ENGINE_NAME}],
88+ search_query = SearchQuery('test', [{'category': 'general', 'name': PRIVATE_ENGINE_NAME}],
88− ['general'], 'en-US', SAFESEARCH, PAGENO, None, 2.0,
89+ ['general'], 'en-US', SAFESEARCH, PAGENO, None, 2.0,
89− preferences=Preferences(['oscar'], ['general'], engines, []))
90+ preferences=Preferences(['oscar'], ['general'], engines, []))
90 search = searx.search.Search(search_query)
91 search = searx.search.Search(search_query)
91 results = search.search()
92 results = search.search()
92 self.assertEqual(results.results_length(), 0)
93 self.assertEqual(results.results_length(), 0)
@@ -94,9 +95,9 @@ class SearchTestCase(SearxTestCase):
94 def test_query_private_engine_with_incorrect_token(self):
95 def test_query_private_engine_with_incorrect_token(self):
95 preferences_with_tokens = Preferences(['oscar'], ['general'], engines, [])
96 preferences_with_tokens = Preferences(['oscar'], ['general'], engines, [])
96 preferences_with_tokens.parse_dict({'tokens': 'bad-token'})
97 preferences_with_tokens.parse_dict({'tokens': 'bad-token'})
97− search_query = searx.query.SearchQuery('test', [{'category': 'general', 'name': PRIVATE_ENGINE_NAME}],
98+ search_query = SearchQuery('test', [{'category': 'general', 'name': PRIVATE_ENGINE_NAME}],
98− ['general'], 'en-US', SAFESEARCH, PAGENO, None, 2.0,
99+ ['general'], 'en-US', SAFESEARCH, PAGENO, None, 2.0,
99− preferences=preferences_with_tokens)
100+ preferences=preferences_with_tokens)
100 search = searx.search.Search(search_query)
101 search = searx.search.Search(search_query)
101 results = search.search()
102 results = search.search()
102 self.assertEqual(results.results_length(), 0)
103 self.assertEqual(results.results_length(), 0)
@@ -104,28 +105,28 @@ class SearchTestCase(SearxTestCase):
104 def test_query_private_engine_with_correct_token(self):
105 def test_query_private_engine_with_correct_token(self):
105 preferences_with_tokens = Preferences(['oscar'], ['general'], engines, [])
106 preferences_with_tokens = Preferences(['oscar'], ['general'], engines, [])
106 preferences_with_tokens.parse_dict({'tokens': 'my-token'})
107 preferences_with_tokens.parse_dict({'tokens': 'my-token'})
107− search_query = searx.query.SearchQuery('test', [{'category': 'general', 'name': PRIVATE_ENGINE_NAME}],
108+ search_query = SearchQuery('test', [{'category': 'general', 'name': PRIVATE_ENGINE_NAME}],
108− ['general'], 'en-US', SAFESEARCH, PAGENO, None, 2.0,
109+ ['general'], 'en-US', SAFESEARCH, PAGENO, None, 2.0,
109− preferences=preferences_with_tokens)
110+ preferences=preferences_with_tokens)
110 search = searx.search.Search(search_query)
111 search = searx.search.Search(search_query)
111 results = search.search()
112 results = search.search()
112 self.assertEqual(results.results_length(), 1)
113 self.assertEqual(results.results_length(), 1)
114 def test_external_bang(self):
115 def test_external_bang(self):
115− search_query = searx.query.SearchQuery('yes yes',
116+ search_query = SearchQuery('yes yes',
116− [{'category': 'general', 'name': PUBLIC_ENGINE_NAME}],
117+ [{'category': 'general', 'name': PUBLIC_ENGINE_NAME}],
117− ['general'], 'en-US', SAFESEARCH, PAGENO, None, None,
118+ ['general'], 'en-US', SAFESEARCH, PAGENO, None, None,
118− preferences=Preferences(['oscar'], ['general'], engines, [],),
119+ preferences=Preferences(['oscar'], ['general'], engines, [],),
119− external_bang="yt")
120+ external_bang="yt")
120 search = searx.search.Search(search_query)
121 search = searx.search.Search(search_query)
121 results = search.search()
122 results = search.search()
122 # For checking if the user redirected with the youtube external bang
123 # For checking if the user redirected with the youtube external bang
123 self.assertTrue(results.redirect_url is not None)
124 self.assertTrue(results.redirect_url is not None)
125− search_query = searx.query.SearchQuery('youtube never gonna give you up',
126+ search_query = SearchQuery('youtube never gonna give you up',
126− [{'category': 'general', 'name': PUBLIC_ENGINE_NAME}],
127+ [{'category': 'general', 'name': PUBLIC_ENGINE_NAME}],
127− ['general'], 'en-US', SAFESEARCH, PAGENO, None, None,
128+ ['general'], 'en-US', SAFESEARCH, PAGENO, None, None,
128− preferences=Preferences(['oscar'], ['general'], engines, []),)
129+ preferences=Preferences(['oscar'], ['general'], engines, []),)
130 search = searx.search.Search(search_query)
131 search = searx.search.Search(search_query)
131 results = search.search()
132 results = search.search()