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

fix serialization problem

This commit is contained in:
Kevin McCarthy
2014-05-10 09:59:48 -10:00
parent b671e7ab99
commit 2c895eb5e9

View File

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