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()