class FakeResponse(object):
def __init__(self, status, body='', headers=None, slow=0):
self.status = status
self.body = body
self.headers = HeadersDict(headers)
self.stream = BytesIO(body)
self.slow = slow
def getheader(self, name, default=None):
return self.headers.get(name, default)
def getheaders(self):
if 'Content-Length' not in self.headers:
self.headers['Content-Length'] = len(self.body)
return self.headers.items()
def _slow(self):
sleep(self.slow)
def read(self, amt=0):
if self.slow:
self._slow()
return self.stream.read(amt)
def __repr__(self):
return 'FakeResponse(status=%s)' % self.status
def __init__(self, status, body='', headers=None, slow=0):
self.status = status
self.body = body
self.headers = HeadersDict(headers)
self.stream = BytesIO(body)
self.slow = slow
def getheaders(self):
headers = HeadersDict({
'content-length': len(self.body),
})
headers.update(self.headers)
return headers.items()
