123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158 |
- # -*- 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=[],
- 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'<div class="title"><h1>searx</h1></div>', result.data)
-
- def test_index_html(self):
- result = self.app.post('/', data={'q': 'test'})
- self.assertIn(
- b'<h3 class="result_title"><img width="14" height="14" class="favicon" src="/static/themes/legacy/img/icons/icon_youtube.ico" alt="youtube" /><a href="http://second.test.xyz" rel="noreferrer">Second <span class="highlight">Test</span></a></h3>', # noqa
- result.data
- )
- self.assertIn(
- b'<p class="content">first <span class="highlight">test</span> content<br class="last"/></p>', # 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'<description>Search results for "test" - searx</description>',
- result.data
- )
-
- self.assertIn(
- b'<opensearch:totalResults>3</opensearch:totalResults>',
- 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'<description>first test content</description>',
- result.data
- )
-
- def test_about(self):
- result = self.app.get('/about')
- self.assertEqual(result.status_code, 200)
- self.assertIn(b'<h1>About <a href="/">searx</a></h1>', result.data)
-
- 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'<legend>Default categories</legend>',
- result.data
- )
- self.assertIn(
- b'<legend>Interface language</legend>',
- result.data
- )
-
- def test_stats(self):
- result = self.app.get('/stats')
- self.assertEqual(result.status_code, 200)
- self.assertIn(b'<h2>Engine stats</h2>', 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'<Description>a privacy-respecting, hackable metasearch engine</Description>', result.data)
-
- def test_favicon(self):
- result = self.app.get('/favicon.ico')
- self.assertEqual(result.status_code, 200)
|