def test_tracer_custom_annotation(mocker, dummy_response, provider_stub):
put_annotation_mock = mocker.MagicMock()
annotation_key = "BookingId"
annotation_value = "123456"
provider = provider_stub(put_annotation_mock=put_annotation_mock)
tracer = Tracer(provider=provider, service="booking")
tracer.put_annotation(annotation_key, annotation_value)
assert put_annotation_mock.call_count == 1
assert put_annotation_mock.call_args == mocker.call(key=annotation_key,
value=annotation_value)
def test_tracer_custom_annotation(mocker, dummy_response, provider_stub):
# GIVEN Tracer is initialized
put_annotation_mock = mocker.MagicMock()
provider = provider_stub(put_annotation_mock=put_annotation_mock)
tracer = Tracer(provider=provider)
# WHEN put_metadata is used
annotation_key = "BookingId"
annotation_value = "123456"
tracer.put_annotation(annotation_key, annotation_value)
# THEN we should have an annotation as expected
assert put_annotation_mock.call_count == 1
assert put_annotation_mock.call_args == mocker.call(key=annotation_key, value=annotation_value)
