| 
									
										
										
										
											2015-06-16 19:56:16 +02:00
										 |  |  | # -*- coding: utf-8 -*- | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | from searx.testing import SearxTestCase | 
					
						
							| 
									
										
										
										
											2020-02-01 11:01:17 +01:00
										 |  |  | from searx.preferences import Preferences | 
					
						
							|  |  |  | from searx.engines import engines | 
					
						
							| 
									
										
										
										
											2015-06-16 19:56:16 +02:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-08-02 13:50:51 +02:00
										 |  |  | import searx.search | 
					
						
							| 
									
										
										
										
											2020-02-01 11:01:17 +01:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | SAFESEARCH = 0 | 
					
						
							|  |  |  | PAGENO = 1 | 
					
						
							|  |  |  | PUBLIC_ENGINE_NAME = 'general dummy' | 
					
						
							|  |  |  | PRIVATE_ENGINE_NAME = 'general private offline' | 
					
						
							|  |  |  | TEST_ENGINES = [ | 
					
						
							|  |  |  |     { | 
					
						
							|  |  |  |         'name': PUBLIC_ENGINE_NAME, | 
					
						
							|  |  |  |         'engine': 'dummy', | 
					
						
							|  |  |  |         'categories': 'general', | 
					
						
							|  |  |  |         'shortcut': 'gd', | 
					
						
							|  |  |  |         'timeout': 3.0, | 
					
						
							|  |  |  |         'tokens': [], | 
					
						
							|  |  |  |     }, | 
					
						
							|  |  |  |     { | 
					
						
							|  |  |  |         'name': PRIVATE_ENGINE_NAME, | 
					
						
							|  |  |  |         'engine': 'dummy-offline', | 
					
						
							|  |  |  |         'categories': 'general', | 
					
						
							|  |  |  |         'shortcut': 'do', | 
					
						
							|  |  |  |         'timeout': 3.0, | 
					
						
							|  |  |  |         'offline': True, | 
					
						
							|  |  |  |         'tokens': ['my-token'], | 
					
						
							|  |  |  |     }, | 
					
						
							|  |  |  | ] | 
					
						
							| 
									
										
										
										
											2019-08-02 13:50:51 +02:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-06-16 19:56:16 +02:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-10-03 17:26:07 +02:00
										 |  |  | class SearchTestCase(SearxTestCase): | 
					
						
							| 
									
										
										
										
											2015-06-16 19:56:16 +02:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-08-02 13:50:51 +02:00
										 |  |  |     @classmethod | 
					
						
							|  |  |  |     def setUpClass(cls): | 
					
						
							| 
									
										
										
										
											2020-02-01 11:01:17 +01:00
										 |  |  |         searx.engines.initialize_engines(TEST_ENGINES) | 
					
						
							| 
									
										
										
										
											2019-08-02 13:50:51 +02:00
										 |  |  | 
 | 
					
						
							|  |  |  |     def test_timeout_simple(self): | 
					
						
							|  |  |  |         searx.search.max_request_timeout = None | 
					
						
							| 
									
										
										
										
											2020-02-01 11:01:17 +01:00
										 |  |  |         search_query = searx.query.SearchQuery('test', [{'category': 'general', 'name': PUBLIC_ENGINE_NAME}], | 
					
						
							|  |  |  |                                                ['general'], 'en-US', SAFESEARCH, PAGENO, None, None, | 
					
						
							|  |  |  |                                                preferences=Preferences(['oscar'], ['general'], engines, [])) | 
					
						
							| 
									
										
										
										
											2019-08-02 13:50:51 +02:00
										 |  |  |         search = searx.search.Search(search_query) | 
					
						
							|  |  |  |         search.search() | 
					
						
							|  |  |  |         self.assertEquals(search.actual_timeout, 3.0) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_timeout_query_above_default_nomax(self): | 
					
						
							|  |  |  |         searx.search.max_request_timeout = None | 
					
						
							| 
									
										
										
										
											2020-02-01 11:01:17 +01:00
										 |  |  |         search_query = searx.query.SearchQuery('test', [{'category': 'general', 'name': PUBLIC_ENGINE_NAME}], | 
					
						
							|  |  |  |                                                ['general'], 'en-US', SAFESEARCH, PAGENO, None, 5.0, | 
					
						
							|  |  |  |                                                preferences=Preferences(['oscar'], ['general'], engines, [])) | 
					
						
							| 
									
										
										
										
											2019-08-02 13:50:51 +02:00
										 |  |  |         search = searx.search.Search(search_query) | 
					
						
							|  |  |  |         search.search() | 
					
						
							|  |  |  |         self.assertEquals(search.actual_timeout, 3.0) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_timeout_query_below_default_nomax(self): | 
					
						
							|  |  |  |         searx.search.max_request_timeout = None | 
					
						
							| 
									
										
										
										
											2020-02-01 11:01:17 +01:00
										 |  |  |         search_query = searx.query.SearchQuery('test', [{'category': 'general', 'name': PUBLIC_ENGINE_NAME}], | 
					
						
							|  |  |  |                                                ['general'], 'en-US', SAFESEARCH, PAGENO, None, 1.0, | 
					
						
							|  |  |  |                                                preferences=Preferences(['oscar'], ['general'], engines, [])) | 
					
						
							| 
									
										
										
										
											2019-08-02 13:50:51 +02:00
										 |  |  |         search = searx.search.Search(search_query) | 
					
						
							|  |  |  |         search.search() | 
					
						
							|  |  |  |         self.assertEquals(search.actual_timeout, 1.0) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_timeout_query_below_max(self): | 
					
						
							|  |  |  |         searx.search.max_request_timeout = 10.0 | 
					
						
							| 
									
										
										
										
											2020-02-01 11:01:17 +01:00
										 |  |  |         search_query = searx.query.SearchQuery('test', [{'category': 'general', 'name': PUBLIC_ENGINE_NAME}], | 
					
						
							|  |  |  |                                                ['general'], 'en-US', SAFESEARCH, PAGENO, None, 5.0, | 
					
						
							|  |  |  |                                                preferences=Preferences(['oscar'], ['general'], engines, [])) | 
					
						
							| 
									
										
										
										
											2019-08-02 13:50:51 +02:00
										 |  |  |         search = searx.search.Search(search_query) | 
					
						
							|  |  |  |         search.search() | 
					
						
							|  |  |  |         self.assertEquals(search.actual_timeout, 5.0) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_timeout_query_above_max(self): | 
					
						
							|  |  |  |         searx.search.max_request_timeout = 10.0 | 
					
						
							| 
									
										
										
										
											2020-02-01 11:01:17 +01:00
										 |  |  |         search_query = searx.query.SearchQuery('test', [{'category': 'general', 'name': PUBLIC_ENGINE_NAME}], | 
					
						
							|  |  |  |                                                ['general'], 'en-US', SAFESEARCH, PAGENO, None, 15.0, | 
					
						
							|  |  |  |                                                preferences=Preferences(['oscar'], ['general'], engines, [])) | 
					
						
							| 
									
										
										
										
											2019-08-02 13:50:51 +02:00
										 |  |  |         search = searx.search.Search(search_query) | 
					
						
							|  |  |  |         search.search() | 
					
						
							|  |  |  |         self.assertEquals(search.actual_timeout, 10.0) | 
					
						
							| 
									
										
										
										
											2020-02-01 11:01:17 +01:00
										 |  |  | 
 | 
					
						
							|  |  |  |     def test_query_private_engine_without_token(self): | 
					
						
							|  |  |  |         search_query = searx.query.SearchQuery('test', [{'category': 'general', 'name': PRIVATE_ENGINE_NAME}], | 
					
						
							|  |  |  |                                                ['general'], 'en-US', SAFESEARCH, PAGENO, None, 2.0, | 
					
						
							|  |  |  |                                                preferences=Preferences(['oscar'], ['general'], engines, [])) | 
					
						
							|  |  |  |         search = searx.search.Search(search_query) | 
					
						
							|  |  |  |         results = search.search() | 
					
						
							|  |  |  |         self.assertEquals(results.results_length(), 0) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_query_private_engine_with_incorrect_token(self): | 
					
						
							|  |  |  |         preferences_with_tokens = Preferences(['oscar'], ['general'], engines, []) | 
					
						
							|  |  |  |         preferences_with_tokens.parse_dict({'tokens': 'bad-token'}) | 
					
						
							|  |  |  |         search_query = searx.query.SearchQuery('test', [{'category': 'general', 'name': PRIVATE_ENGINE_NAME}], | 
					
						
							|  |  |  |                                                ['general'], 'en-US', SAFESEARCH, PAGENO, None, 2.0, | 
					
						
							|  |  |  |                                                preferences=preferences_with_tokens) | 
					
						
							|  |  |  |         search = searx.search.Search(search_query) | 
					
						
							|  |  |  |         results = search.search() | 
					
						
							|  |  |  |         self.assertEquals(results.results_length(), 0) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def test_query_private_engine_with_correct_token(self): | 
					
						
							|  |  |  |         preferences_with_tokens = Preferences(['oscar'], ['general'], engines, []) | 
					
						
							|  |  |  |         preferences_with_tokens.parse_dict({'tokens': 'my-token'}) | 
					
						
							|  |  |  |         search_query = searx.query.SearchQuery('test', [{'category': 'general', 'name': PRIVATE_ENGINE_NAME}], | 
					
						
							|  |  |  |                                                ['general'], 'en-US', SAFESEARCH, PAGENO, None, 2.0, | 
					
						
							|  |  |  |                                                preferences=preferences_with_tokens) | 
					
						
							|  |  |  |         search = searx.search.Search(search_query) | 
					
						
							|  |  |  |         results = search.search() | 
					
						
							|  |  |  |         self.assertEquals(results.results_length(), 1) | 
					
						
							| 
									
										
										
										
											2020-07-03 15:25:04 +02:00
										 |  |  | 
 | 
					
						
							|  |  |  |     def test_external_bang(self): | 
					
						
							|  |  |  |         search_query = searx.query.SearchQuery('yes yes', | 
					
						
							|  |  |  |                                                [{'category': 'general', 'name': PUBLIC_ENGINE_NAME}], | 
					
						
							|  |  |  |                                                ['general'], 'en-US', SAFESEARCH, PAGENO, None, None, | 
					
						
							|  |  |  |                                                preferences=Preferences(['oscar'], ['general'], engines, [],), | 
					
						
							|  |  |  |                                                external_bang="yt") | 
					
						
							|  |  |  |         search = searx.search.Search(search_query) | 
					
						
							|  |  |  |         results = search.search() | 
					
						
							|  |  |  |         # For checking if the user redirected with the youtube external bang | 
					
						
							|  |  |  |         self.assertTrue(results.redirect_url is not None) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         search_query = searx.query.SearchQuery('youtube never gonna give you up', | 
					
						
							|  |  |  |                                                [{'category': 'general', 'name': PUBLIC_ENGINE_NAME}], | 
					
						
							|  |  |  |                                                ['general'], 'en-US', SAFESEARCH, PAGENO, None, None, | 
					
						
							|  |  |  |                                                preferences=Preferences(['oscar'], ['general'], engines, []),) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         search = searx.search.Search(search_query) | 
					
						
							|  |  |  |         results = search.search() | 
					
						
							|  |  |  |         # This should not redirect | 
					
						
							|  |  |  |         self.assertTrue(results.redirect_url is None) |