def _filter_data(self, data, request_id):
body_with_script = configuration.generate_script_tags(request_id) + '</body>'
return data.replace('</body>', body_with_script)
def middleware_inserts_script_tags_in_returned_data():
middleware = Middleware(create_application())
response, request_id = output_and_id_from_application(middleware)
script_tags = configuration.generate_script_tags(request_id)
expected = '<html><body>{0}</body></html>'.format(script_tags)
assert_equal(expected, response)
