[test] Support additional args and kwargs in report_warning() mocks

1 files changed, 4 insertions, 4 deletions

diff --git a/test/helper.py b/test/helper.py
index 6f2129eff..fac069d25 100644
--- a/test/helper.py
+++ b/test/helper.py
@@ -85,10 +85,10 @@ class FakeYDL(YoutubeDL):
         # Silence an expected warning matching a regex
         old_report_warning = self.report_warning
 
-        def report_warning(self, message):
+        def report_warning(self, message, *args, **kwargs):
             if re.match(regex, message):
                 return
-            old_report_warning(message)
+            old_report_warning(message, *args, **kwargs)
         self.report_warning = types.MethodType(report_warning, self)
 
 
@@ -265,11 +265,11 @@ def assertRegexpMatches(self, text, regexp, msg=None):
 def expect_warnings(ydl, warnings_re):
     real_warning = ydl.report_warning
 
-    def _report_warning(w):
+    def _report_warning(self, w, *args, **kwargs):
         if not any(re.search(w_re, w) for w_re in warnings_re):
             real_warning(w)
 
-    ydl.report_warning = _report_warning
+    ydl.report_warning = types.MethodType(_report_warning, ydl)
 
 
 def http_server_port(httpd):