1
0
mirror of https://github.com/kevin1024/vcrpy.git synced 2025-12-09 17:15:35 +00:00

Merge pull request #437 from steinnes/minimal-aiohttp-streamreader-support

Minimal aiohttp streamreader support
This commit is contained in:
Arthur Hamon
2019-07-01 07:16:54 +02:00
committed by GitHub
3 changed files with 26 additions and 1 deletions

View File

@@ -5,12 +5,16 @@ import asyncio
import functools
import json
from aiohttp import ClientResponse
from aiohttp import ClientResponse, streams
from yarl import URL
from vcr.request import Request
class MockStream(asyncio.StreamReader, streams.AsyncStreamReaderMixin):
pass
class MockClientResponse(ClientResponse):
def __init__(self, method, url):
super().__init__(
@@ -37,6 +41,13 @@ class MockClientResponse(ClientResponse):
def release(self):
pass
@property
def content(self):
s = MockStream()
s.feed_data(self._body)
s.feed_eof()
return s
def vcr_request(cassette, real_request):
@functools.wraps(real_request)