diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_all_urls.py | 14 | ||||
| -rw-r--r-- | test/test_youtube_lists.py | 58 | ||||
| -rw-r--r-- | test/tests.json | 30 | 
3 files changed, 62 insertions, 40 deletions
diff --git a/test/test_all_urls.py b/test/test_all_urls.py index 06de8e7b8..69717b3fc 100644 --- a/test/test_all_urls.py +++ b/test/test_all_urls.py @@ -11,12 +11,18 @@ from youtube_dl.InfoExtractors import YoutubeIE, YoutubePlaylistIE  class TestAllURLsMatching(unittest.TestCase):      def test_youtube_playlist_matching(self): -        self.assertTrue(YoutubePlaylistIE().suitable(u'ECUl4u3cNGP61MdtwGTqZA0MreSaDybji8')) -        self.assertTrue(YoutubePlaylistIE().suitable(u'PL63F0C78739B09958')) -        self.assertFalse(YoutubePlaylistIE().suitable(u'PLtS2H6bU1M')) +        self.assertTrue(YoutubePlaylistIE.suitable(u'ECUl4u3cNGP61MdtwGTqZA0MreSaDybji8')) +        self.assertTrue(YoutubePlaylistIE.suitable(u'UUBABnxM4Ar9ten8Mdjj1j0Q')) #585 +        self.assertTrue(YoutubePlaylistIE.suitable(u'PL63F0C78739B09958')) +        self.assertTrue(YoutubePlaylistIE.suitable(u'https://www.youtube.com/playlist?list=UUBABnxM4Ar9ten8Mdjj1j0Q')) +        self.assertTrue(YoutubePlaylistIE.suitable(u'https://www.youtube.com/course?list=ECUl4u3cNGP61MdtwGTqZA0MreSaDybji8')) +        self.assertTrue(YoutubePlaylistIE.suitable(u'https://www.youtube.com/playlist?list=PLwP_SiAcdui0KVebT0mU9Apz359a4ubsC')) +        self.assertTrue(YoutubePlaylistIE.suitable(u'https://www.youtube.com/watch?v=AV6J6_AeFEQ&playnext=1&list=PL4023E734DA416012')) #668 +        self.assertFalse(YoutubePlaylistIE.suitable(u'PLtS2H6bU1M'))      def test_youtube_matching(self): -        self.assertTrue(YoutubeIE().suitable(u'PLtS2H6bU1M')) +        self.assertTrue(YoutubeIE.suitable(u'PLtS2H6bU1M')) +        self.assertFalse(YoutubeIE.suitable(u'https://www.youtube.com/watch?v=AV6J6_AeFEQ&playnext=1&list=PL4023E734DA416012')) #668      def test_youtube_extract(self):          self.assertEqual(YoutubeIE()._extract_id('http://www.youtube.com/watch?&v=BaW_jenozKc'), 'BaW_jenozKc') diff --git a/test/test_youtube_lists.py b/test/test_youtube_lists.py index 3044e0852..f4705bc5b 100644 --- a/test/test_youtube_lists.py +++ b/test/test_youtube_lists.py @@ -8,7 +8,7 @@ import json  import os  sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) -from youtube_dl.InfoExtractors import YoutubeUserIE,YoutubePlaylistIE +from youtube_dl.InfoExtractors import YoutubeUserIE, YoutubePlaylistIE, YoutubeIE  from youtube_dl.utils import *  PARAMETERS_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "parameters.json") @@ -35,39 +35,51 @@ class FakeDownloader(object):  class TestYoutubeLists(unittest.TestCase):      def test_youtube_playlist(self): -        DL = FakeDownloader() -        IE = YoutubePlaylistIE(DL) -        IE.extract('https://www.youtube.com/playlist?list=PLwiyx1dc3P2JR9N8gQaQN_BCvlSlap7re') -        self.assertEqual(DL.result, [ -            ['http://www.youtube.com/watch?v=bV9L5Ht9LgY'], -            ['http://www.youtube.com/watch?v=FXxLjLQi3Fg'], -            ['http://www.youtube.com/watch?v=tU3Bgo5qJZE'] -        ]) +        dl = FakeDownloader() +        ie = YoutubePlaylistIE(dl) +        ie.extract('https://www.youtube.com/playlist?list=PLwiyx1dc3P2JR9N8gQaQN_BCvlSlap7re') +        ytie_results = [YoutubeIE()._extract_id(r[0]) for r in dl.result] +        self.assertEqual(ytie_results, [ 'bV9L5Ht9LgY', 'FXxLjLQi3Fg', 'tU3Bgo5qJZE']) + +    def test_issue_673(self): +        dl = FakeDownloader() +        ie = YoutubePlaylistIE(dl) +        ie.extract('PLBB231211A4F62143') +        self.assertTrue(len(dl.result) > 40)      def test_youtube_playlist_long(self): -        DL = FakeDownloader() -        IE = YoutubePlaylistIE(DL) -        IE.extract('https://www.youtube.com/playlist?list=UUBABnxM4Ar9ten8Mdjj1j0Q') -        self.assertTrue(len(DL.result) >= 799) +        dl = FakeDownloader() +        ie = YoutubePlaylistIE(dl) +        ie.extract('https://www.youtube.com/playlist?list=UUBABnxM4Ar9ten8Mdjj1j0Q') +        self.assertTrue(len(dl.result) >= 799) + +    def test_youtube_playlist_with_deleted(self): +        #651 +        dl = FakeDownloader() +        ie = YoutubePlaylistIE(dl) +        ie.extract('https://www.youtube.com/playlist?list=PLwP_SiAcdui0KVebT0mU9Apz359a4ubsC') +        ytie_results = [YoutubeIE()._extract_id(r[0]) for r in dl.result] +        self.assertFalse('pElCt5oNDuI' in ytie_results) +        self.assertFalse('KdPEApIVdWM' in ytie_results)      def test_youtube_course(self): -        DL = FakeDownloader() -        IE = YoutubePlaylistIE(DL) +        dl = FakeDownloader() +        ie = YoutubePlaylistIE(dl)          # TODO find a > 100 (paginating?) videos course -        IE.extract('https://www.youtube.com/course?list=ECUl4u3cNGP61MdtwGTqZA0MreSaDybji8') -        self.assertEqual(DL.result[0], ['http://www.youtube.com/watch?v=j9WZyLZCBzs']) -        self.assertEqual(len(DL.result), 25) -        self.assertEqual(DL.result[-1], ['http://www.youtube.com/watch?v=rYefUsYuEp0']) +        ie.extract('https://www.youtube.com/course?list=ECUl4u3cNGP61MdtwGTqZA0MreSaDybji8') +        self.assertEqual(YoutubeIE()._extract_id(dl.result[0][0]), 'j9WZyLZCBzs') +        self.assertEqual(len(dl.result), 25) +        self.assertEqual(YoutubeIE()._extract_id(dl.result[-1][0]), 'rYefUsYuEp0')      def test_youtube_channel(self):          # I give up, please find a channel that does paginate and test this like test_youtube_playlist_long          pass # TODO      def test_youtube_user(self): -        DL = FakeDownloader() -        IE = YoutubeUserIE(DL) -        IE.extract('https://www.youtube.com/user/TheLinuxFoundation') -        self.assertTrue(len(DL.result) >= 320) +        dl = FakeDownloader() +        ie = YoutubeUserIE(dl) +        ie.extract('https://www.youtube.com/user/TheLinuxFoundation') +        self.assertTrue(len(dl.result) >= 320)  if __name__ == '__main__':      unittest.main() diff --git a/test/tests.json b/test/tests.json index a3c31ae51..fd9d33332 100644 --- a/test/tests.json +++ b/test/tests.json @@ -129,18 +129,6 @@      "md5": "f647e9e90064b53b6e046e75d0241fbd"    },    { -    "name": "TweetReel", -    "url": "http://tweetreel.com/?77smq", -    "file": "77smq.mov", -    "md5": "56b4d9ca9de467920f3f99a6d91255d6", -    "info_dict": { -        "uploader": "itszero", -        "uploader_id": "itszero", -        "upload_date": "20091225", -        "description": "Installing Gentoo Linux on Powerbook G4, it turns out the sleep indicator becomes HDD activity indicator :D" -    } -  }, -  {      "name": "Steam",      "url": "http://store.steampowered.com/video/105600/",      "playlist": [ @@ -293,7 +281,8 @@      "file": "102.mp4",      "md5": "7bc087e71d16f18f9b8ab9fa62a8a031",      "info_dict": { -        "title": "Dan Dennett: The illusion of consciousness" +        "title": "Dan Dennett: The illusion of consciousness", +        "thumbnail": "http://images.ted.com/images/ted/488_389x292.jpg"      }    },    { @@ -304,5 +293,20 @@      "info_dict": {          "title": "Absolute Mehrheit vom 17.02.2013 - Die Highlights, Teil 2"      } +  }, +  { +    "name": "Generic", +    "url": "http://www.hodiho.fr/2013/02/regis-plante-sa-jeep.html", +    "file": "13601338388002.mp4", +    "md5": "85b90ccc9d73b4acd9138d3af4c27f89" +  }, +  { +    "name": "Spiegel", +    "url": "http://www.spiegel.de/video/vulkan-tungurahua-in-ecuador-ist-wieder-aktiv-video-1259285.html", +    "file": "1259285.mp4", +    "md5": "2c2754212136f35fb4b19767d242f66e", +    "info_dict": { +        "title": "Vulkanausbruch in Ecuador: Der \"Feuerschlund\" ist wieder aktiv" +    }    }  ]  | 
