[generic] Add a test case for direct links with broken HEAD (#4032)

This commit is contained in:
Philipp Hagemeister 2014-10-26 20:49:51 +01:00
parent 579657ad87
commit 70b7e3fbb6
3 changed files with 28 additions and 1 deletions

View file

@ -171,3 +171,13 @@ def assertGreaterEqual(self, got, expected, msg=None):
if msg is None:
msg = '%r not greater than or equal to %r' % (got, expected)
self.assertTrue(got >= expected, msg)
def expect_warnings(ydl, warnings_re):
real_warning = ydl.report_warning
def _report_warning(w):
if not any(re.search(w_re, w) for w_re in warnings_re):
real_warning(w)
ydl.report_warning = _report_warning