| 
									
										
										
										
											2015-06-16 19:56:16 +02:00
										 |  |  | # -*- coding: utf-8 -*- | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | from searx.testing import SearxTestCase | 
					
						
							| 
									
										
										
										
											2020-09-22 16:22:22 +02:00
										 |  |  | from searx.search import SearchQuery, EngineRef | 
					
						
							| 
									
										
										
										
											2020-09-22 16:55:59 +02:00
										 |  |  | import searx.search | 
					
						
							| 
									
										
										
										
											2020-09-10 18:08:14 +02:00
										 |  |  | import searx.engines | 
					
						
							| 
									
										
										
										
											2020-02-01 11:01:17 +01:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | SAFESEARCH = 0 | 
					
						
							|  |  |  | PAGENO = 1 | 
					
						
							|  |  |  | PUBLIC_ENGINE_NAME = 'general dummy' | 
					
						
							|  |  |  | TEST_ENGINES = [ | 
					
						
							|  |  |  |     { | 
					
						
							|  |  |  |         'name': PUBLIC_ENGINE_NAME, | 
					
						
							|  |  |  |         'engine': 'dummy', | 
					
						
							|  |  |  |         'categories': 'general', | 
					
						
							|  |  |  |         'shortcut': 'gd', | 
					
						
							|  |  |  |         'timeout': 3.0, | 
					
						
							|  |  |  |         'tokens': [], | 
					
						
							|  |  |  |     }, | 
					
						
							|  |  |  | ] | 
					
						
							| 
									
										
										
										
											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-09-22 16:22:22 +02:00
										 |  |  |         search_query = SearchQuery('test', [EngineRef(PUBLIC_ENGINE_NAME, 'general')], | 
					
						
							| 
									
										
										
										
											2020-09-22 16:55:59 +02:00
										 |  |  |                                    ['general'], 'en-US', SAFESEARCH, PAGENO, None, None) | 
					
						
							| 
									
										
										
										
											2019-08-02 13:50:51 +02:00
										 |  |  |         search = searx.search.Search(search_query) | 
					
						
							|  |  |  |         search.search() | 
					
						
							| 
									
										
										
										
											2020-08-06 17:42:46 +02:00
										 |  |  |         self.assertEqual(search.actual_timeout, 3.0) | 
					
						
							| 
									
										
										
										
											2019-08-02 13:50:51 +02:00
										 |  |  | 
 | 
					
						
							|  |  |  |     def test_timeout_query_above_default_nomax(self): | 
					
						
							|  |  |  |         searx.search.max_request_timeout = None | 
					
						
							| 
									
										
										
										
											2020-09-22 16:22:22 +02:00
										 |  |  |         search_query = SearchQuery('test', [EngineRef(PUBLIC_ENGINE_NAME, 'general')], | 
					
						
							| 
									
										
										
										
											2020-09-22 16:55:59 +02:00
										 |  |  |                                    ['general'], 'en-US', SAFESEARCH, PAGENO, None, 5.0) | 
					
						
							| 
									
										
										
										
											2019-08-02 13:50:51 +02:00
										 |  |  |         search = searx.search.Search(search_query) | 
					
						
							|  |  |  |         search.search() | 
					
						
							| 
									
										
										
										
											2020-08-06 17:42:46 +02:00
										 |  |  |         self.assertEqual(search.actual_timeout, 3.0) | 
					
						
							| 
									
										
										
										
											2019-08-02 13:50:51 +02:00
										 |  |  | 
 | 
					
						
							|  |  |  |     def test_timeout_query_below_default_nomax(self): | 
					
						
							|  |  |  |         searx.search.max_request_timeout = None | 
					
						
							| 
									
										
										
										
											2020-09-22 16:22:22 +02:00
										 |  |  |         search_query = SearchQuery('test', [EngineRef(PUBLIC_ENGINE_NAME, 'general')], | 
					
						
							| 
									
										
										
										
											2020-09-22 16:55:59 +02:00
										 |  |  |                                    ['general'], 'en-US', SAFESEARCH, PAGENO, None, 1.0) | 
					
						
							| 
									
										
										
										
											2019-08-02 13:50:51 +02:00
										 |  |  |         search = searx.search.Search(search_query) | 
					
						
							|  |  |  |         search.search() | 
					
						
							| 
									
										
										
										
											2020-08-06 17:42:46 +02:00
										 |  |  |         self.assertEqual(search.actual_timeout, 1.0) | 
					
						
							| 
									
										
										
										
											2019-08-02 13:50:51 +02:00
										 |  |  | 
 | 
					
						
							|  |  |  |     def test_timeout_query_below_max(self): | 
					
						
							|  |  |  |         searx.search.max_request_timeout = 10.0 | 
					
						
							| 
									
										
										
										
											2020-09-22 16:22:22 +02:00
										 |  |  |         search_query = SearchQuery('test', [EngineRef(PUBLIC_ENGINE_NAME, 'general')], | 
					
						
							| 
									
										
										
										
											2020-09-22 16:55:59 +02:00
										 |  |  |                                    ['general'], 'en-US', SAFESEARCH, PAGENO, None, 5.0) | 
					
						
							| 
									
										
										
										
											2019-08-02 13:50:51 +02:00
										 |  |  |         search = searx.search.Search(search_query) | 
					
						
							|  |  |  |         search.search() | 
					
						
							| 
									
										
										
										
											2020-08-06 17:42:46 +02:00
										 |  |  |         self.assertEqual(search.actual_timeout, 5.0) | 
					
						
							| 
									
										
										
										
											2019-08-02 13:50:51 +02:00
										 |  |  | 
 | 
					
						
							|  |  |  |     def test_timeout_query_above_max(self): | 
					
						
							|  |  |  |         searx.search.max_request_timeout = 10.0 | 
					
						
							| 
									
										
										
										
											2020-09-22 16:22:22 +02:00
										 |  |  |         search_query = SearchQuery('test', [EngineRef(PUBLIC_ENGINE_NAME, 'general')], | 
					
						
							| 
									
										
										
										
											2020-09-22 16:55:59 +02:00
										 |  |  |                                    ['general'], 'en-US', SAFESEARCH, PAGENO, None, 15.0) | 
					
						
							| 
									
										
										
										
											2019-08-02 13:50:51 +02:00
										 |  |  |         search = searx.search.Search(search_query) | 
					
						
							|  |  |  |         search.search() | 
					
						
							| 
									
										
										
										
											2020-08-06 17:42:46 +02:00
										 |  |  |         self.assertEqual(search.actual_timeout, 10.0) | 
					
						
							| 
									
										
										
										
											2020-02-01 11:01:17 +01:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-07-03 15:25:04 +02:00
										 |  |  |     def test_external_bang(self): | 
					
						
							| 
									
										
										
										
											2020-09-22 13:59:27 +02:00
										 |  |  |         search_query = SearchQuery('yes yes', | 
					
						
							| 
									
										
										
										
											2020-09-22 16:22:22 +02:00
										 |  |  |                                    [EngineRef(PUBLIC_ENGINE_NAME, 'general')], | 
					
						
							| 
									
										
										
										
											2020-09-22 13:59:27 +02:00
										 |  |  |                                    ['general'], 'en-US', SAFESEARCH, PAGENO, None, None, | 
					
						
							|  |  |  |                                    external_bang="yt") | 
					
						
							| 
									
										
										
										
											2020-07-03 15:25:04 +02:00
										 |  |  |         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) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-09-22 13:59:27 +02:00
										 |  |  |         search_query = SearchQuery('youtube never gonna give you up', | 
					
						
							| 
									
										
										
										
											2020-09-22 16:22:22 +02:00
										 |  |  |                                    [EngineRef(PUBLIC_ENGINE_NAME, 'general')], | 
					
						
							| 
									
										
										
										
											2020-09-22 16:55:59 +02:00
										 |  |  |                                    ['general'], 'en-US', SAFESEARCH, PAGENO, None, None) | 
					
						
							| 
									
										
										
										
											2020-07-03 15:25:04 +02:00
										 |  |  | 
 | 
					
						
							|  |  |  |         search = searx.search.Search(search_query) | 
					
						
							|  |  |  |         results = search.search() | 
					
						
							|  |  |  |         # This should not redirect | 
					
						
							|  |  |  |         self.assertTrue(results.redirect_url is None) |