mirror of
				https://github.com/yt-dlp/yt-dlp.git
				synced 2025-10-31 06:35:12 +00:00 
			
		
		
		
	adding download test with md5 check
This commit is contained in:
		
							
								
								
									
										37
									
								
								test/test_download.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										37
									
								
								test/test_download.py
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,37 @@ | ||||
| #!/usr/bin/env python2 | ||||
| import unittest | ||||
| import hashlib | ||||
| import os | ||||
|  | ||||
| from youtube_dl.FileDownloader import FileDownloader | ||||
| from youtube_dl.InfoExtractors  import YoutubeIE | ||||
|  | ||||
| class DownloadTest(unittest.TestCase): | ||||
|     #calculated with the md5sum utility | ||||
|     #md5sum (GNU coreutils) 8.19 | ||||
|     YOUTUBE_MD5 = "ba4092da68c9ded8ef3aaace5ffd1860" | ||||
|     YOUTUBE_URL = "http://www.youtube.com/watch?v=u0VbyYcljx8&feature=related" | ||||
|     YOUTUBE_FILE = "u0VbyYcljx8.flv" | ||||
|  | ||||
|     def test_youtube(self): | ||||
|         #let's download a file from youtube | ||||
|         global YOUTUBE_URL | ||||
|         fd = FileDownloader({}) | ||||
|         fd.add_info_extractor(YoutubeIE()) | ||||
|         fd.download([DownloadTest.YOUTUBE_URL]) | ||||
|         self.assertTrue(os.path.exists(DownloadTest.YOUTUBE_FILE)) | ||||
|         md5_down_file = md5_for_file(DownloadTest.YOUTUBE_FILE) | ||||
|         self.assertEqual(md5_down_file, DownloadTest.YOUTUBE_MD5) | ||||
|  | ||||
|     def cleanUp(self): | ||||
|         if os.path.exists(DownloadTest.YOUTUBE_FILE): | ||||
|             os.remove(DownloadTest.YOUTUBE_FILE) | ||||
|  | ||||
| def md5_for_file(f, block_size=2**20): | ||||
|     md5 = hashlib.md5() | ||||
|     while True: | ||||
|         data = f.read(block_size) | ||||
|         if not data: | ||||
|             break | ||||
|         md5.update(data) | ||||
|         return md5.digest() | ||||
		Reference in New Issue
	
	Block a user
	 gcmalloc
					gcmalloc