test_youtube_noapi.py 7.5KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155
  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. class TestYoutubeNoAPIEngine(SearxTestCase):
  7. def test_request(self):
  8. query = 'test_query'
  9. dicto = defaultdict(dict)
  10. dicto['pageno'] = 0
  11. dicto['time_range'] = ''
  12. params = youtube_noapi.request(query, dicto)
  13. self.assertIn('url', params)
  14. self.assertIn(query, params['url'])
  15. self.assertIn('youtube.com', params['url'])
  16. def test_response(self):
  17. self.assertRaises(AttributeError, youtube_noapi.response, None)
  18. self.assertRaises(AttributeError, youtube_noapi.response, [])
  19. self.assertRaises(AttributeError, youtube_noapi.response, '')
  20. self.assertRaises(AttributeError, youtube_noapi.response, '[]')
  21. response = mock.Mock(text='<html></html>')
  22. self.assertEqual(youtube_noapi.response(response), [])
  23. html = """
  24. <ol id="item-section-063864" class="item-section">
  25. <li>
  26. <div class="yt-lockup yt-lockup-tile yt-lockup-video vve-check clearfix yt-uix-tile"
  27. data-context-item-id="DIVZCPfAOeM"
  28. data-visibility-tracking="CBgQ3DAYACITCPGXnYau6sUCFZEIHAod-VQASCj0JECx_-GK5uqMpcIB">
  29. <div class="yt-lockup-dismissable"><div class="yt-lockup-thumbnail contains-addto">
  30. <a aria-hidden="true" href="/watch?v=DIVZCPfAOeM" class=" yt-uix-sessionlink pf-link"
  31. data-sessionlink="itct=CBgQ3DAYACITCPGXnYau6sUCFZEIHAod-VQASCj0JFIEdGVzdA">
  32. <div class="yt-thumb video-thumb"><img src="//i.ytimg.com/vi/DIVZCPfAOeM/mqdefault.jpg"
  33. width="196" height="110"/></div><span class="video-time" aria-hidden="true">11:35</span></a>
  34. <span class="thumb-menu dark-overflow-action-menu video-actions">
  35. </span>
  36. </div>
  37. <div class="yt-lockup-content">
  38. <h3 class="yt-lockup-title">
  39. <a href="/watch?v=DIVZCPfAOeM"
  40. class="yt-uix-tile-link yt-ui-ellipsis yt-ui-ellipsis-2 yt-uix-sessionlink spf-link"
  41. data-sessionlink="itct=CBgQ3DAYACITCPGXnYau6sUCFZEIHAod-VQASCj0JFIEdGVzdA"
  42. title="Top Speed Test Kawasaki Ninja H2 (Thailand) By. MEHAY SUPERBIKE"
  43. aria-describedby="description-id-259079" rel="spf-prefetch" dir="ltr">
  44. Title
  45. </a>
  46. <span class="accessible-description" id="description-id-259079"> - Durée : 11:35.</span>
  47. </h3>
  48. <div class="yt-lockup-byline">de
  49. <a href="/user/mheejapan" class=" yt-uix-sessionlink spf-link g-hovercard"
  50. data-sessionlink="itct=CBgQ3DAYACITCPGXnYau6sUCFZEIHAod-VQASCj0JA" data-ytid="UCzEesu54Hjs0uRKmpy66qeA"
  51. data-name="">MEHAY SUPERBIKE</a></div><div class="yt-lockup-meta">
  52. <ul class="yt-lockup-meta-info">
  53. <li>il y a 20 heures</li>
  54. <li>8 424 vues</li>
  55. </ul>
  56. </div>
  57. <div class="yt-lockup-description yt-ui-ellipsis yt-ui-ellipsis-2" dir="ltr">
  58. Description
  59. </div>
  60. <div class="yt-lockup-badges">
  61. <ul class="yt-badge-list ">
  62. <li class="yt-badge-item" >
  63. <span class="yt-badge">Nouveauté</span>
  64. </li>
  65. <li class="yt-badge-item" ><span class="yt-badge " >HD</span></li>
  66. </ul>
  67. </div>
  68. <div class="yt-lockup-action-menu yt-uix-menu-container">
  69. <div class="yt-uix-menu yt-uix-videoactionmenu hide-until-delayloaded"
  70. data-video-id="DIVZCPfAOeM" data-menu-content-id="yt-uix-videoactionmenu-menu">
  71. </div>
  72. </div>
  73. </div>
  74. </div>
  75. </div>
  76. </li>
  77. </ol>
  78. """
  79. response = mock.Mock(text=html)
  80. results = youtube_noapi.response(response)
  81. self.assertEqual(type(results), list)
  82. self.assertEqual(len(results), 1)
  83. self.assertEqual(results[0]['title'], 'Title')
  84. self.assertEqual(results[0]['url'], 'https://www.youtube.com/watch?v=DIVZCPfAOeM')
  85. self.assertEqual(results[0]['content'], 'Description')
  86. self.assertEqual(results[0]['thumbnail'], 'https://i.ytimg.com/vi/DIVZCPfAOeM/hqdefault.jpg')
  87. self.assertTrue('DIVZCPfAOeM' in results[0]['embedded'])
  88. html = """
  89. <ol id="item-section-063864" class="item-section">
  90. <li>
  91. <div class="yt-lockup yt-lockup-tile yt-lockup-video vve-check clearfix yt-uix-tile"
  92. data-context-item-id="DIVZCPfAOeM"
  93. data-visibility-tracking="CBgQ3DAYACITCPGXnYau6sUCFZEIHAod-VQASCj0JECx_-GK5uqMpcIB">
  94. <div class="yt-lockup-dismissable"><div class="yt-lockup-thumbnail contains-addto">
  95. <a aria-hidden="true" href="/watch?v=DIVZCPfAOeM" class=" yt-uix-sessionlink pf-link"
  96. data-sessionlink="itct=CBgQ3DAYACITCPGXnYau6sUCFZEIHAod-VQASCj0JFIEdGVzdA">
  97. <div class="yt-thumb video-thumb"><img src="//i.ytimg.com/vi/DIVZCPfAOeM/mqdefault.jpg"
  98. width="196" height="110"/></div><span class="video-time" aria-hidden="true">11:35</span></a>
  99. <span class="thumb-menu dark-overflow-action-menu video-actions">
  100. </span>
  101. </div>
  102. <div class="yt-lockup-content">
  103. <h3 class="yt-lockup-title">
  104. <span class="accessible-description" id="description-id-259079"> - Durée : 11:35.</span>
  105. </h3>
  106. <div class="yt-lockup-byline">de
  107. <a href="/user/mheejapan" class=" yt-uix-sessionlink spf-link g-hovercard"
  108. data-sessionlink="itct=CBgQ3DAYACITCPGXnYau6sUCFZEIHAod-VQASCj0JA" data-ytid="UCzEesu54Hjs0uRKmpy66qeA"
  109. data-name="">MEHAY SUPERBIKE</a></div><div class="yt-lockup-meta">
  110. <ul class="yt-lockup-meta-info">
  111. <li>il y a 20 heures</li>
  112. <li>8 424 vues</li>
  113. </ul>
  114. </div>
  115. <div class="yt-lockup-badges">
  116. <ul class="yt-badge-list ">
  117. <li class="yt-badge-item" >
  118. <span class="yt-badge">Nouveauté</span>
  119. </li>
  120. <li class="yt-badge-item" ><span class="yt-badge " >HD</span></li>
  121. </ul>
  122. </div>
  123. <div class="yt-lockup-action-menu yt-uix-menu-container">
  124. <div class="yt-uix-menu yt-uix-videoactionmenu hide-until-delayloaded"
  125. data-video-id="DIVZCPfAOeM" data-menu-content-id="yt-uix-videoactionmenu-menu">
  126. </div>
  127. </div>
  128. </div>
  129. </div>
  130. </div>
  131. </li>
  132. </ol>
  133. """
  134. response = mock.Mock(text=html)
  135. results = youtube_noapi.response(response)
  136. self.assertEqual(type(results), list)
  137. self.assertEqual(len(results), 1)
  138. html = """
  139. <ol id="item-section-063864" class="item-section">
  140. <li>
  141. </li>
  142. </ol>
  143. """
  144. response = mock.Mock(text=html)
  145. results = youtube_noapi.response(response)
  146. self.assertEqual(type(results), list)
  147. self.assertEqual(len(results), 0)