mirror of
				https://github.com/searxng/searxng
				synced 2024-01-01 19:24:07 +01:00 
			
		
		
		
	[fix] implement tests and remove usage of gen_useragent in engines
This commit is contained in:
		
							parent
							
								
									1f908a6222
								
							
						
					
					
						commit
						9a9ca307fe
					
				
					 3 changed files with 56 additions and 2 deletions
				
			
		| 
						 | 
					@ -6,7 +6,7 @@ from urllib.parse import urlencode
 | 
				
			||||||
from lxml import html
 | 
					from lxml import html
 | 
				
			||||||
 | 
					
 | 
				
			||||||
from dateutil.relativedelta import relativedelta
 | 
					from dateutil.relativedelta import relativedelta
 | 
				
			||||||
from searx.utils import eval_xpath, eval_xpath_list, extract_text, gen_useragent
 | 
					from searx.utils import eval_xpath, eval_xpath_list, extract_text
 | 
				
			||||||
 | 
					
 | 
				
			||||||
about = {
 | 
					about = {
 | 
				
			||||||
    'website': 'https://mojeek.com',
 | 
					    'website': 'https://mojeek.com',
 | 
				
			||||||
| 
						 | 
					@ -63,7 +63,6 @@ def request(query, params):
 | 
				
			||||||
        logger.debug(args["since"])
 | 
					        logger.debug(args["since"])
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    params['url'] = f"{base_url}/search?{urlencode(args)}"
 | 
					    params['url'] = f"{base_url}/search?{urlencode(args)}"
 | 
				
			||||||
    params['headers'] = {'User-Agent': gen_useragent()}
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
    return params
 | 
					    return params
 | 
				
			||||||
 | 
					
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
							
								
								
									
										2
									
								
								tests/unit/processors/__init__.py
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										2
									
								
								tests/unit/processors/__init__.py
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
					@ -0,0 +1,2 @@
 | 
				
			||||||
 | 
					# SPDX-License-Identifier: AGPL-3.0-or-later
 | 
				
			||||||
 | 
					# pylint: disable=missing-module-docstring
 | 
				
			||||||
							
								
								
									
										53
									
								
								tests/unit/processors/test_online.py
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										53
									
								
								tests/unit/processors/test_online.py
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
					@ -0,0 +1,53 @@
 | 
				
			||||||
 | 
					# SPDX-License-Identifier: AGPL-3.0-or-later
 | 
				
			||||||
 | 
					# pylint: disable=missing-module-docstring
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					from searx.search import SearchQuery, EngineRef
 | 
				
			||||||
 | 
					from searx.search.processors import online
 | 
				
			||||||
 | 
					from searx.engines import load_engines
 | 
				
			||||||
 | 
					from searx import engines
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					from tests import SearxTestCase
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					TEST_ENGINE_NAME = 'dummy engine'
 | 
				
			||||||
 | 
					TEST_ENGINE = {
 | 
				
			||||||
 | 
					    'name': TEST_ENGINE_NAME,
 | 
				
			||||||
 | 
					    'engine': 'dummy',
 | 
				
			||||||
 | 
					    'categories': 'general',
 | 
				
			||||||
 | 
					    'shortcut': 'du',
 | 
				
			||||||
 | 
					    'timeout': 3.0,
 | 
				
			||||||
 | 
					    'tokens': [],
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					class TestOnlineProcessor(SearxTestCase):  # pylint: disable=missing-class-docstring
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def setUp(self):
 | 
				
			||||||
 | 
					        load_engines([TEST_ENGINE])
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def tearDown(self):
 | 
				
			||||||
 | 
					        load_engines([])
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def _get_params(self, online_processor, search_query, engine_category):
 | 
				
			||||||
 | 
					        params = online_processor.get_params(search_query, engine_category)
 | 
				
			||||||
 | 
					        self.assertIsNotNone(params)
 | 
				
			||||||
 | 
					        assert params is not None
 | 
				
			||||||
 | 
					        return params
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_get_params_default_params(self):
 | 
				
			||||||
 | 
					        engine = engines.engines[TEST_ENGINE_NAME]
 | 
				
			||||||
 | 
					        online_processor = online.OnlineProcessor(engine, TEST_ENGINE_NAME)
 | 
				
			||||||
 | 
					        search_query = SearchQuery('test', [EngineRef(TEST_ENGINE_NAME, 'general')], 'all', 0, 1, None, None, None)
 | 
				
			||||||
 | 
					        params = self._get_params(online_processor, search_query, 'general')
 | 
				
			||||||
 | 
					        self.assertIn('method', params)
 | 
				
			||||||
 | 
					        self.assertIn('headers', params)
 | 
				
			||||||
 | 
					        self.assertIn('data', params)
 | 
				
			||||||
 | 
					        self.assertIn('url', params)
 | 
				
			||||||
 | 
					        self.assertIn('cookies', params)
 | 
				
			||||||
 | 
					        self.assertIn('auth', params)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_get_params_useragent(self):
 | 
				
			||||||
 | 
					        engine = engines.engines[TEST_ENGINE_NAME]
 | 
				
			||||||
 | 
					        online_processor = online.OnlineProcessor(engine, TEST_ENGINE_NAME)
 | 
				
			||||||
 | 
					        search_query = SearchQuery('test', [EngineRef(TEST_ENGINE_NAME, 'general')], 'all', 0, 1, None, None, None)
 | 
				
			||||||
 | 
					        params = self._get_params(online_processor, search_query, 'general')
 | 
				
			||||||
 | 
					        self.assertIn('User-Agent', params['headers'])
 | 
				
			||||||
		Loading…
	
	Add table
		
		Reference in a new issue