diff options
Diffstat (limited to 'tests/unit/test_webapp.py')
-rw-r--r-- | tests/unit/test_webapp.py | 120 |
1 files changed, 45 insertions, 75 deletions
diff --git a/tests/unit/test_webapp.py b/tests/unit/test_webapp.py index 43b631cb8..920a346a5 100644 --- a/tests/unit/test_webapp.py +++ b/tests/unit/test_webapp.py @@ -10,11 +10,11 @@ from tests import SearxTestCase class ViewsTestCase(SearxTestCase): - def setUp(self): # skip init function (no external HTTP request) def dummy(*args, **kwargs): pass + self.setattr4test(searx.search.processors, 'initialize_processor', dummy) from searx import webapp # pylint disable=import-outside-toplevel @@ -30,43 +30,39 @@ class ViewsTestCase(SearxTestCase): 'url': 'http://first.test.xyz', 'engines': ['youtube', 'startpage'], 'engine': 'startpage', - 'parsed_url': ParseResult(scheme='http', netloc='first.test.xyz', path='/', params='', query='', fragment=''), # noqa - }, { + '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 + 'parsed_url': ParseResult( + scheme='http', netloc='second.test.xyz', path='/', params='', query='', fragment='' + ), # noqa }, ] - timings = [ - { - 'engine': 'startpage', - 'total': 0.8, - 'load': 0.7 - }, - { - 'engine': 'youtube', - 'total': 0.9, - 'load': 0.6 - } - ] + timings = [{'engine': 'startpage', 'total': 0.8, 'load': 0.7}, {'engine': 'youtube', 'total': 0.9, 'load': 0.6}] def search_mock(search_self, *args): - search_self.result_container = Mock(get_ordered_results=lambda: test_results, - answers=dict(), - corrections=set(), - suggestions=set(), - infoboxes=[], - unresponsive_engines=set(), - results=test_results, - results_number=lambda: 3, - results_length=lambda: len(test_results), - get_timings=lambda: timings, - redirect_url=None, - engine_data={}) + search_self.result_container = Mock( + get_ordered_results=lambda: test_results, + answers=dict(), + corrections=set(), + suggestions=set(), + infoboxes=[], + unresponsive_engines=set(), + results=test_results, + results_number=lambda: 3, + results_length=lambda: len(test_results), + get_timings=lambda: timings, + redirect_url=None, + engine_data={}, + ) self.setattr4test(Search, 'search', search_mock) @@ -82,9 +78,12 @@ class ViewsTestCase(SearxTestCase): def test_index_empty(self): result = self.app.post('/') self.assertEqual(result.status_code, 200) - self.assertIn(b'<div class="text-hide center-block" id="main-logo">' - + b'<img class="center-block img-responsive" src="/static/themes/oscar/img/searxng.svg"' - + b' alt="searx logo" />SearXNG</div>', result.data) + self.assertIn( + b'<div class="text-hide center-block" id="main-logo">' + + b'<img class="center-block img-responsive" src="/static/themes/oscar/img/searxng.svg"' + + b' alt="searx logo" />SearXNG</div>', + result.data, + ) def test_index_html_post(self): result = self.app.post('/', data={'q': 'test'}) @@ -120,11 +119,10 @@ class ViewsTestCase(SearxTestCase): b'<h4 class="result_header" id="result-2"><img width="32" height="32" class="favicon"' + b' src="/static/themes/oscar/img/icons/youtube.png" alt="youtube" /><a href="http://second.test.xyz"' + b' rel="noreferrer" aria-labelledby="result-2">Second <span class="highlight">Test</span></a></h4>', # noqa - result.data + result.data, ) self.assertIn( - b'<p class="result-content">second <span class="highlight">test</span> content</p>', # noqa - result.data + b'<p class="result-content">second <span class="highlight">test</span> content</p>', result.data # noqa ) def test_index_json(self): @@ -151,7 +149,7 @@ class ViewsTestCase(SearxTestCase): b'title,url,content,host,engine,score,type\r\n' b'First Test,http://first.test.xyz,first test content,first.test.xyz,startpage,,result\r\n' # noqa b'Second Test,http://second.test.xyz,second test content,second.test.xyz,youtube,,result\r\n', # noqa - result.data + result.data, ) def test_index_rss(self): @@ -161,30 +159,15 @@ class ViewsTestCase(SearxTestCase): def test_search_rss(self): result = self.app.post('/search', data={'q': 'test', 'format': 'rss'}) - self.assertIn( - b'<description>Search results for "test" - searx</description>', - result.data - ) + self.assertIn(b'<description>Search results for "test" - searx</description>', result.data) - self.assertIn( - b'<opensearch:totalResults>3</opensearch:totalResults>', - result.data - ) + self.assertIn(b'<opensearch:totalResults>3</opensearch:totalResults>', result.data) - self.assertIn( - b'<title>First Test</title>', - result.data - ) + self.assertIn(b'<title>First Test</title>', result.data) - self.assertIn( - b'<link>http://first.test.xyz</link>', - result.data - ) + self.assertIn(b'<link>http://first.test.xyz</link>', result.data) - self.assertIn( - b'<description>first test content</description>', - result.data - ) + self.assertIn(b'<description>first test content</description>', result.data) def test_about(self): result = self.app.get('/about') @@ -199,18 +182,9 @@ class ViewsTestCase(SearxTestCase): def test_preferences(self): result = self.app.get('/preferences') self.assertEqual(result.status_code, 200) - self.assertIn( - b'<form method="post" action="/preferences" id="search_form">', - result.data - ) - self.assertIn( - b'<label class="col-sm-3 col-md-2" for="categories">Default categories</label>', - result.data - ) - self.assertIn( - b'<label class="col-sm-3 col-md-2" for="locale">Interface language</label>', - result.data - ) + self.assertIn(b'<form method="post" action="/preferences" id="search_form">', result.data) + self.assertIn(b'<label class="col-sm-3 col-md-2" for="categories">Default categories</label>', result.data) + self.assertIn(b'<label class="col-sm-3 col-md-2" for="locale">Interface language</label>', result.data) def test_browser_locale(self): result = self.app.get('/preferences', headers={'Accept-Language': 'zh-tw;q=0.8'}) @@ -218,30 +192,26 @@ class ViewsTestCase(SearxTestCase): self.assertIn( b'<option value="zh-Hant-TW" selected="selected">', result.data, - 'Interface locale ignored browser preference.' + 'Interface locale ignored browser preference.', ) self.assertIn( b'<option value="zh-Hant-TW" selected="selected">', result.data, - 'Search language ignored browser preference.' + 'Search language ignored browser preference.', ) def test_brower_empty_locale(self): result = self.app.get('/preferences', headers={'Accept-Language': ''}) self.assertEqual(result.status_code, 200) self.assertIn( - b'<option value="en" selected="selected">', - result.data, - 'Interface locale ignored browser preference.' + b'<option value="en" selected="selected">', result.data, 'Interface locale ignored browser preference.' ) def test_locale_occitan(self): result = self.app.get('/preferences?locale=oc') self.assertEqual(result.status_code, 200) self.assertIn( - b'<option value="oc" selected="selected">', - result.data, - 'Interface locale ignored browser preference.' + b'<option value="oc" selected="selected">', result.data, 'Interface locale ignored browser preference.' ) def test_stats(self): |