@@ -11,15 +11,19 @@ class TestQwantEngine(SearxTestCase):
11 dicto = defaultdict(dict)
11 dicto = defaultdict(dict)
12 dicto['pageno'] = 0
12 dicto['pageno'] = 0
13 dicto['language'] = 'fr_FR'
13 dicto['language'] = 'fr_FR'
14+ qwant.categories = ['']
14 params = qwant.request(query, dicto)
15 params = qwant.request(query, dicto)
15 self.assertIn('url', params)
16 self.assertIn('url', params)
16 self.assertIn(query, params['url'])
17 self.assertIn(query, params['url'])
18+ self.assertIn('web', params['url'])
17 self.assertIn('qwant.com', params['url'])
19 self.assertIn('qwant.com', params['url'])
18 self.assertIn('fr_fr', params['url'])
20 self.assertIn('fr_fr', params['url'])
20 dicto['language'] = 'all'
22 dicto['language'] = 'all'
23+ qwant.categories = ['news']
21 params = qwant.request(query, dicto)
24 params = qwant.request(query, dicto)
22 self.assertFalse('fr' in params['url'])
25 self.assertFalse('fr' in params['url'])
26+ self.assertIn('news', params['url'])
24 def test_response(self):
28 def test_response(self):
25 self.assertRaises(AttributeError, qwant.response, None)
29 self.assertRaises(AttributeError, qwant.response, None)
@@ -68,7 +72,7 @@ class TestQwantEngine(SearxTestCase):
70 response = mock.Mock(text=json)
74 response = mock.Mock(text=json)
71− qwant.search_url_keyword = 'web'
75+ qwant.categories = ['general']
72 results = qwant.response(response)
76 results = qwant.response(response)
73 self.assertEqual(type(results), list)
77 self.assertEqual(type(results), list)
74 self.assertEqual(len(results), 1)
78 self.assertEqual(len(results), 1)
@@ -113,7 +117,7 @@ class TestQwantEngine(SearxTestCase):
115 response = mock.Mock(text=json)
119 response = mock.Mock(text=json)
116− qwant.search_url_keyword = 'images'
120+ qwant.categories = ['images']
117 results = qwant.response(response)
121 results = qwant.response(response)
118 self.assertEqual(type(results), list)
122 self.assertEqual(type(results), list)
119 self.assertEqual(len(results), 1)
123 self.assertEqual(len(results), 1)
@@ -158,7 +162,7 @@ class TestQwantEngine(SearxTestCase):
160 response = mock.Mock(text=json)
164 response = mock.Mock(text=json)
161− qwant.search_url_keyword = 'news'
165+ qwant.categories = ['news']
162 results = qwant.response(response)
166 results = qwant.response(response)
163 self.assertEqual(type(results), list)
167 self.assertEqual(type(results), list)
164 self.assertEqual(len(results), 1)
168 self.assertEqual(len(results), 1)
@@ -202,7 +206,7 @@ class TestQwantEngine(SearxTestCase):
204 response = mock.Mock(text=json)
208 response = mock.Mock(text=json)
205− qwant.search_url_keyword = 'social'
209+ qwant.categories = ['social media']
206 results = qwant.response(response)
210 results = qwant.response(response)
207 self.assertEqual(type(results), list)
211 self.assertEqual(type(results), list)
208 self.assertEqual(len(results), 1)
212 self.assertEqual(len(results), 1)
@@ -246,7 +250,7 @@ class TestQwantEngine(SearxTestCase):
248 response = mock.Mock(text=json)
252 response = mock.Mock(text=json)
249− qwant.search_url_keyword = ''
253+ qwant.categories = ['']
250 results = qwant.response(response)
254 results = qwant.response(response)
251 self.assertEqual(type(results), list)
255 self.assertEqual(type(results), list)
252 self.assertEqual(len(results), 0)
256 self.assertEqual(len(results), 0)