123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154 |
- # -*- coding: utf-8 -*-
- from collections import defaultdict
- import mock
- from searx.engines import youtube_noapi
- from searx.testing import SearxTestCase
-
-
- class TestYoutubeNoAPIEngine(SearxTestCase):
-
- def test_request(self):
- query = 'test_query'
- dicto = defaultdict(dict)
- dicto['pageno'] = 0
- params = youtube_noapi.request(query, dicto)
- self.assertIn('url', params)
- self.assertIn(query, params['url'])
- self.assertIn('youtube.com', params['url'])
-
- def test_response(self):
- self.assertRaises(AttributeError, youtube_noapi.response, None)
- self.assertRaises(AttributeError, youtube_noapi.response, [])
- self.assertRaises(AttributeError, youtube_noapi.response, '')
- self.assertRaises(AttributeError, youtube_noapi.response, '[]')
-
- response = mock.Mock(text='<html></html>')
- self.assertEqual(youtube_noapi.response(response), [])
-
- html = """
- <ol id="item-section-063864" class="item-section">
- <li>
- <div class="yt-lockup yt-lockup-tile yt-lockup-video vve-check clearfix yt-uix-tile"
- data-context-item-id="DIVZCPfAOeM"
- data-visibility-tracking="CBgQ3DAYACITCPGXnYau6sUCFZEIHAod-VQASCj0JECx_-GK5uqMpcIB">
- <div class="yt-lockup-dismissable"><div class="yt-lockup-thumbnail contains-addto">
- <a aria-hidden="true" href="/watch?v=DIVZCPfAOeM" class=" yt-uix-sessionlink pf-link"
- data-sessionlink="itct=CBgQ3DAYACITCPGXnYau6sUCFZEIHAod-VQASCj0JFIEdGVzdA">
- <div class="yt-thumb video-thumb"><img src="//i.ytimg.com/vi/DIVZCPfAOeM/mqdefault.jpg"
- width="196" height="110"/></div><span class="video-time" aria-hidden="true">11:35</span></a>
- <span class="thumb-menu dark-overflow-action-menu video-actions">
- </span>
- </div>
- <div class="yt-lockup-content">
- <h3 class="yt-lockup-title">
- <a href="/watch?v=DIVZCPfAOeM"
- class="yt-uix-tile-link yt-ui-ellipsis yt-ui-ellipsis-2 yt-uix-sessionlink spf-link"
- data-sessionlink="itct=CBgQ3DAYACITCPGXnYau6sUCFZEIHAod-VQASCj0JFIEdGVzdA"
- title="Top Speed Test Kawasaki Ninja H2 (Thailand) By. MEHAY SUPERBIKE"
- aria-describedby="description-id-259079" rel="spf-prefetch" dir="ltr">
- Title
- </a>
- <span class="accessible-description" id="description-id-259079"> - Durée : 11:35.</span>
- </h3>
- <div class="yt-lockup-byline">de
- <a href="/user/mheejapan" class=" yt-uix-sessionlink spf-link g-hovercard"
- data-sessionlink="itct=CBgQ3DAYACITCPGXnYau6sUCFZEIHAod-VQASCj0JA" data-ytid="UCzEesu54Hjs0uRKmpy66qeA"
- data-name="">MEHAY SUPERBIKE</a></div><div class="yt-lockup-meta">
- <ul class="yt-lockup-meta-info">
- <li>il y a 20 heures</li>
- <li>8 424 vues</li>
- </ul>
- </div>
- <div class="yt-lockup-description yt-ui-ellipsis yt-ui-ellipsis-2" dir="ltr">
- Description
- </div>
- <div class="yt-lockup-badges">
- <ul class="yt-badge-list ">
- <li class="yt-badge-item" >
- <span class="yt-badge">Nouveauté</span>
- </li>
- <li class="yt-badge-item" ><span class="yt-badge " >HD</span></li>
- </ul>
- </div>
- <div class="yt-lockup-action-menu yt-uix-menu-container">
- <div class="yt-uix-menu yt-uix-videoactionmenu hide-until-delayloaded"
- data-video-id="DIVZCPfAOeM" data-menu-content-id="yt-uix-videoactionmenu-menu">
- </div>
- </div>
- </div>
- </div>
- </div>
- </li>
- </ol>
- """
- response = mock.Mock(text=html)
- results = youtube_noapi.response(response)
- self.assertEqual(type(results), list)
- self.assertEqual(len(results), 1)
- self.assertEqual(results[0]['title'], 'Title')
- self.assertEqual(results[0]['url'], 'https://www.youtube.com/watch?v=DIVZCPfAOeM')
- self.assertEqual(results[0]['content'], 'Description')
- self.assertEqual(results[0]['thumbnail'], 'https://i.ytimg.com/vi/DIVZCPfAOeM/hqdefault.jpg')
- self.assertTrue('DIVZCPfAOeM' in results[0]['embedded'])
-
- html = """
- <ol id="item-section-063864" class="item-section">
- <li>
- <div class="yt-lockup yt-lockup-tile yt-lockup-video vve-check clearfix yt-uix-tile"
- data-context-item-id="DIVZCPfAOeM"
- data-visibility-tracking="CBgQ3DAYACITCPGXnYau6sUCFZEIHAod-VQASCj0JECx_-GK5uqMpcIB">
- <div class="yt-lockup-dismissable"><div class="yt-lockup-thumbnail contains-addto">
- <a aria-hidden="true" href="/watch?v=DIVZCPfAOeM" class=" yt-uix-sessionlink pf-link"
- data-sessionlink="itct=CBgQ3DAYACITCPGXnYau6sUCFZEIHAod-VQASCj0JFIEdGVzdA">
- <div class="yt-thumb video-thumb"><img src="//i.ytimg.com/vi/DIVZCPfAOeM/mqdefault.jpg"
- width="196" height="110"/></div><span class="video-time" aria-hidden="true">11:35</span></a>
- <span class="thumb-menu dark-overflow-action-menu video-actions">
- </span>
- </div>
- <div class="yt-lockup-content">
- <h3 class="yt-lockup-title">
- <span class="accessible-description" id="description-id-259079"> - Durée : 11:35.</span>
- </h3>
- <div class="yt-lockup-byline">de
- <a href="/user/mheejapan" class=" yt-uix-sessionlink spf-link g-hovercard"
- data-sessionlink="itct=CBgQ3DAYACITCPGXnYau6sUCFZEIHAod-VQASCj0JA" data-ytid="UCzEesu54Hjs0uRKmpy66qeA"
- data-name="">MEHAY SUPERBIKE</a></div><div class="yt-lockup-meta">
- <ul class="yt-lockup-meta-info">
- <li>il y a 20 heures</li>
- <li>8 424 vues</li>
- </ul>
- </div>
- <div class="yt-lockup-badges">
- <ul class="yt-badge-list ">
- <li class="yt-badge-item" >
- <span class="yt-badge">Nouveauté</span>
- </li>
- <li class="yt-badge-item" ><span class="yt-badge " >HD</span></li>
- </ul>
- </div>
- <div class="yt-lockup-action-menu yt-uix-menu-container">
- <div class="yt-uix-menu yt-uix-videoactionmenu hide-until-delayloaded"
- data-video-id="DIVZCPfAOeM" data-menu-content-id="yt-uix-videoactionmenu-menu">
- </div>
- </div>
- </div>
- </div>
- </div>
- </li>
- </ol>
- """
- response = mock.Mock(text=html)
- results = youtube_noapi.response(response)
- self.assertEqual(type(results), list)
- self.assertEqual(len(results), 1)
-
- html = """
- <ol id="item-section-063864" class="item-section">
- <li>
- </li>
- </ol>
- """
- response = mock.Mock(text=html)
- results = youtube_noapi.response(response)
- self.assertEqual(type(results), list)
- self.assertEqual(len(results), 0)
|