Mirroring/vcrpy
1
0
Fork 0
mirror of https://github.com/kevin1024/vcrpy.git synced 2025-12-09 17:15:35 +00:00
Code Releases Activity

Fix multiple header behaviour.

Browse Source 
Join multiple header values together, rather than losing duplicates with a
dict.
This commit is contained in:
Matt Thomson
2014-09-06 17:15:15 +01:00
parent 3c410b5f9d
commit f719f90e63
2 changed files with 15 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
vcr/stubs/__init__.py
View File

@@ -57,7 +57,7 @@ def parse_headers(header_list):
def serialize_headers(response):
out = {}
for key, values in compat.get_headers(response):
for key, values in compat.get_headers(response.msg):
out.setdefault(key, [])
out[key].extend(values)
return out
@@ -108,11 +108,15 @@ class VCRHTTPResponse(HTTPResponse):
def getheaders(self):
message = parse_headers(self.recorded_response['headers'])
return compat.get_header_items(message)
return list(compat.get_header_items(message))
def getheader(self, header, default=None):
headers = dict(((k.lower(), v) for k, v in self.getheaders()))
return headers.get(header.lower(), default)
values = [v for (k, v) in self.getheaders() if k.lower() == header.lower()]
if values:
return ', '.join(values)
else:
return default
class VCRConnection: