Move testcase generator to helper

This commit is contained in:
Philipp Hagemeister 2013-06-27 21:15:16 +02:00
parent 20db33e299
commit fc2c063e1e
3 changed files with 14 additions and 14 deletions

View file

@ -2,6 +2,7 @@ import io
import json
import os.path
import youtube_dl.extractor
from youtube_dl import YoutubeDL, YoutubeDLHandler
from youtube_dl.utils import (
compat_cookiejar,
@ -30,4 +31,14 @@ class FakeYDL(YoutubeDL):
def trouble(self, s, tb=None):
raise Exception(s)
def download(self, x):
self.result.append(x)
self.result.append(x)
def get_testcases():
for ie in youtube_dl.extractor.gen_extractors():
t = getattr(ie, '_TEST', None)
if t:
t['name'] = type(ie).__name__[:-len('IE')]
yield t
for t in getattr(ie, '_TESTS', []):
t['name'] = type(ie).__name__[:-len('IE')]
yield t