print WARNINGs during test + minor fix to NBAIE

1 files changed, 9 insertions, 0 deletions

diff --git a/test/test_download.py b/test/test_download.py
index 3eca333f2..3e6bdd44e 100644
--- a/test/test_download.py
+++ b/test/test_download.py
@@ -40,9 +40,18 @@ def _try_rm(filename):
 
 class FileDownloader(youtube_dl.FileDownloader):
     def __init__(self, *args, **kwargs):
+        self._to_stderr = self.to_stderr
         self.to_stderr = self.to_screen
         self.processed_info_dicts = []
         return youtube_dl.FileDownloader.__init__(self, *args, **kwargs)
+    def report_warning(self, message):
+        # let warnings pass to output
+        if sys.stderr.isatty() and os.name != 'nt':
+            _msg_header=u'\033[0;33mWARNING:\033[0m'
+        else:
+            _msg_header=u'WARNING:'
+        warning_message=u'%s %s' % (_msg_header,message)
+        self._to_stderr(warning_message)
     def process_info(self, info_dict):
         self.processed_info_dicts.append(info_dict)
         return youtube_dl.FileDownloader.process_info(self, info_dict)