Fix error when serializing reverted SSL flows.

This commit is contained in:
Aldo Cortesi 2012-07-05 11:52:56 +12:00
parent 4acc9aca27
commit aa708a2d28
2 changed files with 8 additions and 1 deletions

View File

@ -538,7 +538,7 @@ class Response(HTTPMsg):
self.headers = ODictCaseless._from_state(state["headers"])
self.content = state["content"]
self.timestamp = state["timestamp"]
self.cert = state["cert"]
self.cert = certutils.SSLCert.from_pem(state["cert"]) if state["cert"] else None
def _get_state(self):
return dict(

View File

@ -192,6 +192,13 @@ class TestFlow:
f.revert()
assert f.request.content == "foo"
def test_backup_idempotence(self):
f = tutils.tflow_full()
f.backup()
f.revert()
f.backup()
f.revert()
def test_getset_state(self):
f = tutils.tflow()
f.response = tutils.tresp(f.request)