1
0
mirror of https://github.com/kevin1024/vcrpy.git synced 2025-12-09 01:03:24 +00:00
This commit is contained in:
Max Shytikov
2014-04-30 02:41:21 +02:00
parent 5d1f35973d
commit 7b253ebc6f

View File

@@ -1,10 +1,9 @@
'''The container for recorded requests and responses''' '''The container for recorded requests and responses'''
try: try:
from collections import Counter, OrderedDict from collections import Counter
except ImportError: except ImportError:
from .compat.counter import Counter from .compat.counter import Counter
from .compat.ordereddict import OrderedDict
from contextdecorator import ContextDecorator from contextdecorator import ContextDecorator
@@ -31,7 +30,7 @@ class Cassette(ContextDecorator):
path, path,
serializer=yamlserializer, serializer=yamlserializer,
record_mode='once', record_mode='once',
match_on=[uri, method]): match_on=[uri, method],
filter_headers=[], filter_headers=[],
filter_query_parameters=[], filter_query_parameters=[],
before_record=None, before_record=None,
@@ -70,10 +69,10 @@ class Cassette(ContextDecorator):
def append(self, request, response): def append(self, request, response):
'''Add a request, response pair to this cassette''' '''Add a request, response pair to this cassette'''
request = filter_request( request = filter_request(
request = request, request=request,
filter_headers = self._filter_headers, filter_headers=self._filter_headers,
filter_query_parameters = self._filter_query_parameters, filter_query_parameters=self._filter_query_parameters,
before_record = self._before_record before_record=self._before_record
) )
if not request: if not request:
return return
@@ -86,10 +85,10 @@ class Cassette(ContextDecorator):
the request. the request.
""" """
request = filter_request( request = filter_request(
request = request, request=request,
filter_headers = self._filter_headers, filter_headers=self._filter_headers,
filter_query_parameters = self._filter_query_parameters, filter_query_parameters=self._filter_query_parameters,
before_record = self._before_record before_record=self._before_record
) )
if not request: if not request:
return return