# -*- coding: utf-8 -*- import json from mock import Mock from searx import webapp from searx.testing import SearxTestCase from searx.search import Search from searx.url_utils import ParseResult class ViewsTestCase(SearxTestCase): def setUp(self): webapp.app.config['TESTING'] = True # to get better error messages self.app = webapp.app.test_client() # set some defaults self.test_results = [ { 'content': 'first test content', 'title': 'First Test', 'url': 'http://first.test.xyz', 'engines': ['youtube', 'startpage'], 'engine': 'startpage', 'parsed_url': ParseResult(scheme='http', netloc='first.test.xyz', path='/', params='', query='', fragment=''), # noqa }, { 'content': 'second test content', 'title': 'Second Test', 'url': 'http://second.test.xyz', 'engines': ['youtube', 'startpage'], 'engine': 'youtube', 'parsed_url': ParseResult(scheme='http', netloc='second.test.xyz', path='/', params='', query='', fragment=''), # noqa }, ] def search_mock(search_self, *args): search_self.result_container = Mock(get_ordered_results=lambda: self.test_results, answers=set(), corrections=set(), suggestions=set(), infoboxes=[], unresponsive_engines=set(), results=self.test_results, results_number=lambda: 3, results_length=lambda: len(self.test_results)) Search.search = search_mock def get_current_theme_name_mock(override=None): if override: return override return 'legacy' webapp.get_current_theme_name = get_current_theme_name_mock self.maxDiff = None # to see full diffs def test_index_empty(self): result = self.app.post('/') self.assertEqual(result.status_code, 200) self.assertIn(b'

searx

', result.data) def test_index_html(self): result = self.app.post('/', data={'q': 'test'}) self.assertIn( b'

youtubeSecond Test

', # noqa result.data ) self.assertIn( b'

first test content

', # noqa result.data ) def test_index_json(self): result = self.app.post('/', data={'q': 'test', 'format': 'json'}) result_dict = json.loads(result.data.decode('utf-8')) self.assertEqual('test', result_dict['query']) self.assertEqual(result_dict['results'][0]['content'], 'first test content') self.assertEqual(result_dict['results'][0]['url'], 'http://first.test.xyz') def test_index_csv(self): result = self.app.post('/', data={'q': 'test', 'format': 'csv'}) self.assertEqual( b'title,url,content,host,engine,score\r\n' b'First Test,http://first.test.xyz,first test content,first.test.xyz,startpage,\r\n' # noqa b'Second Test,http://second.test.xyz,second test content,second.test.xyz,youtube,\r\n', # noqa result.data ) def test_index_rss(self): result = self.app.post('/', data={'q': 'test', 'format': 'rss'}) self.assertIn( b'Search results for "test" - searx', result.data ) self.assertIn( b'3', result.data ) self.assertIn( b'First Test', result.data ) self.assertIn( b'http://first.test.xyz', result.data ) self.assertIn( b'first test content', result.data ) def test_about(self): result = self.app.get('/about') self.assertEqual(result.status_code, 200) self.assertIn(b'

About searx

', result.data) def test_preferences(self): result = self.app.get('/preferences') self.assertEqual(result.status_code, 200) self.assertIn( b'
', result.data ) self.assertIn( b'Default categories', result.data ) self.assertIn( b'Interface language', result.data ) def test_stats(self): result = self.app.get('/stats') self.assertEqual(result.status_code, 200) self.assertIn(b'

Engine stats

', result.data) def test_robots_txt(self): result = self.app.get('/robots.txt') self.assertEqual(result.status_code, 200) self.assertIn(b'Allow: /', result.data) def test_opensearch_xml(self): result = self.app.get('/opensearch.xml') self.assertEqual(result.status_code, 200) self.assertIn(b'a privacy-respecting, hackable metasearch engine', result.data) def test_favicon(self): result = self.app.get('/favicon.ico') self.assertEqual(result.status_code, 200) def test_config(self): result = self.app.get('/config') self.assertEqual(result.status_code, 200) json_result = result.get_json() self.assertTrue(json_result)