Exemple #1
0
    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
Exemple #2
0
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)