tests: fix mock YoutubeDLWrapper after refactor, #141

Author: ikreymer

pywb/rewrite/rewrite_live.py

@@ -284,7 +284,6 @@ def __init__(self):
         self.ydl.add_default_info_extractors()
 
     def extract_info(self, url):
-        print('YDL', self.ydl)
         if not self.ydl:
             return None
 

tests/test_live_proxy.py

@@ -10,7 +10,7 @@
 import webtest
 import shutil
 
-import pywb.webapp.live_rewrite_handler
+import pywb.rewrite.rewrite_live
 
 
 #=================================================================
@@ -53,7 +53,7 @@ def extract_info(self, url):
         return {'mock': 'youtube_dl_data'}
 
 
-pywb.webapp.live_rewrite_handler.YoutubeDLWrapper = MockYTDWrapper
+pywb.rewrite.rewrite_live.youtubedl = MockYTDWrapper()
 
 
 #=================================================================

tests/test_live_rewriter.py

@@ -2,7 +2,7 @@
 from pywb.apps.cli import LiveCli
 from pywb.framework.wsgi_wrappers import init_app
 import webtest
-import pywb.webapp.live_rewrite_handler
+import pywb.rewrite.rewrite_live
 
 
 #=================================================================
@@ -11,7 +11,7 @@ def extract_info(self, url):
         return {'mock': 'youtube_dl_data'}
 
 
-pywb.webapp.live_rewrite_handler.YoutubeDLWrapper = MockYTDWrapper
+pywb.rewrite.rewrite_live.youtubedl = MockYTDWrapper()
 
 
 def setup_module():