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
  8. class TestBingImagesEngine(SearxTestCase):
  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'])
  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'])
  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'])
  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, '[]')
  43.         response = mock.Mock(text='<html></html>')
  44.         self.assertEqual(bing_images.response(response), [])
  46.         response = mock.Mock(text='<html></html>')
  47.         self.assertEqual(bing_images.response(response), [])
  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')
  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)