Mirroring/vcrpy
1
0
Fork 0
mirror of https://github.com/kevin1024/vcrpy.git synced 2025-12-08 16:53:23 +00:00
Code Releases Activity

fix proxy for Python 2

Browse Source
This commit is contained in:
Samuel Fekete
2017-11-03 11:39:13 +00:00
committed by Samuel Fekete
parent 4b6b5effc7
commit ff7dd06f47
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/integration/test_proxy.py
View File

@@ -23,8 +23,15 @@ class Proxy(SimpleHTTPServer.SimpleHTTPRequestHandler):
'''
def do_GET(self):
upstream_response = urlopen(self.path)
self.send_response(upstream_response.status, upstream_response.msg)
for header in upstream_response.headers.items():
try:
status = upstream_response.status
headers = upstream_response.headers.items()
except AttributeError:
# In Python 2 the response is an addinfourl instance.
status = upstream_response.code
headers = upstream_response.info().items()
self.send_response(status, upstream_response.msg)
for header in headers:
self.send_header(*header)
self.end_headers()
self.copyfile(upstream_response, self.wfile)