mirror of
				https://github.com/searxng/searxng
				synced 2024-01-01 19:24:07 +01:00 
			
		
		
		
	[fix] engine unit tests.
Enables unit tests in the engines directory by adding __init__.py, and fixups for the enabled tests.
This commit is contained in:
		
							parent
							
								
									f6f622f7e5
								
							
						
					
					
						commit
						1f908a6222
					
				
					 4 changed files with 44 additions and 35 deletions
				
			
		
							
								
								
									
										2
									
								
								tests/unit/engines/__init__.py
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										2
									
								
								tests/unit/engines/__init__.py
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
					@ -0,0 +1,2 @@
 | 
				
			||||||
 | 
					# SPDX-License-Identifier: AGPL-3.0-or-later
 | 
				
			||||||
 | 
					# pylint: disable=missing-module-docstring
 | 
				
			||||||
| 
						 | 
					@ -21,7 +21,7 @@ from searx.engines import command as command_engine
 | 
				
			||||||
from tests import SearxTestCase
 | 
					from tests import SearxTestCase
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
class TestCommandEngine(SearxTestCase):
 | 
					class TestCommandEngine(SearxTestCase):  # pylint: disable=missing-class-docstring
 | 
				
			||||||
    def test_basic_seq_command_engine(self):
 | 
					    def test_basic_seq_command_engine(self):
 | 
				
			||||||
        ls_engine = command_engine
 | 
					        ls_engine = command_engine
 | 
				
			||||||
        ls_engine.command = ['seq', '{{QUERY}}']
 | 
					        ls_engine.command = ['seq', '{{QUERY}}']
 | 
				
			||||||
| 
						 | 
					@ -33,10 +33,10 @@ class TestCommandEngine(SearxTestCase):
 | 
				
			||||||
            {'number': '4', 'template': 'key-value.html'},
 | 
					            {'number': '4', 'template': 'key-value.html'},
 | 
				
			||||||
            {'number': '5', 'template': 'key-value.html'},
 | 
					            {'number': '5', 'template': 'key-value.html'},
 | 
				
			||||||
        ]
 | 
					        ]
 | 
				
			||||||
        results = ls_engine.search('5'.encode('utf-8'), {'pageno': 1})
 | 
					        results = ls_engine.search('5', {'pageno': 1})
 | 
				
			||||||
        self.assertEqual(results, expected_results)
 | 
					        self.assertEqual(results, expected_results)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def test_delimiter_parsing_command_engine(self):
 | 
					    def test_delimiter_parsing(self):
 | 
				
			||||||
        searx_logs = '''DEBUG:searx.webapp:static directory is /home/n/p/searx/searx/static
 | 
					        searx_logs = '''DEBUG:searx.webapp:static directory is /home/n/p/searx/searx/static
 | 
				
			||||||
DEBUG:searx.webapp:templates directory is /home/n/p/searx/searx/templates
 | 
					DEBUG:searx.webapp:templates directory is /home/n/p/searx/searx/templates
 | 
				
			||||||
DEBUG:searx.engines:soundcloud engine: Starting background initialization
 | 
					DEBUG:searx.engines:soundcloud engine: Starting background initialization
 | 
				
			||||||
| 
						 | 
					@ -140,10 +140,10 @@ INFO:werkzeug: * Debugger PIN: 299-578-362'''
 | 
				
			||||||
        ]
 | 
					        ]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        for i in [0, 1]:
 | 
					        for i in [0, 1]:
 | 
				
			||||||
            results = echo_engine.search(''.encode('utf-8'), {'pageno': i + 1})
 | 
					            results = echo_engine.search('', {'pageno': i + 1})
 | 
				
			||||||
            self.assertEqual(results, expected_results_by_page[i])
 | 
					            self.assertEqual(results, expected_results_by_page[i])
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def test_regex_parsing_command_engine(self):
 | 
					    def test_regex_parsing(self):
 | 
				
			||||||
        txt = '''commit 35f9a8c81d162a361b826bbcd4a1081a4fbe76a7
 | 
					        txt = '''commit 35f9a8c81d162a361b826bbcd4a1081a4fbe76a7
 | 
				
			||||||
Author: Noémi Ványi <sitbackandwait@gmail.com>
 | 
					Author: Noémi Ványi <sitbackandwait@gmail.com>
 | 
				
			||||||
Date:   Tue Oct 15 11:31:33 2019 +0200
 | 
					Date:   Tue Oct 15 11:31:33 2019 +0200
 | 
				
			||||||
| 
						 | 
					@ -168,11 +168,12 @@ commit '''
 | 
				
			||||||
        git_log_engine.result_separator = '\n\ncommit '
 | 
					        git_log_engine.result_separator = '\n\ncommit '
 | 
				
			||||||
        git_log_engine.delimiter = {}
 | 
					        git_log_engine.delimiter = {}
 | 
				
			||||||
        git_log_engine.parse_regex = {
 | 
					        git_log_engine.parse_regex = {
 | 
				
			||||||
            'commit': '\w{40}',
 | 
					            'commit': r'\w{40}',
 | 
				
			||||||
            'author': '[\w* ]* <\w*@?\w*\.?\w*>',
 | 
					            'author': r'[\w* ]* <\w*@?\w*\.?\w*>',
 | 
				
			||||||
            'date': 'Date: .*',
 | 
					            'date': r'Date: .*',
 | 
				
			||||||
            'message': '\n\n.*$',
 | 
					            'message': r'\n\n.*$',
 | 
				
			||||||
        }
 | 
					        }
 | 
				
			||||||
 | 
					        git_log_engine.init({"command": git_log_engine.command, "parse_regex": git_log_engine.parse_regex})
 | 
				
			||||||
        expected_results = [
 | 
					        expected_results = [
 | 
				
			||||||
            {
 | 
					            {
 | 
				
			||||||
                'commit': '35f9a8c81d162a361b826bbcd4a1081a4fbe76a7',
 | 
					                'commit': '35f9a8c81d162a361b826bbcd4a1081a4fbe76a7',
 | 
				
			||||||
| 
						 | 
					@ -197,7 +198,7 @@ commit '''
 | 
				
			||||||
            },
 | 
					            },
 | 
				
			||||||
        ]
 | 
					        ]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        results = git_log_engine.search(''.encode('utf-8'), {'pageno': 1})
 | 
					        results = git_log_engine.search('', {'pageno': 1})
 | 
				
			||||||
        self.assertEqual(results, expected_results)
 | 
					        self.assertEqual(results, expected_results)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def test_working_dir_path_query(self):
 | 
					    def test_working_dir_path_query(self):
 | 
				
			||||||
| 
						 | 
					@ -207,7 +208,7 @@ commit '''
 | 
				
			||||||
        ls_engine.delimiter = {'chars': ' ', 'keys': ['file']}
 | 
					        ls_engine.delimiter = {'chars': ' ', 'keys': ['file']}
 | 
				
			||||||
        ls_engine.query_type = 'path'
 | 
					        ls_engine.query_type = 'path'
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        results = ls_engine.search('.'.encode(), {'pageno': 1})
 | 
					        results = ls_engine.search('.', {'pageno': 1})
 | 
				
			||||||
        self.assertTrue(len(results) != 0)
 | 
					        self.assertTrue(len(results) != 0)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        forbidden_paths = [
 | 
					        forbidden_paths = [
 | 
				
			||||||
| 
						 | 
					@ -218,7 +219,7 @@ commit '''
 | 
				
			||||||
            '/var',
 | 
					            '/var',
 | 
				
			||||||
        ]
 | 
					        ]
 | 
				
			||||||
        for forbidden_path in forbidden_paths:
 | 
					        for forbidden_path in forbidden_paths:
 | 
				
			||||||
            self.assertRaises(ValueError, ls_engine.search, '..'.encode(), {'pageno': 1})
 | 
					            self.assertRaises(ValueError, ls_engine.search, forbidden_path, {'pageno': 1})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def test_enum_queries(self):
 | 
					    def test_enum_queries(self):
 | 
				
			||||||
        echo_engine = command_engine
 | 
					        echo_engine = command_engine
 | 
				
			||||||
| 
						 | 
					@ -227,7 +228,7 @@ commit '''
 | 
				
			||||||
        echo_engine.query_enum = ['i-am-allowed-to-say-this', 'and-that']
 | 
					        echo_engine.query_enum = ['i-am-allowed-to-say-this', 'and-that']
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        for allowed in echo_engine.query_enum:
 | 
					        for allowed in echo_engine.query_enum:
 | 
				
			||||||
            results = echo_engine.search(allowed.encode(), {'pageno': 1})
 | 
					            results = echo_engine.search(allowed, {'pageno': 1})
 | 
				
			||||||
            self.assertTrue(len(results) != 0)
 | 
					            self.assertTrue(len(results) != 0)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        forbidden_queries = [
 | 
					        forbidden_queries = [
 | 
				
			||||||
| 
						 | 
					@ -236,4 +237,4 @@ commit '''
 | 
				
			||||||
            'prohibited',
 | 
					            'prohibited',
 | 
				
			||||||
        ]
 | 
					        ]
 | 
				
			||||||
        for forbidden in forbidden_queries:
 | 
					        for forbidden in forbidden_queries:
 | 
				
			||||||
            self.assertRaises(ValueError, echo_engine.search, forbidden.encode(), {'pageno': 1})
 | 
					            self.assertRaises(ValueError, echo_engine.search, forbidden, {'pageno': 1})
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
| 
						 | 
					@ -7,41 +7,8 @@ from searx.engines import xpath
 | 
				
			||||||
from tests import SearxTestCase
 | 
					from tests import SearxTestCase
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
class TestXpathEngine(SearxTestCase):
 | 
					class TestXpathEngine(SearxTestCase):  # pylint: disable=missing-class-docstring
 | 
				
			||||||
    def test_request(self):
 | 
					    html = """
 | 
				
			||||||
        xpath.search_url = 'https://url.com/{query}'
 | 
					 | 
				
			||||||
        xpath.categories = []
 | 
					 | 
				
			||||||
        xpath.paging = False
 | 
					 | 
				
			||||||
        query = 'test_query'
 | 
					 | 
				
			||||||
        dicto = defaultdict(dict)
 | 
					 | 
				
			||||||
        params = xpath.request(query, dicto)
 | 
					 | 
				
			||||||
        self.assertIn('url', params)
 | 
					 | 
				
			||||||
        self.assertEquals('https://url.com/test_query', params['url'])
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        xpath.search_url = 'https://url.com/q={query}&p={pageno}'
 | 
					 | 
				
			||||||
        xpath.paging = True
 | 
					 | 
				
			||||||
        query = 'test_query'
 | 
					 | 
				
			||||||
        dicto = defaultdict(dict)
 | 
					 | 
				
			||||||
        dicto['pageno'] = 1
 | 
					 | 
				
			||||||
        params = xpath.request(query, dicto)
 | 
					 | 
				
			||||||
        self.assertIn('url', params)
 | 
					 | 
				
			||||||
        self.assertEquals('https://url.com/q=test_query&p=1', params['url'])
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    def test_response(self):
 | 
					 | 
				
			||||||
        # without results_xpath
 | 
					 | 
				
			||||||
        xpath.url_xpath = '//div[@class="search_result"]//a[@class="result"]/@href'
 | 
					 | 
				
			||||||
        xpath.title_xpath = '//div[@class="search_result"]//a[@class="result"]'
 | 
					 | 
				
			||||||
        xpath.content_xpath = '//div[@class="search_result"]//p[@class="content"]'
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        self.assertRaises(AttributeError, xpath.response, None)
 | 
					 | 
				
			||||||
        self.assertRaises(AttributeError, xpath.response, [])
 | 
					 | 
				
			||||||
        self.assertRaises(AttributeError, xpath.response, '')
 | 
					 | 
				
			||||||
        self.assertRaises(AttributeError, xpath.response, '[]')
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        response = mock.Mock(text='<html></html>')
 | 
					 | 
				
			||||||
        self.assertEqual(xpath.response(response), [])
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        html = u"""
 | 
					 | 
				
			||||||
    <div>
 | 
					    <div>
 | 
				
			||||||
        <div class="search_result">
 | 
					        <div class="search_result">
 | 
				
			||||||
            <a class="result" href="https://result1.com">Result 1</a>
 | 
					            <a class="result" href="https://result1.com">Result 1</a>
 | 
				
			||||||
| 
						 | 
					@ -55,7 +22,44 @@ class TestXpathEngine(SearxTestCase):
 | 
				
			||||||
        </div>
 | 
					        </div>
 | 
				
			||||||
    </div>
 | 
					    </div>
 | 
				
			||||||
    """
 | 
					    """
 | 
				
			||||||
        response = mock.Mock(text=html)
 | 
					
 | 
				
			||||||
 | 
					    def test_request(self):
 | 
				
			||||||
 | 
					        xpath.search_url = 'https://url.com/{query}'
 | 
				
			||||||
 | 
					        xpath.categories = []
 | 
				
			||||||
 | 
					        xpath.paging = False
 | 
				
			||||||
 | 
					        query = 'test_query'
 | 
				
			||||||
 | 
					        dicto = defaultdict(dict)
 | 
				
			||||||
 | 
					        dicto['language'] = 'all'
 | 
				
			||||||
 | 
					        dicto['pageno'] = 1
 | 
				
			||||||
 | 
					        params = xpath.request(query, dicto)
 | 
				
			||||||
 | 
					        self.assertIn('url', params)
 | 
				
			||||||
 | 
					        self.assertEqual('https://url.com/test_query', params['url'])
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        xpath.search_url = 'https://url.com/q={query}&p={pageno}'
 | 
				
			||||||
 | 
					        xpath.paging = True
 | 
				
			||||||
 | 
					        query = 'test_query'
 | 
				
			||||||
 | 
					        dicto = defaultdict(dict)
 | 
				
			||||||
 | 
					        dicto['language'] = 'all'
 | 
				
			||||||
 | 
					        dicto['pageno'] = 1
 | 
				
			||||||
 | 
					        params = xpath.request(query, dicto)
 | 
				
			||||||
 | 
					        self.assertIn('url', params)
 | 
				
			||||||
 | 
					        self.assertEqual('https://url.com/q=test_query&p=1', params['url'])
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_response(self):
 | 
				
			||||||
 | 
					        # without results_xpath
 | 
				
			||||||
 | 
					        xpath.url_xpath = '//div[@class="search_result"]//a[@class="result"]/@href'
 | 
				
			||||||
 | 
					        xpath.title_xpath = '//div[@class="search_result"]//a[@class="result"]'
 | 
				
			||||||
 | 
					        xpath.content_xpath = '//div[@class="search_result"]//p[@class="content"]'
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        self.assertRaises(AttributeError, xpath.response, None)
 | 
				
			||||||
 | 
					        self.assertRaises(AttributeError, xpath.response, [])
 | 
				
			||||||
 | 
					        self.assertRaises(AttributeError, xpath.response, '')
 | 
				
			||||||
 | 
					        self.assertRaises(AttributeError, xpath.response, '[]')
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        response = mock.Mock(text='<html></html>', status_code=200)
 | 
				
			||||||
 | 
					        self.assertEqual(xpath.response(response), [])
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        response = mock.Mock(text=self.html, status_code=200)
 | 
				
			||||||
        results = xpath.response(response)
 | 
					        results = xpath.response(response)
 | 
				
			||||||
        self.assertEqual(type(results), list)
 | 
					        self.assertEqual(type(results), list)
 | 
				
			||||||
        self.assertEqual(len(results), 2)
 | 
					        self.assertEqual(len(results), 2)
 | 
				
			||||||
| 
						 | 
					@ -80,6 +84,7 @@ class TestXpathEngine(SearxTestCase):
 | 
				
			||||||
        results = xpath.response(response)
 | 
					        results = xpath.response(response)
 | 
				
			||||||
        self.assertTrue(results[0]['is_onion'])
 | 
					        self.assertTrue(results[0]['is_onion'])
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def test_response_results_xpath(self):
 | 
				
			||||||
        # with results_xpath
 | 
					        # with results_xpath
 | 
				
			||||||
        xpath.results_xpath = '//div[@class="search_result"]'
 | 
					        xpath.results_xpath = '//div[@class="search_result"]'
 | 
				
			||||||
        xpath.url_xpath = './/a[@class="result"]/@href'
 | 
					        xpath.url_xpath = './/a[@class="result"]/@href'
 | 
				
			||||||
| 
						 | 
					@ -93,10 +98,10 @@ class TestXpathEngine(SearxTestCase):
 | 
				
			||||||
        self.assertRaises(AttributeError, xpath.response, '')
 | 
					        self.assertRaises(AttributeError, xpath.response, '')
 | 
				
			||||||
        self.assertRaises(AttributeError, xpath.response, '[]')
 | 
					        self.assertRaises(AttributeError, xpath.response, '[]')
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        response = mock.Mock(text='<html></html>')
 | 
					        response = mock.Mock(text='<html></html>', status_code=200)
 | 
				
			||||||
        self.assertEqual(xpath.response(response), [])
 | 
					        self.assertEqual(xpath.response(response), [])
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        response = mock.Mock(text=html)
 | 
					        response = mock.Mock(text=self.html, status_code=200)
 | 
				
			||||||
        results = xpath.response(response)
 | 
					        results = xpath.response(response)
 | 
				
			||||||
        self.assertEqual(type(results), list)
 | 
					        self.assertEqual(type(results), list)
 | 
				
			||||||
        self.assertEqual(len(results), 2)
 | 
					        self.assertEqual(len(results), 2)
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
| 
						 | 
					@ -247,6 +247,7 @@ class TestBang(SearxTestCase):  # pylint:disable=missing-class-docstring
 | 
				
			||||||
                self.assertEqual(query.user_query_parts, TestBang.THE_QUERY.split(' '))
 | 
					                self.assertEqual(query.user_query_parts, TestBang.THE_QUERY.split(' '))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    def test_specific(self):
 | 
					    def test_specific(self):
 | 
				
			||||||
 | 
					        load_engines(TEST_ENGINES)
 | 
				
			||||||
        for bang in TestBang.SPECIFIC_BANGS:
 | 
					        for bang in TestBang.SPECIFIC_BANGS:
 | 
				
			||||||
            with self.subTest(msg="Check bang is specific", bang=bang):
 | 
					            with self.subTest(msg="Check bang is specific", bang=bang):
 | 
				
			||||||
                query_text = TestBang.THE_QUERY + ' ' + bang
 | 
					                query_text = TestBang.THE_QUERY + ' ' + bang
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
		Loading…
	
	Add table
		
		Reference in a new issue