Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge remote-tracking branch 'origin/master' into fork_master
New tests - merged with md5 correction
- Loading branch information
Showing
10 changed files
with
387 additions
and
208 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file was deleted.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,128 @@ | ||
#!/usr/bin/env python3 | ||
|
||
import io # for python 2 | ||
import json | ||
import os | ||
import sys | ||
import unittest | ||
|
||
# Allow direct execution | ||
import os | ||
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) | ||
|
||
import youtube_dl.InfoExtractors | ||
|
||
HEADER = u'''#!/usr/bin/env python | ||
# DO NOT EDIT THIS FILE BY HAND! | ||
# It is auto-generated from tests.json and gentests.py. | ||
import hashlib | ||
import io | ||
import os | ||
import json | ||
import unittest | ||
import sys | ||
# Allow direct execution | ||
import os | ||
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) | ||
from youtube_dl.FileDownloader import FileDownloader | ||
import youtube_dl.InfoExtractors | ||
def _file_md5(fn): | ||
with open(fn, 'rb') as f: | ||
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: | ||
_skip_unless = unittest.skipUnless | ||
except AttributeError: # Python 2.6 | ||
def _skip_unless(cond, reason='No reason given'): | ||
def resfunc(f): | ||
def wfunc(*args, **kwargs): | ||
if cond: | ||
return f(*args, **kwargs) | ||
else: | ||
print('Skipped test') | ||
return | ||
return wfunc | ||
return resfunc | ||
_skip = lambda *args, **kwargs: _skip_unless(False, *args, **kwargs) | ||
class DownloadTest(unittest.TestCase): | ||
PARAMETERS_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "parameters.json") | ||
def setUp(self): | ||
# Clear old files | ||
self.tearDown() | ||
with io.open(self.PARAMETERS_FILE, encoding='utf-8') as pf: | ||
self.parameters = json.load(pf) | ||
''' | ||
|
||
FOOTER = u''' | ||
if __name__ == '__main__': | ||
unittest.main() | ||
''' | ||
|
||
DEF_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'tests.json') | ||
TEST_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'test_download.py') | ||
|
||
def gentests(): | ||
with io.open(DEF_FILE, encoding='utf-8') as deff: | ||
defs = json.load(deff) | ||
with io.open(TEST_FILE, 'w', encoding='utf-8') as testf: | ||
testf.write(HEADER) | ||
spaces = ' ' * 4 | ||
write = lambda l: testf.write(spaces + l + '\n') | ||
|
||
for d in defs: | ||
name = d['name'] | ||
ie = getattr(youtube_dl.InfoExtractors, name + 'IE') | ||
testf.write('\n') | ||
write('@_skip_unless(youtube_dl.InfoExtractors.' + name + 'IE._WORKING, "IE marked as not _WORKING")') | ||
if not d['file']: | ||
write('@_skip("No output file specified")') | ||
elif 'skip' in d: | ||
write('@_skip(' + repr(d['skip']) + ')') | ||
write('def test_' + name + '(self):') | ||
write(' filename = ' + repr(d['file'])) | ||
write(' fd = FileDownloader(self.parameters)') | ||
write(' fd.add_info_extractor(youtube_dl.InfoExtractors.' + name + 'IE())') | ||
for ien in d.get('addIEs', []): | ||
write(' fd.add_info_extractor(youtube_dl.InfoExtractors.' + ien + 'IE())') | ||
write(' fd.download([' + repr(d['url']) + '])') | ||
write(' self.assertTrue(os.path.exists(filename))') | ||
if 'size' in d: | ||
write(' self.assertEqual(os.path.getsize(filename), ' + repr(d['size']) + ')') | ||
if 'md5' in d: | ||
write(' md5_for_file = _file_md5(filename)') | ||
write(' self.assertEqual(md5_for_file, ' + repr(d['md5']) + ')') | ||
|
||
testf.write('\n\n') | ||
write('def tearDown(self):') | ||
for d in defs: | ||
if d['file']: | ||
write(' if os.path.exists(' + repr(d['file']) + '):') | ||
write(' os.remove(' + repr(d['file']) + ')') | ||
else: | ||
write(' # No file specified for ' + d['name']) | ||
testf.write('\n') | ||
testf.write(FOOTER) | ||
|
||
if __name__ == '__main__': | ||
gentests() |
Oops, something went wrong.