Source code for testsuite.plugins.tracing

import uuid

import pytest

from testsuite.tracing import TraceidManager

_TRACE_ID_PREFIX = 'testsuite-'




@pytest.fixture(scope='session')
def testsuite_traceid_generator():
    """
    Fill free to override this fixture with our own.
    """

    def generator():
        return _TRACE_ID_PREFIX + uuid.uuid4().hex

    return generator





@pytest.fixture
def testsuite_trace_id(
    testsuite_traceid_generator, _testsuite_traceid_history
) -> str:
    """
    Testcase trace id.
    """
    trace_id = testsuite_traceid_generator()
    _testsuite_traceid_history.add(trace_id)
    return trace_id





@pytest.fixture
def testsuite_traceid_manager(
    testsuite_trace_id: str, _testsuite_traceid_history
) -> TraceidManager:
    """TraceidManager associated with current testcase.

    :returns: :py:class:`testsuite.tracing.TraceidManager`
    """
    return TraceidManager(testsuite_trace_id, _testsuite_traceid_history)



@pytest.fixture(scope='session')
def _testsuite_traceid_history():
    return set()