diff options
| author | Philipp Hagemeister <phihag@phihag.de> | 2014-11-23 21:33:31 +0100 | 
|---|---|---|
| committer | Philipp Hagemeister <phihag@phihag.de> | 2014-11-23 21:33:31 +0100 | 
| commit | 784b6d3a9bc79fe55a8b132fd10555c1e9a61c31 (patch) | |
| tree | e9a7996c44fd8d07f0bef5cdebbafdbe9c0b7bb4 /test/test_download.py | |
| parent | f3a34072266c0a7595f73fa032685cf8a50d2ab4 (diff) | |
| parent | 2514d2635e0c8ff0fb72d00a093a849001df2bdd (diff) | |
Merge remote-tracking branch 'jtwaleson/master'
Diffstat (limited to 'test/test_download.py')
| -rw-r--r-- | test/test_download.py | 13 | 
1 files changed, 11 insertions, 2 deletions
| diff --git a/test/test_download.py b/test/test_download.py index 12cfb5cbe..b2615f338 100644 --- a/test/test_download.py +++ b/test/test_download.py @@ -40,18 +40,22 @@ from youtube_dl.extractor import get_info_extractor  RETRIES = 3 +  class YoutubeDL(youtube_dl.YoutubeDL):      def __init__(self, *args, **kwargs):          self.to_stderr = self.to_screen          self.processed_info_dicts = []          super(YoutubeDL, self).__init__(*args, **kwargs) +      def report_warning(self, message):          # Don't accept warnings during tests          raise ExtractorError(message) +      def process_info(self, info_dict):          self.processed_info_dicts.append(info_dict)          return super(YoutubeDL, self).process_info(info_dict) +  def _file_md5(fn):      with open(fn, 'rb') as f:          return hashlib.md5(f.read()).hexdigest() @@ -61,10 +65,13 @@ defs = gettestcases()  class TestDownload(unittest.TestCase):      maxDiff = None +      def setUp(self):          self.defs = defs -### Dynamically generate tests +# Dynamically generate tests + +  def generator(test_case):      def test_template(self): @@ -101,6 +108,7 @@ def generator(test_case):          ydl = YoutubeDL(params, auto_init=False)          ydl.add_default_info_extractors()          finished_hook_called = set() +          def _hook(status):              if status['status'] == 'finished':                  finished_hook_called.add(status['filename']) @@ -111,6 +119,7 @@ def generator(test_case):              return tc.get('file') or ydl.prepare_filename(tc.get('info_dict', {}))          res_dict = None +          def try_rm_tcs_files(tcs=None):              if tcs is None:                  tcs = test_cases @@ -206,7 +215,7 @@ def generator(test_case):      return test_template -### And add them to TestDownload +# And add them to TestDownload  for n, test_case in enumerate(defs):      test_method = generator(test_case)      tname = 'test_' + str(test_case['name']) | 
