def test_response_processing(self, *mocks):
from flaskext.gae_mini_profiler import GAEMiniProfiler
app = mock.Mock()
rendering = "GAEMiniProfiler"
class Response(object):
"""Mock response"""
status_code = 200
is_sequence = True
charset = "utf-8"
data = u"<body>Hello World!</body>"
with mock.patch_object(GAEMiniProfiler, '_render') as render_mock:
render_mock.return_value = rendering
ext = GAEMiniProfiler(app)
new_response = ext._process_response(Response())
self.assertEquals([u"<body>Hello World!GAEMiniProfiler</body>"],
new_response.response)
