fjmwkj
/
searxng
mirror of https://github.com/searxng/searxng


			
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124
							# -*- 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
        dicto['time_range'] = ''
        params = youtube_noapi.request(query, dicto)
        self.assertIn('url', params)
        self.assertIn(query, params['url'])
        self.assertIn('youtube.com', params['url'])

    def test_time_range_search(self):
        dicto = defaultdict(dict)
        query = 'test_query'
        dicto['time_range'] = 'year'
        params = youtube_noapi.request(query, dicto)
        self.assertIn('&sp=EgIIBQ%253D%253D', params['url'])

        dicto['time_range'] = 'month'
        params = youtube_noapi.request(query, dicto)
        self.assertIn('&sp=EgIIBA%253D%253D', params['url'])

        dicto['time_range'] = 'week'
        params = youtube_noapi.request(query, dicto)
        self.assertIn('&sp=EgIIAw%253D%253D', params['url'])

        dicto['time_range'] = 'day'
        params = youtube_noapi.request(query, dicto)
        self.assertIn('&sp=EgIIAg%253D%253D', 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 = """
        <div></div>
        <script>
            window["ytInitialData"] = {
                "contents": {
                    "twoColumnSearchResultsRenderer": {
                        "primaryContents": {
                            "sectionListRenderer": {
                                "contents": [
                                    {
                                        "itemSectionRenderer": {
                                            "contents": [
                                                {
                                                    "videoRenderer": {
                                                        "videoId": "DIVZCPfAOeM",
                                                        "title": {
                                                            "simpleText": "Title"
                                                        },
                                                        "descriptionSnippet": {
                                                            "runs": [
                                                                {
                                                                    "text": "Des"
                                                                },
                                                                {
                                                                    "text": "cription"
                                                                }
                                                            ]
                                                        }
                                                    }
                                                },
                                                {
                                                    "videoRenderer": {
                                                        "videoId": "9C_HReR_McQ",
                                                        "title": {
                                                            "simpleText": "Title"
                                                        },
                                                        "descriptionSnippet": {
                                                            "simpleText": "Description"
                                                        }
                                                    }
                                                }
                                            ]
                                        }
                                    }
                                ]
                            }
                        }
                    }
                }
            };
        </script>
        """
        response = mock.Mock(text=html)
        results = youtube_noapi.response(response)
        self.assertEqual(type(results), list)
        self.assertEqual(len(results), 2)
        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'])
        self.assertEqual(results[1]['title'], 'Title')
        self.assertEqual(results[1]['url'], 'https://www.youtube.com/watch?v=9C_HReR_McQ')
        self.assertEqual(results[1]['content'], 'Description')
        self.assertEqual(results[1]['thumbnail'], 'https://i.ytimg.com/vi/9C_HReR_McQ/hqdefault.jpg')
        self.assertTrue('9C_HReR_McQ' in results[1]['embedded'])

        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)