@@ -9,20 +9,15 @@ from searx.testing import SearxTestCase
9class TestWikidataEngine(SearxTestCase):
9class TestWikidataEngine(SearxTestCase):
11 def test_request(self):
11 def test_request(self):
12− wikidata.supported_languages = ['en', 'es']
13 query = 'test_query'
12 query = 'test_query'
14 dicto = defaultdict(dict)
13 dicto = defaultdict(dict)
15− dicto['language'] = 'en-US'
16 params = wikidata.request(query, dicto)
14 params = wikidata.request(query, dicto)
17 self.assertIn('url', params)
15 self.assertIn('url', params)
18 self.assertIn(query, params['url'])
16 self.assertIn(query, params['url'])
19 self.assertIn('wikidata.org', params['url'])
17 self.assertIn('wikidata.org', params['url'])
20− self.assertIn('en', params['url'])
22− dicto['language'] = 'es-ES'
23 params = wikidata.request(query, dicto)
19 params = wikidata.request(query, dicto)
24 self.assertIn(query, params['url'])
20 self.assertIn(query, params['url'])
25− self.assertIn('es', params['url'])
27 # successful cases are not tested here to avoid sending additional requests
22 # successful cases are not tested here to avoid sending additional requests
28 def test_response(self):
23 def test_response(self):
@@ -31,6 +26,7 @@ class TestWikidataEngine(SearxTestCase):
31 self.assertRaises(AttributeError, wikidata.response, '')
26 self.assertRaises(AttributeError, wikidata.response, '')
32 self.assertRaises(AttributeError, wikidata.response, '[]')
27 self.assertRaises(AttributeError, wikidata.response, '[]')
29+ wikidata.supported_languages = ['en', 'es']
34 response = mock.Mock(text='<html></html>', search_params={"language": "en"})
30 response = mock.Mock(text='<html></html>', search_params={"language": "en"})
35 self.assertEqual(wikidata.response(response), [])
31 self.assertEqual(wikidata.response(response), [])