logo

searx

My custom branche(s) on searx, a meta-search engine git clone https://hacktivis.me/git/searx.git

test_duckduckgo_images.py (2990B)

    

  1. # -*- coding: utf-8 -*-
  2. from collections import defaultdict
  3. import mock
  4. from searx.engines import duckduckgo_images
  5. from searx.testing import SearxTestCase
  8. class TestDuckduckgoImagesEngine(SearxTestCase):
  10.     def test_request(self):
  11.         duckduckgo_images.supported_languages = ['de-CH', 'en-US']
  12.         query = 'test_query'
  13.         dicto = defaultdict(dict)
  14.         dicto['is_test'] = True
  15.         dicto['pageno'] = 1
  16.         dicto['safesearch'] = 0
  17.         dicto['language'] = 'en-US'
  18.         params = duckduckgo_images.request(query, dicto)
  19.         self.assertIn('url', params)
  20.         self.assertIn(query, params['url'])
  21.         self.assertIn('duckduckgo.com', params['url'])
  22.         self.assertIn('s=0', params['url'])
  23.         self.assertIn('p=-1', params['url'])
  24.         self.assertIn('vqd=12345', params['url'])
  26.         # test paging, safe search and language
  27.         dicto['pageno'] = 2
  28.         dicto['safesearch'] = 2
  29.         dicto['language'] = 'de'
  30.         params = duckduckgo_images.request(query, dicto)
  31.         self.assertIn('url', params)
  32.         self.assertIn(query, params['url'])
  33.         self.assertIn('s=50', params['url'])
  34.         self.assertIn('p=1', params['url'])
  35.         self.assertIn('ch-de', params['url'])
  37.     def test_response(self):
  38.         self.assertRaises(AttributeError, duckduckgo_images.response, None)
  39.         self.assertRaises(AttributeError, duckduckgo_images.response, [])
  40.         self.assertRaises(AttributeError, duckduckgo_images.response, '')
  41.         self.assertRaises(AttributeError, duckduckgo_images.response, '[]')
  43.         response = mock.Mock(text='If this error persists, please let us know: ops@duckduckgo.com')
  44.         self.assertEqual(duckduckgo_images.response(response), [])
  46.         json = u"""
  47.         {
  48.             "query": "test_query",
  49.             "results": [
  50.                 {
  51.                     "title": "Result 1",
  52.                     "url": "https://site1.url",
  53.                     "thumbnail": "https://thumb1.nail",
  54.                     "image": "https://image1"
  55.                 },
  56.                 {
  57.                     "title": "Result 2",
  58.                     "url": "https://site2.url",
  59.                     "thumbnail": "https://thumb2.nail",
  60.                     "image": "https://image2"
  61.                 }
  62.             ]
  63.         }
  64.         """
  65.         response = mock.Mock(text=json)
  66.         results = duckduckgo_images.response(response)
  67.         self.assertEqual(len(results), 2)
  68.         self.assertEqual(results[0]['title'], 'Result 1')
  69.         self.assertEqual(results[0]['url'], 'https://site1.url')
  70.         self.assertEqual(results[0]['thumbnail_src'], 'https://thumb1.nail')
  71.         self.assertEqual(results[0]['img_src'], 'https://image1')
  72.         self.assertEqual(results[1]['title'], 'Result 2')
  73.         self.assertEqual(results[1]['url'], 'https://site2.url')
  74.         self.assertEqual(results[1]['thumbnail_src'], 'https://thumb2.nail')
  75.         self.assertEqual(results[1]['img_src'], 'https://image2')