brave.propagation.Propagation.Getter Java Examples

@Test
public void testGetterOnGetSpanId() {
  Getter<Invocation, String> getter = Deencapsulation
      .getField(ZipkinProviderDelegate.class, "INVOCATION_STRING_GETTER");

  Invocation invocation = Mockito.mock(Invocation.class);
  Map<String, String> context = new HashMap<>();

  Mockito.when(invocation.getContext()).thenReturn(context);

  // if there is no spanId or traceId, then result is null
  String spanId = getter.get(invocation, ZipkinProviderDelegate.SPAN_ID_HEADER_NAME);
  Assert.assertNull(spanId);

  // if there is no spanId but traceId, then traceId will be returned as result
  final String testTraceId = "testTraceId";
  context.put(ZipkinProviderDelegate.TRACE_ID_HEADER_NAME, testTraceId);
  spanId = getter.get(invocation, ZipkinProviderDelegate.SPAN_ID_HEADER_NAME);
  Assert.assertEquals(testTraceId, spanId);

  // if there is spanId, then spanId will be returned
  final String testSpanId = "testSpanId";
  context.put(ZipkinProviderDelegate.SPAN_ID_HEADER_NAME, testSpanId);
  spanId = getter.get(invocation, ZipkinProviderDelegate.SPAN_ID_HEADER_NAME);
  Assert.assertEquals(testSpanId, spanId);
}
 

@Test
public void testGetterOnGetOtherContent() {
  Getter<Invocation, String> getter = Deencapsulation
      .getField(ZipkinProviderDelegate.class, "INVOCATION_STRING_GETTER");

  Invocation invocation = Mockito.mock(Invocation.class);
  Map<String, String> context = new HashMap<>();

  Mockito.when(invocation.getContext()).thenReturn(context);

  final String key = "key";
  String value = getter.get(invocation, key);
  Assert.assertNull(value);

  final String expectedValue = "value";
  context.put(key, expectedValue);
  value = getter.get(invocation, key);
  Assert.assertEquals(expectedValue, value);
}
 

TextMapExtractor(
    Propagation<String> propagation,
    Set<String> allNames,
    Getter<Map<String, String>, String> getter) {
  this.allNames = allNames;
  this.delegate = propagation.extractor(getter);
}
 

TracingMethodMessageHandlerAdapter(MessagingTracing messagingTracing,
		Getter<MessageHeaderAccessor, String> getter) {
	this.tracing = messagingTracing.tracing();
	this.tracer = tracing.tracer();
	this.extractor = tracing.propagation().extractor(MessageConsumerRequest.GETTER);
	this.getter = getter;
}
 

TextMapExtractor(
    Propagation<String> propagation,
    Set<String> allNames,
    Getter<Map<String, String>, String> getter) {
  this.allNames = allNames;
  this.delegate = propagation.extractor(getter);
}
 

@Test public void getterAdapter() {
  Map<String, Integer> map = new LinkedHashMap<>();
  map.put("1", 1);
  map.put("2", 2);

  Getter<Map<Integer, String>, String> wrappedGetter = new GetterAdapter<>(getter, map);

  Map<Integer, String> request = new LinkedHashMap<>();
  request.put(1, "one");
  request.put(2, "two");

  assertThat(wrappedGetter.get(request, "1")).isEqualTo("one");
  assertThat(wrappedGetter.get(request, "2")).isEqualTo("two");
  assertThat(wrappedGetter.get(request, "3")).isNull();
}
 

private Getter<Invocation, String> extractor() {
  return INVOCATION_STRING_GETTER;
}
 

AWSExtractor(AWSPropagation propagation, Getter<R, String> getter) {
  this.propagation = propagation;
  this.getter = getter;
}
 

XCloudTraceContextExtractor(Propagation<String> primary, Getter<R, String> getter) {
  this.primary = primary.extractor(getter);
  this.getter = getter;
}
 

MessageConsumerRequest(Message delegate,
		Getter<MessageHeaderAccessor, String> getter) {
	this.delegate = delegate;
	this.mutableHeaders = MessageHeaderAccessor.getMutableAccessor(delegate);
	this.getter = getter;
}
 

@Test public void getter_equalsHashCodeString() {
  assertDelegates(
      () -> mock(Getter.class),
      g -> new StringPropagationAdapter.GetterAdapter<>(g, Collections.emptyMap())
  );
}