Merge pull request #27 from onna/fix-streamreader

Fix streamreader init

CHANGELOG.rst

@@ -1,6 +1,11 @@
 Changelog
 =========
 
+2.1.1
+-----
+
+- Add required parameter to streamreader
+
 2.1.0
 -----
 

cassettedeck/service.py

@@ -5,6 +5,7 @@
 from aiohttp import hdrs
 from aiohttp.helpers import TimerNoop
 from unittest.mock import Mock
+from cassettedeck.store import DEFAULT_STREAM_LIMIT
 import json
 import asyncio
 
@@ -66,7 +67,7 @@ async def build_response(self, session, method, url,
         resp._headers = CIMultiDict({hdrs.CONTENT_TYPE: ct})
         loop = asyncio.get_event_loop()
         protocol = Mock(_reading_paused=False)
-        resp.content = StreamReader(protocol, loop=loop)
+        resp.content = StreamReader(protocol, loop=loop, limit=DEFAULT_STREAM_LIMIT)
         if isinstance(data, str):
             data = data.encode('utf8')
         resp.content.feed_data(data)

cassettedeck/store.py

@@ -12,6 +12,8 @@
 
 default_library = os.path.join(os.path.dirname(__file__), 'cassettes/')
 
+DEFAULT_STREAM_LIMIT = 6 ** 16
+
 
 class CassetteStore(object):
     """This class wraps all logic related to cassettes, such as storing
@@ -176,7 +178,7 @@ def build_response(self, method, url, params, data, headers):
 
         # Set headers and content
         resp._headers = CIMultiDict(resp_json['headers'])
-        resp.content = StreamReader(Mock())
+        resp.content = StreamReader(Mock(), limit=DEFAULT_STREAM_LIMIT)
 
         # Get the data
         data = resp_json['body']['data']

setup.py

@@ -4,7 +4,7 @@
 
 setup(
     name='cassettedeck',
-    version='2.1.0',
+    version='2.1.1',
     description='A library store and replay aiohttp requests',
     long_description='To simplify and speed up tests that make HTTP requests',
     author='Developer team at Onna Technologies',