commit: 4dba3739fb3b98572cbd51adab226376b5844105
parent 8f040e30adbbd615155a5075bec28ccadff10eff
Author: Cqoicebordel <Cqoicebordel@users.noreply.github.com>
Date: Mon, 26 Jan 2015 18:24:08 +0100
Youtube's unit test
Diffstat:
3 files changed, 207 insertions(+), 2 deletions(-)
diff --git a/searx/engines/youtube.py b/searx/engines/youtube.py
@@ -57,7 +57,7 @@ def response(resp):
url = [x['href'] for x in result['link'] if x['type'] == 'text/html']
if not url:
- return
+ continue
# remove tracking
url = url[0].replace('feature=youtube_gdata', '')
@@ -73,7 +73,7 @@ def response(resp):
pubdate = result['published']['$t']
publishedDate = parser.parse(pubdate)
- if result['media$group']['media$thumbnail']:
+ if 'media$thumbnail' in result['media$group']:
thumbnail = result['media$group']['media$thumbnail'][0]['url']
content = result['content']['$t']
diff --git a/searx/tests/engines/test_youtube.py b/searx/tests/engines/test_youtube.py
@@ -0,0 +1,204 @@
+from collections import defaultdict
+import mock
+from searx.engines import youtube
+from searx.testing import SearxTestCase
+
+
+class TestYoutubeEngine(SearxTestCase):
+
+ def test_request(self):
+ query = 'test_query'
+ dicto = defaultdict(dict)
+ dicto['pageno'] = 0
+ dicto['language'] = 'fr_FR'
+ params = youtube.request(query, dicto)
+ self.assertTrue('url' in params)
+ self.assertTrue(query in params['url'])
+ self.assertTrue('youtube.com' in params['url'])
+ self.assertTrue('fr' in params['url'])
+
+ dicto['language'] = 'all'
+ params = youtube.request(query, dicto)
+ self.assertFalse('fr' in params['url'])
+
+ def test_response(self):
+ self.assertRaises(AttributeError, youtube.response, None)
+ self.assertRaises(AttributeError, youtube.response, [])
+ self.assertRaises(AttributeError, youtube.response, '')
+ self.assertRaises(AttributeError, youtube.response, '[]')
+
+ response = mock.Mock(text='{}')
+ self.assertEqual(youtube.response(response), [])
+
+ response = mock.Mock(text='{"data": []}')
+ self.assertEqual(youtube.response(response), [])
+
+ json = """
+ {"feed":{"entry":[{
+ "id":{"$t":"http://gdata.youtube.com/feeds/api/videos/DIVZCPfAOeM"},
+ "published":{"$t":"2015-01-23T21:25:00.000Z"},
+ "updated":{"$t":"2015-01-26T14:38:15.000Z"},
+ "title":{"$t":"Title",
+ "type":"text"},"content":{"$t":"Description","type":"text"},
+ "link":[{"rel":"alternate","type":"text/html",
+ "href":"https://www.youtube.com/watch?v=DIVZCPfAOeM&feature=youtube_gdata"},
+ {"rel":"http://gdata.youtube.com/schemas/2007#video.related",
+ "type":"application/atom+xml",
+ "href":"https://gdata.youtube.com/feeds/api/videos/DIVZCPfAOeM/related"},
+ {"rel":"http://gdata.youtube.com/schemas/2007#mobile","type":"text/html",
+ "href":"https://m.youtube.com/details?v=DIVZCPfAOeM"},
+ {"rel":"self","type":"application/atom+xml",
+ "href":"https://gdata.youtube.com/feeds/api/videos/DIVZCPfAOeM"}],
+ "author":[{"name":{"$t":"Cauet"},
+ "uri":{"$t":"https://gdata.youtube.com/feeds/api/users/cauetofficiel"} }],
+ "gd$comments":{"gd$feedLink":{"rel":"http://gdata.youtube.com/schemas/2007#comments",
+ "href":"https://gdata.youtube.com/feeds/api/videos/DIVZCPfAOeM/comments",
+ "countHint":8} },
+ "media$group":{"media$category":[{"$t":"Comedy","label":"Comedy",
+ "scheme":"http://gdata.youtube.com/schemas/2007/categories.cat"}],
+ "media$content":[{"url":"https://www.youtube.com/v/DIVZCPfAOeM?version=3&f=videos&app=youtube_gdata",
+ "type":"application/x-shockwave-flash","medium":"video",
+ "isDefault":"true","expression":"full","duration":354,"yt$format":5},
+ {"url":"rtsp://r1---sn-cg07luel.c.youtube.com/CiILENy73wIaGQnjOcD3CFmFDBMYDSANFEgGUgZ2aWRlb3MM/0/0/0/video.3gp",
+ "type":"video/3gpp","medium":"video","expression":"full","duration":354,
+ "yt$format":1},
+ {"url":"rtsp://r1---sn-cg07luel.c.youtube.com/CiILENy73wIaGQnjOcD3CFmFDBMYESARFEgGUgZ2aWRlb3MM/0/0/0/video.3gp",
+ "type":"video/3gpp","medium":"video","expression":"full","duration":354,"yt$format":6}],
+ "media$description":{"$t":"Desc","type":"plain"},
+ "media$keywords":{},
+ "media$player":[{"url":"https://www.youtube.com/watch?v=DIVZCPfAOeM&feature=youtube_gdata_player"}],
+ "media$thumbnail":[{"url":"https://i.ytimg.com/vi/DIVZCPfAOeM/0.jpg",
+ "height":360,"width":480,"time":"00:02:57"},
+ {"url":"https://i.ytimg.com/vi/DIVZCPfAOeM/1.jpg","height":90,"width":120,"time":"00:01:28.500"},
+ {"url":"https://i.ytimg.com/vi/DIVZCPfAOeM/2.jpg","height":90,"width":120,"time":"00:02:57"},
+ {"url":"https://i.ytimg.com/vi/DIVZCPfAOeM/3.jpg","height":90,"width":120,"time":"00:04:25.500"}],
+ "media$title":{"$t":"Title","type":"plain"},
+ "yt$duration":{"seconds":"354"} },
+ "gd$rating":{"average":4.932159,"max":5,"min":1,"numRaters":1533,
+ "rel":"http://schemas.google.com/g/2005#overall"},
+ "yt$statistics":{"favoriteCount":"0","viewCount":"92464"} }
+ ]
+ }
+ }
+ """
+ response = mock.Mock(text=json)
+ results = youtube.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/0.jpg')
+ self.assertTrue('DIVZCPfAOeM' in results[0]['embedded'])
+
+ json = """
+ {"feed":{"entry":[{
+ "id":{"$t":"http://gdata.youtube.com/feeds/api/videos/DIVZCPfAOeM"},
+ "published":{"$t":"2015-01-23T21:25:00.000Z"},
+ "updated":{"$t":"2015-01-26T14:38:15.000Z"},
+ "title":{"$t":"Title",
+ "type":"text"},"content":{"$t":"Description","type":"text"},
+ "link":[{"rel":"http://gdata.youtube.com/schemas/2007#video.related",
+ "type":"application/atom+xml",
+ "href":"https://gdata.youtube.com/feeds/api/videos/DIVZCPfAOeM/related"},
+ {"rel":"self","type":"application/atom+xml",
+ "href":"https://gdata.youtube.com/feeds/api/videos/DIVZCPfAOeM"}],
+ "author":[{"name":{"$t":"Cauet"},
+ "uri":{"$t":"https://gdata.youtube.com/feeds/api/users/cauetofficiel"} }],
+ "gd$comments":{"gd$feedLink":{"rel":"http://gdata.youtube.com/schemas/2007#comments",
+ "href":"https://gdata.youtube.com/feeds/api/videos/DIVZCPfAOeM/comments",
+ "countHint":8} },
+ "media$group":{"media$category":[{"$t":"Comedy","label":"Comedy",
+ "scheme":"http://gdata.youtube.com/schemas/2007/categories.cat"}],
+ "media$content":[{"url":"https://www.youtube.com/v/DIVZCPfAOeM?version=3&f=videos&app=youtube_gdata",
+ "type":"application/x-shockwave-flash","medium":"video",
+ "isDefault":"true","expression":"full","duration":354,"yt$format":5},
+ {"url":"rtsp://r1---sn-cg07luel.c.youtube.com/CiILENy73wIaGQnjOcD3CFmFDBMYDSANFEgGUgZ2aWRlb3MM/0/0/0/video.3gp",
+ "type":"video/3gpp","medium":"video","expression":"full","duration":354,
+ "yt$format":1},
+ {"url":"rtsp://r1---sn-cg07luel.c.youtube.com/CiILENy73wIaGQnjOcD3CFmFDBMYESARFEgGUgZ2aWRlb3MM/0/0/0/video.3gp",
+ "type":"video/3gpp","medium":"video","expression":"full","duration":354,"yt$format":6}],
+ "media$description":{"$t":"Desc","type":"plain"},
+ "media$keywords":{},
+ "media$player":[{"url":"https://www.youtube.com/watch?v=DIVZCPfAOeM&feature=youtube_gdata_player"}],
+ "media$thumbnail":[{"url":"https://i.ytimg.com/vi/DIVZCPfAOeM/0.jpg",
+ "height":360,"width":480,"time":"00:02:57"},
+ {"url":"https://i.ytimg.com/vi/DIVZCPfAOeM/1.jpg","height":90,"width":120,"time":"00:01:28.500"},
+ {"url":"https://i.ytimg.com/vi/DIVZCPfAOeM/2.jpg","height":90,"width":120,"time":"00:02:57"},
+ {"url":"https://i.ytimg.com/vi/DIVZCPfAOeM/3.jpg","height":90,"width":120,"time":"00:04:25.500"}],
+ "media$title":{"$t":"Title","type":"plain"},
+ "yt$duration":{"seconds":"354"} },
+ "gd$rating":{"average":4.932159,"max":5,"min":1,"numRaters":1533,
+ "rel":"http://schemas.google.com/g/2005#overall"},
+ "yt$statistics":{"favoriteCount":"0","viewCount":"92464"} }
+ ]
+ }
+ }
+ """
+ response = mock.Mock(text=json)
+ results = youtube.response(response)
+ self.assertEqual(type(results), list)
+ self.assertEqual(len(results), 0)
+
+ json = """
+ {"feed":{"entry":[{
+ "id":{"$t":"http://gdata.youtube.com/feeds/api/videos/DIVZCPfAOeM"},
+ "published":{"$t":"2015-01-23T21:25:00.000Z"},
+ "updated":{"$t":"2015-01-26T14:38:15.000Z"},
+ "title":{"$t":"Title",
+ "type":"text"},"content":{"$t":"Description","type":"text"},
+ "link":[{"rel":"alternate","type":"text/html",
+ "href":"https://www.youtube.com/watch?v=DIVZCPfAOeM"},
+ {"rel":"http://gdata.youtube.com/schemas/2007#video.related",
+ "type":"application/atom+xml",
+ "href":"https://gdata.youtube.com/feeds/api/videos/DIVZCPfAOeM/related"},
+ {"rel":"http://gdata.youtube.com/schemas/2007#mobile","type":"text/html",
+ "href":"https://m.youtube.com/details?v=DIVZCPfAOeM"},
+ {"rel":"self","type":"application/atom+xml",
+ "href":"https://gdata.youtube.com/feeds/api/videos/DIVZCPfAOeM"}],
+ "author":[{"name":{"$t":"Cauet"},
+ "uri":{"$t":"https://gdata.youtube.com/feeds/api/users/cauetofficiel"} }],
+ "gd$comments":{"gd$feedLink":{"rel":"http://gdata.youtube.com/schemas/2007#comments",
+ "href":"https://gdata.youtube.com/feeds/api/videos/DIVZCPfAOeM/comments",
+ "countHint":8} },
+ "media$group":{"media$category":[{"$t":"Comedy","label":"Comedy",
+ "scheme":"http://gdata.youtube.com/schemas/2007/categories.cat"}],
+ "media$content":[{"url":"https://www.youtube.com/v/DIVZCPfAOeM?version=3&f=videos&app=youtube_gdata",
+ "type":"application/x-shockwave-flash","medium":"video",
+ "isDefault":"true","expression":"full","duration":354,"yt$format":5},
+ {"url":"rtsp://r1---sn-cg07luel.c.youtube.com/CiILENy73wIaGQnjOcD3CFmFDBMYDSANFEgGUgZ2aWRlb3MM/0/0/0/video.3gp",
+ "type":"video/3gpp","medium":"video","expression":"full","duration":354,
+ "yt$format":1},
+ {"url":"rtsp://r1---sn-cg07luel.c.youtube.com/CiILENy73wIaGQnjOcD3CFmFDBMYESARFEgGUgZ2aWRlb3MM/0/0/0/video.3gp",
+ "type":"video/3gpp","medium":"video","expression":"full","duration":354,"yt$format":6}],
+ "media$description":{"$t":"Desc","type":"plain"},
+ "media$keywords":{},
+ "media$player":[{"url":"https://www.youtube.com/watch?v=DIVZCPfAOeM&feature=youtube_gdata_player"}],
+ "media$title":{"$t":"Title","type":"plain"},
+ "yt$duration":{"seconds":"354"} },
+ "gd$rating":{"average":4.932159,"max":5,"min":1,"numRaters":1533,
+ "rel":"http://schemas.google.com/g/2005#overall"},
+ "yt$statistics":{"favoriteCount":"0","viewCount":"92464"} }
+ ]
+ }
+ }
+ """
+ response = mock.Mock(text=json)
+ results = youtube.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'], '')
+ self.assertTrue('DIVZCPfAOeM' in results[0]['embedded'])
+
+ json = """
+ {"toto":{"entry":[]
+ }
+ }
+ """
+ response = mock.Mock(text=json)
+ results = youtube.response(response)
+ self.assertEqual(type(results), list)
+ self.assertEqual(len(results), 0)
diff --git a/searx/tests/test_engines.py b/searx/tests/test_engines.py
@@ -4,3 +4,4 @@ from searx.tests.engines.test_dummy import * # noqa
from searx.tests.engines.test_flickr import * # noqa
from searx.tests.engines.test_github import * # noqa
from searx.tests.engines.test_mixcloud import * # noqa
+from searx.tests.engines.test_youtube import * # noqa
