diff options
author | Filippo Valsorda <filippo.valsorda@gmail.com> | 2013-03-09 10:05:43 +0100 |
---|---|---|
committer | Filippo Valsorda <filippo.valsorda@gmail.com> | 2013-03-09 10:05:43 +0100 |
commit | 8cc83b8dbea6e4f34f483c4a209158307df566f0 (patch) | |
tree | 276629be87d66640dc3bc3caaa472fb7b9c1da2b /test | |
parent | 3d3423574d35a0fe71062f21dd57ada02a5225b4 (diff) | |
download | youtube-dl-8cc83b8dbea6e4f34f483c4a209158307df566f0.zip youtube-dl-8cc83b8dbea6e4f34f483c4a209158307df566f0.tar.gz youtube-dl-8cc83b8dbea6e4f34f483c4a209158307df566f0.tar.bz2 |
Bubble up all the stack of exceptions and retry download tests on timeout errors
Diffstat (limited to 'test')
-rw-r--r-- | test/test_download.py | 16 |
1 files changed, 15 insertions, 1 deletions
diff --git a/test/test_download.py b/test/test_download.py index f1bccf5..a8de1d0 100644 --- a/test/test_download.py +++ b/test/test_download.py @@ -20,6 +20,8 @@ from youtube_dl.utils import * DEF_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'tests.json') PARAMETERS_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "parameters.json") +RETRIES = 3 + # General configuration (from __init__, not very elegant...) jar = compat_cookiejar.CookieJar() cookie_processor = compat_urllib_request.HTTPCookieProcessor(jar) @@ -94,7 +96,19 @@ def generator(test_case): _try_rm(tc['file'] + '.part') _try_rm(tc['file'] + '.info.json') try: - fd.download([test_case['url']]) + for retry in range(1, RETRIES + 1): + try: + fd.download([test_case['url']]) + except (DownloadError, ExtractorError) as err: + if retry == RETRIES: raise + + # Check if the exception is not a network related one + if not err.exc_info[0] in (ZeroDivisionError, compat_urllib_error.URLError, socket.timeout): + raise + + print('Retrying: {0} failed tries\n\n##########\n\n'.format(retry)) + else: + break for tc in test_cases: if not test_case.get('params', {}).get('skip_download', False): |