logo

searx

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

test_youtube_noapi.py (8353B)

    

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

  2. from collections import defaultdict

  3. import mock

  4. from searx.engines import youtube_noapi

  5. from searx.testing import SearxTestCase

  6. 

  7. 

  8. class TestYoutubeNoAPIEngine(SearxTestCase):

  9. 

  10.     def test_request(self):

  11.         query = 'test_query'

  12.         dicto = defaultdict(dict)

  13.         dicto['pageno'] = 0

  14.         dicto['time_range'] = ''

  15.         params = youtube_noapi.request(query, dicto)

  16.         self.assertIn('url', params)

  17.         self.assertIn(query, params['url'])

  18.         self.assertIn('youtube.com', params['url'])

  19. 

  20.     def test_time_range_search(self):

  21.         dicto = defaultdict(dict)

  22.         query = 'test_query'

  23.         dicto['time_range'] = 'year'

  24.         params = youtube_noapi.request(query, dicto)

  25.         self.assertIn('&sp=EgIIBQ%253D%253D', params['url'])

  26. 

  27.         dicto['time_range'] = 'month'

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

  29.         self.assertIn('&sp=EgIIBA%253D%253D', params['url'])

  30. 

  31.         dicto['time_range'] = 'week'

  32.         params = youtube_noapi.request(query, dicto)

  33.         self.assertIn('&sp=EgIIAw%253D%253D', params['url'])

  34. 

  35.         dicto['time_range'] = 'day'

  36.         params = youtube_noapi.request(query, dicto)

  37.         self.assertIn('&sp=EgIIAg%253D%253D', params['url'])

  38. 

  39.     def test_response(self):

  40.         self.assertRaises(AttributeError, youtube_noapi.response, None)

  41.         self.assertRaises(AttributeError, youtube_noapi.response, [])

  42.         self.assertRaises(AttributeError, youtube_noapi.response, '')

  43.         self.assertRaises(AttributeError, youtube_noapi.response, '[]')

  44. 

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

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

  47. 

  48.         html = """

  49.         <ol id="item-section-063864" class="item-section">

  50.             <li>

  51.                 <div class="yt-lockup yt-lockup-tile yt-lockup-video vve-check clearfix yt-uix-tile"

  52.                 data-context-item-id="DIVZCPfAOeM"

  53.                 data-visibility-tracking="CBgQ3DAYACITCPGXnYau6sUCFZEIHAod-VQASCj0JECx_-GK5uqMpcIB">

  54.                 <div class="yt-lockup-dismissable"><div class="yt-lockup-thumbnail contains-addto">

  55.                 <a aria-hidden="true" href="/watch?v=DIVZCPfAOeM" class=" yt-uix-sessionlink pf-link"

  56.                 data-sessionlink="itct=CBgQ3DAYACITCPGXnYau6sUCFZEIHAod-VQASCj0JFIEdGVzdA">

  57.                 <div class="yt-thumb video-thumb"><img src="//i.ytimg.com/vi/DIVZCPfAOeM/mqdefault.jpg"

  58.                 width="196" height="110"/></div><span class="video-time" aria-hidden="true">11:35</span></a>

  59.                 <span class="thumb-menu dark-overflow-action-menu video-actions">

  60.                 </span>

  61.                 </div>

  62.                 <div class="yt-lockup-content">

  63.                 <h3 class="yt-lockup-title">

  64.                 <a href="/watch?v=DIVZCPfAOeM"

  65.                 class="yt-uix-tile-link yt-ui-ellipsis yt-ui-ellipsis-2 yt-uix-sessionlink spf-link"

  66.                 data-sessionlink="itct=CBgQ3DAYACITCPGXnYau6sUCFZEIHAod-VQASCj0JFIEdGVzdA"

  67.                 title="Top Speed Test Kawasaki Ninja H2 (Thailand) By. MEHAY SUPERBIKE"

  68.                 aria-describedby="description-id-259079" rel="spf-prefetch" dir="ltr">

  69.                 Title

  70.                 </a>

  71.                 <span class="accessible-description" id="description-id-259079"> - Durée : 11:35.</span>

  72.                 </h3>

  73.                 <div class="yt-lockup-byline">de

  74.                 <a href="/user/mheejapan" class=" yt-uix-sessionlink spf-link g-hovercard"

  75.                 data-sessionlink="itct=CBgQ3DAYACITCPGXnYau6sUCFZEIHAod-VQASCj0JA" data-ytid="UCzEesu54Hjs0uRKmpy66qeA"

  76.                 data-name="">MEHAY SUPERBIKE</a></div><div class="yt-lockup-meta">

  77.                 <ul class="yt-lockup-meta-info">

  78.                     <li>il y a 20 heures</li>

  79.                     <li>8 424 vues</li>

  80.                 </ul>

  81.                 </div>

  82.                 <div class="yt-lockup-description yt-ui-ellipsis yt-ui-ellipsis-2" dir="ltr">

  83.                 Description

  84.                 </div>

  85.                 <div class="yt-lockup-badges">

  86.                 <ul class="yt-badge-list ">

  87.                     <li class="yt-badge-item" >

  88.                         <span class="yt-badge">Nouveauté</span>

  89.                     </li>

  90.                     <li class="yt-badge-item" ><span class="yt-badge " >HD</span></li>

  91.                 </ul>

  92.                 </div>

  93.                 <div class="yt-lockup-action-menu yt-uix-menu-container">

  94.                 <div class="yt-uix-menu yt-uix-videoactionmenu hide-until-delayloaded"

  95.                 data-video-id="DIVZCPfAOeM" data-menu-content-id="yt-uix-videoactionmenu-menu">

  96.                 </div>

  97.                 </div>

  98.                 </div>

  99.                 </div>

  100.                 </div>

  101.             </li>

  102.         </ol>

  103.         """

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

  105.         results = youtube_noapi.response(response)

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

  107.         self.assertEqual(len(results), 1)

  108.         self.assertEqual(results[0]['title'], 'Title')

  109.         self.assertEqual(results[0]['url'], 'https://www.youtube.com/watch?v=DIVZCPfAOeM')

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

  111.         self.assertEqual(results[0]['thumbnail'], 'https://i.ytimg.com/vi/DIVZCPfAOeM/hqdefault.jpg')

  112.         self.assertTrue('DIVZCPfAOeM' in results[0]['embedded'])

  113. 

  114.         html = """

  115.         <ol id="item-section-063864" class="item-section">

  116.             <li>

  117.                 <div class="yt-lockup yt-lockup-tile yt-lockup-video vve-check clearfix yt-uix-tile"

  118.                 data-context-item-id="DIVZCPfAOeM"

  119.                 data-visibility-tracking="CBgQ3DAYACITCPGXnYau6sUCFZEIHAod-VQASCj0JECx_-GK5uqMpcIB">

  120.                 <div class="yt-lockup-dismissable"><div class="yt-lockup-thumbnail contains-addto">

  121.                 <a aria-hidden="true" href="/watch?v=DIVZCPfAOeM" class=" yt-uix-sessionlink pf-link"

  122.                 data-sessionlink="itct=CBgQ3DAYACITCPGXnYau6sUCFZEIHAod-VQASCj0JFIEdGVzdA">

  123.                 <div class="yt-thumb video-thumb"><img src="//i.ytimg.com/vi/DIVZCPfAOeM/mqdefault.jpg"

  124.                 width="196" height="110"/></div><span class="video-time" aria-hidden="true">11:35</span></a>

  125.                 <span class="thumb-menu dark-overflow-action-menu video-actions">

  126.                 </span>

  127.                 </div>

  128.                 <div class="yt-lockup-content">

  129.                 <h3 class="yt-lockup-title">

  130.                 <span class="accessible-description" id="description-id-259079"> - Durée : 11:35.</span>

  131.                 </h3>

  132.                 <div class="yt-lockup-byline">de

  133.                 <a href="/user/mheejapan" class=" yt-uix-sessionlink spf-link g-hovercard"

  134.                 data-sessionlink="itct=CBgQ3DAYACITCPGXnYau6sUCFZEIHAod-VQASCj0JA" data-ytid="UCzEesu54Hjs0uRKmpy66qeA"

  135.                 data-name="">MEHAY SUPERBIKE</a></div><div class="yt-lockup-meta">

  136.                 <ul class="yt-lockup-meta-info">

  137.                     <li>il y a 20 heures</li>

  138.                     <li>8 424 vues</li>

  139.                 </ul>

  140.                 </div>

  141.                 <div class="yt-lockup-badges">

  142.                 <ul class="yt-badge-list ">

  143.                     <li class="yt-badge-item" >

  144.                         <span class="yt-badge">Nouveauté</span>

  145.                     </li>

  146.                     <li class="yt-badge-item" ><span class="yt-badge " >HD</span></li>

  147.                 </ul>

  148.                 </div>

  149.                 <div class="yt-lockup-action-menu yt-uix-menu-container">

  150.                 <div class="yt-uix-menu yt-uix-videoactionmenu hide-until-delayloaded"

  151.                 data-video-id="DIVZCPfAOeM" data-menu-content-id="yt-uix-videoactionmenu-menu">

  152.                 </div>

  153.                 </div>

  154.                 </div>

  155.                 </div>

  156.                 </div>

  157.             </li>

  158.         </ol>

  159.         """

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

  161.         results = youtube_noapi.response(response)

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

  163.         self.assertEqual(len(results), 1)

  164. 

  165.         html = """

  166.         <ol id="item-section-063864" class="item-section">

  167.             <li>

  168.             </li>

  169.         </ol>

  170.         """

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

  172.         results = youtube_noapi.response(response)

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

  174.         self.assertEqual(len(results), 0)