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)
