proxy_test: Fix pre_response_callback

ArchiveTeam · Jan 27, 2015 · fb8c4fa · fb8c4fa
1 parent 7abea0f
commit fb8c4fa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/wpull/proxy_test.py b/wpull/proxy_test.py
@@ -47,12 +47,12 @@ def request_callback(request):
             print(request)
             cookie_jar_wrapper.add_cookie_header(request)
 
-        def response_callback(request, response):
+        def pre_response_callback(request, response):
             print(response)
             cookie_jar_wrapper.extract_cookies(response, request)
 
         proxy.request_callback = request_callback
-        proxy.response_callback = response_callback
+        proxy.pre_response_callback = pre_response_callback
 
         yield From(trollius.start_server(proxy, sock=proxy_socket))