FunctionWithTracing implements Function<T, U> { public static <T, U> FunctionWithTracing<T, U> withTracing(Function<T, U> origFunction) { return new FunctionWithTracing<>(origFunction); } FunctionWithTracing(Function<T, U> origFunction);  FunctionWithTracing(Function<T, U> origFunction,
                               Pair<Deque<Span>, Map<String, String>> originalThreadInfo);  FunctionWithTracing(Function<T, U> origFunction,
                               Deque<Span> spanStackForExecution,
                               Map<String, String> mdcContextMapForExecution); static FunctionWithTracing<T, U> withTracing(Function<T, U> origFunction); static FunctionWithTracing<T, U> withTracing(
        Function<T, U> origFunction,
        Pair<Deque<Span>, Map<String, String>> originalThreadInfo
    ); static FunctionWithTracing<T, U> withTracing(Function<T, U> origFunction,
                                                               Deque<Span> spanStackForExecution,
                                                               Map<String, String> mdcContextMapForExecution); @Override U apply(T t);  }
@DataProvider(value = { "true", "false" }) @Test public void current_thread_info_constructor_sets_fields_as_expected(boolean useStaticFactory) { Tracer.getInstance().startRequestWithRootSpan("request-" + UUID.randomUUID().toString()); Deque<Span> spanStackMock = Tracer.getInstance().getCurrentSpanStackCopy(); Map<String, String> mdcInfoMock = MDC.getCopyOfContextMap(); FunctionWithTracing instance = (useStaticFactory) ? withTracing(functionMock) : new FunctionWithTracing(functionMock); assertThat(instance.origFunction).isSameAs(functionMock); assertThat(instance.spanStackForExecution).isEqualTo(spanStackMock); assertThat(instance.mdcContextMapForExecution).isEqualTo(mdcInfoMock); }
@DataProvider(value = { "true | true | true", "true | false | true", "false | true | true", "false | false | true", "true | true | false", "true | false | false", "false | true | false", "false | false | false", }, splitBy = "\\|") @Test public void pair_constructor_sets_fields_as_expected( boolean nullSpanStack, boolean nullMdcInfo, boolean useStaticFactory ) { Deque<Span> spanStackMock = (nullSpanStack) ? null : mock(Deque.class); Map<String, String> mdcInfoMock = (nullMdcInfo) ? null : mock(Map.class); FunctionWithTracing instance = (useStaticFactory) ? withTracing(functionMock, Pair.of(spanStackMock, mdcInfoMock)) : new FunctionWithTracing(functionMock, Pair.of(spanStackMock, mdcInfoMock) ); assertThat(instance.origFunction).isSameAs(functionMock); assertThat(instance.spanStackForExecution).isEqualTo(spanStackMock); assertThat(instance.mdcContextMapForExecution).isEqualTo(mdcInfoMock); }
@DataProvider(value = { "true", "false" }) @Test public void pair_constructor_sets_fields_as_expected_when_pair_is_null(boolean useStaticFactory) { FunctionWithTracing instance = (useStaticFactory) ? withTracing(functionMock, (Pair)null) : new FunctionWithTracing(functionMock, (Pair)null); assertThat(instance.origFunction).isSameAs(functionMock); assertThat(instance.spanStackForExecution).isNull(); assertThat(instance.mdcContextMapForExecution).isNull(); }
@DataProvider(value = { "true", "false" }) @Test public void kitchen_sink_constructor_sets_fields_as_expected(boolean useStaticFactory) { Deque<Span> spanStackMock = mock(Deque.class); Map<String, String> mdcInfoMock = mock(Map.class); FunctionWithTracing instance = (useStaticFactory) ? withTracing(functionMock, spanStackMock, mdcInfoMock) : new FunctionWithTracing(functionMock, spanStackMock, mdcInfoMock); assertThat(instance.origFunction).isSameAs(functionMock); assertThat(instance.spanStackForExecution).isEqualTo(spanStackMock); assertThat(instance.mdcContextMapForExecution).isEqualTo(mdcInfoMock); }
@Test public void constructors_throw_exception_if_passed_null_operator() { final Deque<Span> spanStackMock = mock(Deque.class); final Map<String, String> mdcInfoMock = mock(Map.class); assertThat(catchThrowable(() -> new FunctionWithTracing(null))) .isInstanceOf(IllegalArgumentException.class); assertThat(catchThrowable(() -> withTracing(null))) .isInstanceOf(IllegalArgumentException.class); assertThat(catchThrowable(() -> new FunctionWithTracing(null, Pair.of(spanStackMock, mdcInfoMock)))) .isInstanceOf(IllegalArgumentException.class); assertThat(catchThrowable(() -> withTracing(null, Pair.of(spanStackMock, mdcInfoMock)))) .isInstanceOf(IllegalArgumentException.class); assertThat(catchThrowable(() -> new FunctionWithTracing(null, spanStackMock, mdcInfoMock))) .isInstanceOf(IllegalArgumentException.class); assertThat(catchThrowable(() -> withTracing(null, spanStackMock, mdcInfoMock))) .isInstanceOf(IllegalArgumentException.class); }