tracing.py

import uuid
 
import pytest
 
from testsuite.tracing import TraceidManager
 
 
@pytest.fixture(scope='session')
def testsuite_traceid_generator():
    """
    Fill free to override this fixture with our own.
 
    @ingroup userver_testsuite_fixtures
    Part of the [yandex-taxi-testsuite](https://github.com/yandex/yandex-taxi-testsuite/blob/develop/testsuite/plugins/tracing.py#L9)
    """
 
    def generator():
        return uuid.uuid4().hex
 
    return generator
 
 
@pytest.fixture
def testsuite_trace_id(
    testsuite_traceid_generator, _testsuite_traceid_history
) -> str:
    """
    Testcase trace id.
 
    @ingroup userver_testsuite_fixtures
    Part of the [yandex-taxi-testsuite](https://github.com/yandex/yandex-taxi-testsuite/blob/develop/testsuite/plugins/tracing.py#L21)
    """
    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 @ref testsuite.tracing.TraceidManager
 
    @ingroup userver_testsuite_fixtures
    Part of the [yandex-taxi-testsuite](https://github.com/yandex/yandex-taxi-testsuite/blob/develop/testsuite/plugins/tracing.py#L33)
    """
    return TraceidManager(testsuite_trace_id, _testsuite_traceid_history)
 
 
@pytest.fixture(scope='session')
def _testsuite_traceid_history():
    return set()