logo

searx

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

test_bing_videos.py (4335B)

    

  1. # -*- coding: utf-8 -*-
  2. from collections import defaultdict
  3. import mock
  4. from searx.engines import bing_videos
  5. from searx.testing import SearxTestCase
  8. class TestBingVideosEngine(SearxTestCase):
  10.     def test_request(self):
  11.         bing_videos.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'] = 0
  17.         dicto['time_range'] = ''
  18.         params = bing_videos.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('OFF' in params['cookies']['SRCHHPGUSR'])
  24.         self.assertTrue('_EDGE_S' in params['cookies'])
  25.         self.assertTrue('fr-fr' in params['cookies']['_EDGE_S'])
  27.         dicto['pageno'] = 2
  28.         dicto['time_range'] = 'day'
  29.         dicto['safesearch'] = 2
  30.         params = bing_videos.request(query, dicto)
  31.         self.assertTrue('first=11' in params['url'])
  32.         self.assertTrue('1440' in params['url'])
  33.         self.assertIn('SRCHHPGUSR', params['cookies'])
  34.         self.assertTrue('STRICT' in params['cookies']['SRCHHPGUSR'])
  36.     def test_response(self):
  37.         self.assertRaises(AttributeError, bing_videos.response, None)
  38.         self.assertRaises(AttributeError, bing_videos.response, [])
  39.         self.assertRaises(AttributeError, bing_videos.response, '')
  40.         self.assertRaises(AttributeError, bing_videos.response, '[]')
  42.         response = mock.Mock(text='<html></html>')
  43.         self.assertEqual(bing_videos.response(response), [])
  45.         response = mock.Mock(text='<html></html>')
  46.         self.assertEqual(bing_videos.response(response), [])
  48.         html = """
  49.         <div class="dg_u">
  50.             <div id="mc_vtvc_1" class="mc_vtvc">
  51.                 <a class="mc_vtvc_link" href="/video">
  52.                     <div class="mc_vtvc_th">
  53.                         <div class="cico">
  54.                             <img src="thumb_1.jpg" />
  55.                         </div>
  56.                         <div class="mc_vtvc_ban_lo">
  57.                             <div class="vtbc">
  58.                                 <div class="mc_bc_w b_smText">
  59.                                     <div class="mc_bc pivot bpi_2">
  60.                                         <span title="">
  61.                                              <span class="mv_vtvc_play cipg "></span>
  62.                                         </span>
  63.                                     </div>
  64.                                     <div class="mc_bc items">10:06</div>
  65.                                 </div>
  66.                             </div>
  67.                         </div>
  68.                         </div>
  69.                         <div class="mc_vtvc_meta">
  70.                         <div class="mc_vtvc_title" title="Title 1"></div>
  71.                         <div class="mc_vtvc_meta_block_area">
  72.                         <div class="mc_vtvc_meta_block">
  73.                             <div class="mc_vtvc_meta_row">
  74.                                 <span>65,696,000+ views</span>
  75.                                 <span>1 year ago</span>
  76.                             </div>
  77.                             <div class="mc_vtvc_meta_row mc_vtvc_meta_row_channel">Content 1</div>
  78.                             <div class="mc_vtvc_meta_row"><span>
  79.                                 <div class="cico mc_vtvc_src_ico">
  80.                                     <div></div>
  81.                                 </div>
  82.                                 <span>YouTube</span>
  83.                             </span></div>
  84.                         </div>
  85.                         </div>
  86.                     </div>
  87.                     <div class="vrhdata"></div>
  88.                     </a>
  89.                 </div>
  90.             </div>
  91.         """
  92.         response = mock.Mock(text=html)
  93.         results = bing_videos.response(response)
  94.         self.assertEqual(type(results), list)
  95.         self.assertEqual(len(results), 1)
  96.         self.assertEqual(results[0]['title'], 'Title 1')
  97.         self.assertEqual(results[0]['url'], 'https://bing.com/video')
  98.         self.assertEqual(results[0]['content'], 'Content 1')
  99.         self.assertEqual(results[0]['thumbnail'], 'thumb_1.jpg')