logo

searx

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

test_bing_images.py (4561B)

    

  1. # -*- coding: utf-8 -*-

  2. from collections import defaultdict

  3. import mock

  4. from searx.engines import bing_images

  5. from searx.testing import SearxTestCase

  6. 

  7. 

  8. class TestBingImagesEngine(SearxTestCase):

  9. 

  10.     def test_request(self):

  11.         bing_images.supported_languages = ['fr-FR', 'en-US']

  12.         query = 'test_query'

  13.         dicto = defaultdict(dict)

  14.         dicto['pageno'] = 1

  15.         dicto['language'] = 'fr-FR'

  16.         dicto['safesearch'] = 1

  17.         dicto['time_range'] = ''

  18.         params = bing_images.request(query, dicto)

  19.         self.assertTrue('url' in params)

  20.         self.assertTrue(query in params['url'])

  21.         self.assertTrue('bing.com' in params['url'])

  22.         self.assertTrue('SRCHHPGUSR' in params['cookies'])

  23.         self.assertTrue('DEMOTE' in params['cookies']['SRCHHPGUSR'])

  24.         self.assertTrue('_EDGE_S' in params['cookies'])

  25.         self.assertTrue('fr-fr' in params['cookies']['_EDGE_S'])

  26. 

  27.         dicto['language'] = 'fr'

  28.         params = bing_images.request(query, dicto)

  29.         self.assertTrue('_EDGE_S' in params['cookies'])

  30.         self.assertTrue('fr-fr' in params['cookies']['_EDGE_S'])

  31. 

  32.         dicto['language'] = 'all'

  33.         params = bing_images.request(query, dicto)

  34.         self.assertTrue('_EDGE_S' in params['cookies'])

  35.         self.assertTrue('en-us' in params['cookies']['_EDGE_S'])

  36. 

  37.     def test_response(self):

  38.         self.assertRaises(AttributeError, bing_images.response, None)

  39.         self.assertRaises(AttributeError, bing_images.response, [])

  40.         self.assertRaises(AttributeError, bing_images.response, '')

  41.         self.assertRaises(AttributeError, bing_images.response, '[]')

  42. 

  43.         response = mock.Mock(text='<html></html>')

  44.         self.assertEqual(bing_images.response(response), [])

  45. 

  46.         response = mock.Mock(text='<html></html>')

  47.         self.assertEqual(bing_images.response(response), [])

  48. 

  49.         html = """

  50.         <div id="mmComponent_images_1">

  51.             <ul>

  52.                 <li>

  53.                     <div>

  54.                         <div class="imgpt">

  55.                             <a m='{"purl":"page_url","murl":"img_url","turl":"thumb_url"}'>

  56.                                 <img src="" alt="alt text" />

  57.                             </a>

  58.                         </div>

  59.                         <div></div>

  60.                     </div>

  61.                     <div>

  62.                         <div class="imgpt">

  63.                             <a m='{"purl":"page_url2","murl":"img_url2","turl":"thumb_url2"}'>

  64.                                 <img src="" alt="alt text 2" />

  65.                             </a>

  66.                         </div>

  67.                     </div>

  68.                 </li>

  69.             </ul>

  70.             <ul>

  71.                 <li>

  72.                     <div>

  73.                         <div class="imgpt">

  74.                             <a m='{"purl":"page_url3","murl":"img_url3","turl":"thumb_url3"}'>

  75.                                 <img src="" alt="alt text 3" />

  76.                             </a>

  77.                         </div>

  78.                     </div>

  79.                 </li>

  80.             </ul>

  81.         </div>

  82.         """

  83.         html = html.replace('\r\n', '').replace('\n', '').replace('\r', '')

  84.         response = mock.Mock(text=html)

  85.         results = bing_images.response(response)

  86.         self.assertEqual(type(results), list)

  87.         self.assertEqual(len(results), 3)

  88.         self.assertEqual(results[0]['title'], 'alt text')

  89.         self.assertEqual(results[0]['url'], 'page_url')

  90.         self.assertEqual(results[0]['content'], '')

  91.         self.assertEqual(results[0]['thumbnail_src'], 'thumb_url')

  92.         self.assertEqual(results[0]['img_src'], 'img_url')

  93. 

  94.     def test_fetch_supported_languages(self):

  95.         html = """

  96.         <div>

  97.             <div id="region-section-content">

  98.                 <ul class="b_vList">

  99.                     <li>

  100.                         <a href="https://bing...&setmkt=de-DE&s...">Germany</a>

  101.                         <a href="https://bing...&setmkt=nb-NO&s...">Norway</a>

  102.                     </li>

  103.                 </ul>

  104.                 <ul class="b_vList">

  105.                     <li>

  106.                         <a href="https://bing...&setmkt=es-AR&s...">Argentina</a>

  107.                     </li>

  108.                 </ul>

  109.             </div>

  110.         </div>

  111.         """

  112.         response = mock.Mock(text=html)

  113.         languages = list(bing_images._fetch_supported_languages(response))

  114.         self.assertEqual(len(languages), 3)

  115.         self.assertIn('de-DE', languages)

  116.         self.assertIn('no-NO', languages)

  117.         self.assertIn('es-AR', languages)