diff options
author | Sergey M․ <dstftw@gmail.com> | 2018-02-03 23:24:27 +0700 |
---|---|---|
committer | Sergey M․ <dstftw@gmail.com> | 2018-02-03 23:24:27 +0700 |
commit | 6fcc053947cbb7bc603b0a24ae27e4a6982fe12d (patch) | |
tree | 2aa3e26c9c62b8008b72c77f4ef38fdcae1db8ff /test | |
parent | a3e8146ea8b06459ab5b5acb09ab5169faf198d8 (diff) |
[test_downloader_http] Use try_rm
Diffstat (limited to 'test')
-rw-r--r-- | test/test_downloader_http.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/test/test_downloader_http.py b/test/test_downloader_http.py index 95d152072..5cf2bf1a5 100644 --- a/test/test_downloader_http.py +++ b/test/test_downloader_http.py @@ -9,6 +9,7 @@ import sys import unittest sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) +from test.helper import try_rm from youtube_dl import YoutubeDL from youtube_dl.compat import compat_http_server from youtube_dl.downloader.http import HttpFD @@ -100,11 +101,12 @@ class TestHttpFD(unittest.TestCase): ydl = YoutubeDL(params) downloader = HttpFD(ydl, params) filename = 'testfile.mp4' + try_rm(encodeFilename(filename)) self.assertTrue(downloader.real_download(filename, { 'url': 'http://127.0.0.1:%d/%s' % (self.port, ep), })) self.assertEqual(os.path.getsize(encodeFilename(filename)), TEST_SIZE) - os.remove(encodeFilename(filename)) + try_rm(encodeFilename(filename)) def download_all(self, params): for ep in ('regular', 'no-content-length', 'no-range', 'no-range-no-content-length'): |