Beispiel #1
0
def test_clear_trace_id():
    DistributedTracingContext.set_trace_id("trace-id")
    DistributedTracingContext.clear()

    assert not DistributedTracingContext.has_trace_id()
    assert DistributedTracingContext.get_trace_id() == ""
    assert not DistributedTracingContext.is_trace_id_set_by_user()
Beispiel #2
0
def test_set_trace_id():
    trace_id = "trace-id"
    DistributedTracingContext.set_trace_id(trace_id)

    assert DistributedTracingContext.has_trace_id()
    assert DistributedTracingContext.get_trace_id() == trace_id
    assert DistributedTracingContext.is_trace_id_set_by_user()
Beispiel #3
0
async def test_add_x_trace_id_no_trace_id_set_args_length_too_short():
    args = ["a", 2, "b", 3]
    func_return_value = "value"

    func = AsyncMock()
    func.return_value = func_return_value

    actual_return_value = await add_x_trace_id(func)(*deepcopy(args))

    assert DistributedTracingContext.has_trace_id()
    assert len(DistributedTracingContext.get_trace_id()) == TRACE_ID_LENGTH

    assert actual_return_value == func_return_value
    func.assert_awaited_once_with(*args)
Beispiel #4
0
def test_set_new_trace_id():
    DistributedTracingContext.set_new_trace_id()

    assert DistributedTracingContext.has_trace_id()
    assert len(DistributedTracingContext.get_trace_id()) == TRACE_ID_LENGTH
    assert not DistributedTracingContext.is_trace_id_set_by_user()
Beispiel #5
0
def test_empty_tracing_context():
    assert not DistributedTracingContext.has_trace_id()
    assert DistributedTracingContext.get_trace_id() == ""
    assert not DistributedTracingContext.is_trace_id_set_by_user()