def wrapper(port, request, *args, **kwargs): session.add_sub_step(control_line="%s %s" % (request.method, request.url), body=request.data) try: for status, headers, body in func(port, request, *args, **kwargs): yield status, headers, body except Exception as e: session.add_sub_step(control_line='500', body=format_exc()) raise e else: content_to_log = utils.remove_base64_screenshot(body) session.add_sub_step(control_line=str(status), body=content_to_log) yield status, headers, body
def log_response(session, response, created=None): response_data = utils.remove_base64_screenshot(response.data) session.add_session_step(control_line=response.status_code, body=response_data, created=created)