Browse Source

Fix MD5 calculation

rtmp_test
Philipp Hagemeister 12 years ago
parent
commit
e91d2338d8
3 changed files with 10 additions and 36 deletions
  1. +0
    -13
      test/gentests.py
  2. +5
    -18
      test/test_download.py
  3. +5
    -5
      test/tests.json

+ 0
- 13
test/gentests.py View File

@ -34,19 +34,6 @@ import youtube_dl.InfoExtractors
def _file_md5(fn): def _file_md5(fn):
with open(fn, 'rb') as f: with open(fn, 'rb') as f:
return hashlib.md5(f.read()).hexdigest() return hashlib.md5(f.read()).hexdigest()
def md5_for_file(filename, block_size=2**20):
with open(filename) as f:
md5 = hashlib.md5()
while True:
data = f.read(block_size)
if not data:
break
md5.update(data)
return md5.hexdigest()
_file_md5 = md5_for_file
try: try:
_skip_unless = unittest.skipUnless _skip_unless = unittest.skipUnless
except AttributeError: # Python 2.6 except AttributeError: # Python 2.6


+ 5
- 18
test/test_download.py View File

@ -20,19 +20,6 @@ import youtube_dl.InfoExtractors
def _file_md5(fn): def _file_md5(fn):
with open(fn, 'rb') as f: with open(fn, 'rb') as f:
return hashlib.md5(f.read()).hexdigest() return hashlib.md5(f.read()).hexdigest()
def md5_for_file(filename, block_size=2**20):
with open(filename) as f:
md5 = hashlib.md5()
while True:
data = f.read(block_size)
if not data:
break
md5.update(data)
return md5.hexdigest()
_file_md5 = md5_for_file
try: try:
_skip_unless = unittest.skipUnless _skip_unless = unittest.skipUnless
except AttributeError: # Python 2.6 except AttributeError: # Python 2.6
@ -96,7 +83,7 @@ class DownloadTest(unittest.TestCase):
fd.download(['http://blip.tv/cbr/cbr-exclusive-gotham-city-imposters-bats-vs-jokerz-short-3-5796352']) fd.download(['http://blip.tv/cbr/cbr-exclusive-gotham-city-imposters-bats-vs-jokerz-short-3-5796352'])
self.assertTrue(os.path.exists(filename)) self.assertTrue(os.path.exists(filename))
md5_for_file = _file_md5(filename) md5_for_file = _file_md5(filename)
self.assertEqual(md5_for_file, '93c24d2f4e0782af13b8a7606ea97ba7')
self.assertEqual(md5_for_file, '4962f94441605832eb1008eb820ef47a')
@_skip_unless(youtube_dl.InfoExtractors.XVideosIE._WORKING, "IE marked as not _WORKING") @_skip_unless(youtube_dl.InfoExtractors.XVideosIE._WORKING, "IE marked as not _WORKING")
def test_XVideos(self): def test_XVideos(self):
@ -106,7 +93,7 @@ class DownloadTest(unittest.TestCase):
fd.download(['http://www.xvideos.com/video939581/funny_porns_by_s_-1']) fd.download(['http://www.xvideos.com/video939581/funny_porns_by_s_-1'])
self.assertTrue(os.path.exists(filename)) self.assertTrue(os.path.exists(filename))
md5_for_file = _file_md5(filename) md5_for_file = _file_md5(filename)
self.assertEqual(md5_for_file, '1ab4dedc01f771cb2a65e91caa801aaf')
self.assertEqual(md5_for_file, 'aecab2ea59b7996110a7e409f0c55da3')
@_skip_unless(youtube_dl.InfoExtractors.VimeoIE._WORKING, "IE marked as not _WORKING") @_skip_unless(youtube_dl.InfoExtractors.VimeoIE._WORKING, "IE marked as not _WORKING")
@_skip("No output file specified") @_skip("No output file specified")
@ -127,7 +114,7 @@ class DownloadTest(unittest.TestCase):
fd.download(['http://soundcloud.com/ethmusic/lostin-powers-she-so-heavy']) fd.download(['http://soundcloud.com/ethmusic/lostin-powers-she-so-heavy'])
self.assertTrue(os.path.exists(filename)) self.assertTrue(os.path.exists(filename))
md5_for_file = _file_md5(filename) md5_for_file = _file_md5(filename)
self.assertEqual(md5_for_file, 'ce3775768ebb6432fa8495d446a078ed')
self.assertEqual(md5_for_file, 'c1b9b9ea8bfd620b96b2628664576e1c')
@_skip_unless(youtube_dl.InfoExtractors.StanfordOpenClassroomIE._WORKING, "IE marked as not _WORKING") @_skip_unless(youtube_dl.InfoExtractors.StanfordOpenClassroomIE._WORKING, "IE marked as not _WORKING")
def test_StanfordOpenClassroom(self): def test_StanfordOpenClassroom(self):
@ -137,7 +124,7 @@ class DownloadTest(unittest.TestCase):
fd.download(['http://openclassroom.stanford.edu/MainFolder/VideoPage.php?course=PracticalUnix&video=intro-environment&speed=100']) fd.download(['http://openclassroom.stanford.edu/MainFolder/VideoPage.php?course=PracticalUnix&video=intro-environment&speed=100'])
self.assertTrue(os.path.exists(filename)) self.assertTrue(os.path.exists(filename))
md5_for_file = _file_md5(filename) md5_for_file = _file_md5(filename)
self.assertEqual(md5_for_file, '22c8206291368c4e2c9c1a307f0ea0f4')
self.assertEqual(md5_for_file, '8aac7873a07dcfaed66b1559ab128514')
@_skip_unless(youtube_dl.InfoExtractors.CollegeHumorIE._WORKING, "IE marked as not _WORKING") @_skip_unless(youtube_dl.InfoExtractors.CollegeHumorIE._WORKING, "IE marked as not _WORKING")
@_skip("No output file specified") @_skip("No output file specified")
@ -158,7 +145,7 @@ class DownloadTest(unittest.TestCase):
fd.download(['http://video.xnxx.com/video1135332/lida_naked_funny_actress_5_']) fd.download(['http://video.xnxx.com/video1135332/lida_naked_funny_actress_5_'])
self.assertTrue(os.path.exists(filename)) self.assertTrue(os.path.exists(filename))
md5_for_file = _file_md5(filename) md5_for_file = _file_md5(filename)
self.assertEqual(md5_for_file, '5f0469c8d1dfd1bc38c8e6deb5e0a21d')
self.assertEqual(md5_for_file, 'c5c67df477eb0d9b058200351448ba4c')
def tearDown(self): def tearDown(self):


+ 5
- 5
test/tests.json View File

@ -20,13 +20,13 @@
}, },
{ {
"name": "BlipTV", "name": "BlipTV",
"md5": "93c24d2f4e0782af13b8a7606ea97ba7",
"md5": "4962f94441605832eb1008eb820ef47a",
"url": "http://blip.tv/cbr/cbr-exclusive-gotham-city-imposters-bats-vs-jokerz-short-3-5796352", "url": "http://blip.tv/cbr/cbr-exclusive-gotham-city-imposters-bats-vs-jokerz-short-3-5796352",
"file": "5779306.m4v" "file": "5779306.m4v"
}, },
{ {
"name": "XVideos", "name": "XVideos",
"md5": "1ab4dedc01f771cb2a65e91caa801aaf",
"md5": "aecab2ea59b7996110a7e409f0c55da3",
"url": "http://www.xvideos.com/video939581/funny_porns_by_s_-1", "url": "http://www.xvideos.com/video939581/funny_porns_by_s_-1",
"file": "939581.flv" "file": "939581.flv"
}, },
@ -38,13 +38,13 @@
}, },
{ {
"name": "Soundcloud", "name": "Soundcloud",
"md5": "ce3775768ebb6432fa8495d446a078ed",
"md5": "c1b9b9ea8bfd620b96b2628664576e1c",
"url": "http://soundcloud.com/ethmusic/lostin-powers-she-so-heavy", "url": "http://soundcloud.com/ethmusic/lostin-powers-she-so-heavy",
"file": "n6FLbx6ZzMiu.mp3" "file": "n6FLbx6ZzMiu.mp3"
}, },
{ {
"name": "StanfordOpenClassroom", "name": "StanfordOpenClassroom",
"md5": "22c8206291368c4e2c9c1a307f0ea0f4",
"md5": "8aac7873a07dcfaed66b1559ab128514",
"url": "http://openclassroom.stanford.edu/MainFolder/VideoPage.php?course=PracticalUnix&video=intro-environment&speed=100", "url": "http://openclassroom.stanford.edu/MainFolder/VideoPage.php?course=PracticalUnix&video=intro-environment&speed=100",
"file": "PracticalUnix_intro-environment.mp4" "file": "PracticalUnix_intro-environment.mp4"
}, },
@ -56,7 +56,7 @@
}, },
{ {
"name": "XNXX", "name": "XNXX",
"md5": "5f0469c8d1dfd1bc38c8e6deb5e0a21d",
"md5": "c5c67df477eb0d9b058200351448ba4c",
"url": "http://video.xnxx.com/video1135332/lida_naked_funny_actress_5_", "url": "http://video.xnxx.com/video1135332/lida_naked_funny_actress_5_",
"file": "1135332.flv" "file": "1135332.flv"
} }

Loading…
Cancel
Save