def testRequestImage(self): img_handler = requestreplacer.ImageHandler() img_handler.add_resizer(0.5) replacer = Replacer([('image/jpeg', img_handler)]) reqfilter = mox.MockAnything() logger = mox.MockAnything() response = ProxyResponse() response.setheaders('Content-Type: image/jpeg\r\n') response.version = 'HTTP/1.0' response.status = 200 response.reason = '' response.setbody(open('data/image_resp_body.jpg', 'rb')) class Fetcher: def fetch(self, host, path, body, headers): return response fetcher = Fetcher() self.server.reqfilter = reqfilter self.server.replacer = replacer self.server.fetcher = fetcher self.server.logger = logger # commands = ProxyCommands(reqfilter, replacer, fetcher, logger) req_handler = ProxyRequestHandler(self.request_mock, self.client_addr, self.server) mox.Verify(self.request_mock)
def testLogging(self): logger = mox.MockObject(RequestLogger) logger.log(self.url, True) mox.Replay(logger) response = ProxyResponse() response.setheaders('Content-Type: image/jpeg\r\n') response.version = 'HTTP/1.0' response.status = 200 response.reason = '' response.setbody(open('data/image_resp_body.jpg', 'rb')) class Fetcher: def fetch(self, host, path, body, headers): return response fetcher = Fetcher() self.server.logger = logger self.server.fetcher = fetcher req_handler = ProxyRequestHandler(self.request_mock, self.client_addr, self.server) mox.Verify(self.request_mock) mox.Verify(logger)