Fix Statigram test

[youtube-dl] / test / test_download.py

diff --git a/test/test_download.py b/test/test_download.py
index 86215203361057ede5b4a3ad5dd5b2a22b004a54..6365b0340ce2228e7248209a7a39987984abe20d 100644 (file)
--- a/test/test_download.py
+++ b/test/test_download.py
@@ -38,8 +38,7 @@ def _try_rm(filename):
         if ose.errno != errno.ENOENT:
             raise
 
-def crc32(value):
-    return '%08x' % (binascii.crc32(value.encode('utf8')) & 0xffffffff)
+md5 = lambda s: hashlib.md5(s.encode('utf-8')).hexdigest()
 
 class FileDownloader(youtube_dl.FileDownloader):
     def __init__(self, *args, **kwargs):
@@ -126,14 +125,17 @@ def generator(test_case):
                     self.assertEqual(md5_for_file, tc['md5'])
                 with io.open(tc['file'] + '.info.json', encoding='utf-8') as infof:
                     info_dict = json.load(infof)
-                for (info_field, value) in tc.get('info_dict', {}).items():
-                    if isinstance(value, compat_str) and value.startswith('crc32:'):
-                        self.assertEqual(value, 'crc32:' + crc32(info_dict.get(info_field)))
+                for (info_field, expected) in tc.get('info_dict', {}).items():
+                    if isinstance(expected, compat_str) and expected.startswith('md5:'):
+                        self.assertEqual(expected, 'md5:' + md5(info_dict.get(info_field)))
                     else:
-                        self.assertEqual(value, info_dict.get(info_field))
+                        got = info_dict.get(info_field)
+                        self.assertEqual(
+                            expected, got,
+                            u'invalid value for field %s, expected %r, got %r' % (info_field, expected, got))
 
                 # If checkable fields are missing from the test case, print the info_dict
-                test_info_dict = dict((key, value if not isinstance(value, compat_str) or len(value) < 250 else 'crc32:' + crc32(value))
+                test_info_dict = dict((key, value if not isinstance(value, compat_str) or len(value) < 250 else 'md5:' + md5(value))
                     for key, value in info_dict.items()
                     if value and key in ('title', 'description', 'uploader', 'upload_date', 'uploader_id', 'location'))
                 if not all(key in tc.get('info_dict', {}).keys() for key in test_info_dict.keys()):