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

  6. 

  7. 

  8. class TestBingVideosEngine(SearxTestCase):

  9. 

  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'])

  26. 

  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'])

  35. 

  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, '[]')

  41. 

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

  43.         self.assertEqual(bing_videos.response(response), [])

  44. 

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

  46.         self.assertEqual(bing_videos.response(response), [])

  47. 

  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')