mirror of
https://github.com/kevin1024/vcrpy.git
synced 2025-12-09 01:03:24 +00:00
7caf29735a
Format with line length 110 to match flake8 make black part of linting check Update travis spec for updated black requirements Add diff output for black on failure update changelog
34 lines
970 B
Python
34 lines
970 B
Python
import vcr
|
|
|
|
|
|
class MockSerializer(object):
|
|
def __init__(self):
|
|
self.serialize_count = 0
|
|
self.deserialize_count = 0
|
|
self.load_args = None
|
|
|
|
def deserialize(self, cassette_string):
|
|
self.serialize_count += 1
|
|
self.cassette_string = cassette_string
|
|
return {"interactions": []}
|
|
|
|
def serialize(self, cassette_dict):
|
|
self.deserialize_count += 1
|
|
return ""
|
|
|
|
|
|
def test_registered_serializer(tmpdir):
|
|
ms = MockSerializer()
|
|
my_vcr = vcr.VCR()
|
|
my_vcr.register_serializer("mock", ms)
|
|
tmpdir.join("test.mock").write("test_data")
|
|
with my_vcr.use_cassette(str(tmpdir.join("test.mock")), serializer="mock"):
|
|
# Serializer deserialized once
|
|
assert ms.serialize_count == 1
|
|
# and serialized the test data string
|
|
assert ms.cassette_string == "test_data"
|
|
# and hasn't serialized yet
|
|
assert ms.deserialize_count == 0
|
|
|
|
assert ms.serialize_count == 1
|