def testHandlerNotAvail(self):
resp_mock = create_response_mock()
resp_mock.getheader("Content-Type").AndReturn("text/html")
mox.Replay(resp_mock)
replacer = Replacer([])
resp = replacer.replace(resp_mock)
mox.Verify(resp_mock)
assert resp == resp_mock
def testHandlerAvail(self):
resp_mock = create_response_mock()
resp_mock.getheader("Content-Type").AndReturn("application/xhtml+xml")
mox.Replay(resp_mock)
handler_mock = mox.MockObject(HTMLHandler)
handler_mock.handle(resp_mock).AndReturn(resp_mock)
mox.Replay(handler_mock)
replacer = Replacer([('text/html|application/xhtml\+xml', handler_mock)])
replacer.replace(resp_mock)
mox.Verify(resp_mock)
mox.Verify(handler_mock)
def testResize(self):
ratio = 0.5
resp_mock = create_response_mock()
resp_mock.getheader("Content-Type").AndReturn("image/jpeg")
resp_mock.read().AndReturn(self.img_data)
cmp_len = lambda new_data: len(new_data) < len(self.img_data)
resp_mock.setheader('Content-Length', mox.Func(cmp_len))
cmp_size = lambda im: self._img_cmp_size(self.img, Image.open(im), ratio)
resp_mock.setbody(mox.Func(cmp_size))
mox.Replay(resp_mock)
img_handler = ImageHandler()
img_handler.add_resizer(ratio)
replacer = Replacer([('image/jpeg', img_handler)])
resp = replacer.replace(resp_mock)
mox.Verify(resp_mock)
