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

fix serialization problem

Browse Source
This commit is contained in:
Kevin McCarthy
2014-05-10 09:59:48 -10:00
parent b671e7ab99
commit 2c895eb5e9
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
vcr/migration.py
View File

@@ -20,6 +20,7 @@ import tempfile
import yaml import yaml
from .serializers import compat, yamlserializer from .serializers import compat, yamlserializer
from .serialize import serialize
from . import request from . import request
from .stubs.compat import get_httpmessage from .stubs.compat import get_httpmessage
@@ -95,8 +96,9 @@ def _old_deserialize(cassette_string):
def migrate_yml(in_fp, out_fp): def migrate_yml(in_fp, out_fp):
(requests, responses) = _old_deserialize(in_fp.read()) (requests, responses) = _old_deserialize(in_fp.read())
interactions = [] cassette = {'requests':[], 'responses':[]}
for req, res in zip(requests, responses): for req, res in zip(requests, responses):
import pdb;pdb.set_trace()
if not isinstance(req, request.Request): if not isinstance(req, request.Request):
raise Exception("already migrated") raise Exception("already migrated")
else: else:
@@ -117,12 +119,11 @@ def migrate_yml(in_fp, out_fp):
response_headers.setdefault(k, []) response_headers.setdefault(k, [])
response_headers[k].append(v) response_headers[k].append(v)
res['headers'] = response_headers res['headers'] = response_headers
interactions.append({'request': req._to_dict(), 'response': res}) cassette['requests'].append(req)
cassette['responses'].append(res)
data = yamlserializer.serialize({
"interactions": interactions, data = serialize(cassette, yamlserializer)
"version": 1
})
out_fp.write(data) out_fp.write(data)