]>
jfr.im git - yt-dlp.git/blob - test/test_download.py
b90ce63233142de1b2a0b8b5f9a12c7064891913
6 from youtube_dl
.FileDownloader
import FileDownloader
7 from youtube_dl
.InfoExtractors
import YoutubeIE
, DailymotionIE
9 class DownloadTest(unittest
.TestCase
):
10 #calculated with md5sum:
11 # md5sum (GNU coreutils) 8.19
12 YOUTUBE_MD5
= "8547978241cb87dd6782b10b8e90acc3"
13 YOUTUBE_URL
= "http://www.youtube.com/watch?v=BaW_jenozKc"
14 YOUTUBE_FILE
= "BaW_jenozKc.flv"
17 DAILYMOTION_URL
= "http://www.dailymotion.com/video/x33vw9_tutoriel-de-youtubeur-dl-des-video_tech"
20 def test_youtube(self
):
21 #let's download a file from youtube
22 fd
= FileDownloader({})
23 fd
.add_info_extractor(YoutubeIE())
24 fd
.download([DownloadTest
.YOUTUBE_URL
])
25 self
.assertTrue(os
.path
.exists(DownloadTest
.YOUTUBE_FILE
))
26 md5_down_file
= md5_for_file(DownloadTest
.YOUTUBE_FILE
)
27 self
.assertEqual(md5_down_file
, DownloadTest
.YOUTUBE_MD5
)
29 def test_dailymotion(self
):
30 fd
= FileDownloader({})
31 fd
.add_info_extractor(DailymotionIE())
32 fd
.download([DownloadTest
.DAILYMOTION_URL
])
33 self
.assertTrue(os
.path
.exists(DownloadTest
.DAILYMOTION_FILE
))
34 md5_down_file
= md5_for_file(DownloadTest
.DAILYMOTION_FILE
)
35 self
.assertEqual(md5_down_file
, DownloadTest
.DAILYMOTION_MD5
)
39 if os
.path
.exists(DownloadTest
.YOUTUBE_FILE
):
40 os
.remove(DownloadTest
.YOUTUBE_FILE
)
41 if os
.path
.exists(DownloadTest
.DAILYMOTION_FILE
):
42 os
.remove(DownloadTest
.DAILYMOTION_FILE
)
44 def md5_for_file(f
, block_size
=2**20):
47 data
= f
.read(block_size
)