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

[youtube-dl.git] / test / test_download.py

diff --git a/test/test_download.py b/test/test_download.py
index 8178015eacb5600f99c983a77bfa43abd9d37b6a..608e4528ed55b1bbdc60f922e3d6d50af877ad98 100644 (file)
--- a/test/test_download.py
+++ b/test/test_download.py
@@ -8,6 +8,7 @@ sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
 from test.helper import (
     assertGreaterEqual,
+    expect_warnings,
     get_params,
     gettestcases,
     expect_info_dict,
@@ -100,6 +101,7 @@ def generator(test_case):
             if status['status'] == 'finished':
                 finished_hook_called.add(status['filename'])
         ydl.add_progress_hook(_hook)
+        expect_warnings(ydl, test_case.get('expected_warnings', []))
 
         def get_tc_filename(tc):
             return tc.get('file') or ydl.prepare_filename(tc.get('info_dict', {}))