Python get_span_and_trace Examples

Example #1

File: test_setup.py Project: uk-gov-mirror/ONSdigital.eq-questionnaire-runner

def test_get_span_and_trace_no_xcloud_header():
    header = cloud_trace_header()
    del header["X-Cloud-Trace-Context"]  # pylint: disable=no-member
    span, trace = get_span_and_trace(header)  # pylint: disable=no-member

    assert trace is None
    assert span is None

Example #2

    def before_request():  # pylint: disable=unused-variable
        request_id = str(uuid4())
        logger.new(request_id=request_id)

        span, trace = get_span_and_trace(flask_request.headers)
        if span and trace:
            logger.bind(span=span, trace=trace)

        logger.info(
            "request",
            method=flask_request.method,
            url_path=flask_request.full_path,
            session_cookie_present="session" in flask_request.cookies,
            csrf_token_present="csrf_token" in cookie_session,
            user_agent=flask_request.user_agent.string,
        )

Example #3

File: test_setup.py Project: uk-gov-mirror/ONSdigital.eq-questionnaire-runner

def test_get_span_and_trace():
    header = cloud_trace_header()
    span, trace = get_span_and_trace(header)  # pylint: disable=no-member
    assert trace == "0123456789"
    assert span == "0123456789012345678901"

Example #4

File: test_setup.py Project: uk-gov-mirror/ONSdigital.eq-questionnaire-runner

def test_get_span_and_trace_malformed_xcloud_header():  # pylint: disable=no-member
    header = cloud_trace_header()
    header["X-Cloud-Trace-Context"] = "not a real trace context"
    span, trace = get_span_and_trace(header)  # pylint: disable=no-member
    assert trace is None
    assert span is None