RunnableWithTracing implements Runnable { public static RunnableWithTracing withTracing(Runnable origRunnable) { return new RunnableWithTracing(origRunnable); } RunnableWithTracing(Runnable origRunnable);  RunnableWithTracing(Runnable origRunnable,
                               Pair<Deque<Span>, Map<String, String>> originalThreadInfo);  RunnableWithTracing(Runnable origRunnable,
                               Deque<Span> spanStackForExecution,
                               Map<String, String> mdcContextMapForExecution); static RunnableWithTracing withTracing(Runnable origRunnable); static RunnableWithTracing withTracing(Runnable origRunnable,
                                                   Pair<Deque<Span>, Map<String, String>> originalThreadInfo); static RunnableWithTracing withTracing(Runnable origRunnable,
                                                   Deque<Span> spanStackForExecution,
                                                   Map<String, String> mdcContextMapForExecution); @Override @SuppressWarnings("deprecation") void run();  }
@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(); RunnableWithTracing instance = (useStaticFactory) ? withTracing(runnableMock) : new RunnableWithTracing(runnableMock); assertThat(instance.origRunnable).isSameAs(runnableMock); 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); RunnableWithTracing instance = (useStaticFactory) ? withTracing(runnableMock, Pair.of(spanStackMock, mdcInfoMock)) : new RunnableWithTracing(runnableMock, Pair.of(spanStackMock, mdcInfoMock) ); assertThat(instance.origRunnable).isSameAs(runnableMock); 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) { RunnableWithTracing instance = (useStaticFactory) ? withTracing(runnableMock, (Pair)null) : new RunnableWithTracing(runnableMock, (Pair)null); assertThat(instance.origRunnable).isSameAs(runnableMock); 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); RunnableWithTracing instance = (useStaticFactory) ? withTracing(runnableMock, spanStackMock, mdcInfoMock) : new RunnableWithTracing(runnableMock, spanStackMock, mdcInfoMock); assertThat(instance.origRunnable).isSameAs(runnableMock); 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 ThrowingCallable() { @Override public void call() throws Throwable { new RunnableWithTracing(null); } })).isInstanceOf(IllegalArgumentException.class); assertThat(catchThrowable(new ThrowingCallable() { @Override public void call() throws Throwable { withTracing(null); } })).isInstanceOf(IllegalArgumentException.class); assertThat(catchThrowable(new ThrowingCallable() { @Override public void call() throws Throwable { new RunnableWithTracing(null, Pair.of(spanStackMock, mdcInfoMock)); } })).isInstanceOf(IllegalArgumentException.class); assertThat(catchThrowable(new ThrowingCallable() { @Override public void call() throws Throwable { withTracing(null, Pair.of(spanStackMock, mdcInfoMock)); } })).isInstanceOf(IllegalArgumentException.class); assertThat(catchThrowable(new ThrowingCallable() { @Override public void call() throws Throwable { new RunnableWithTracing(null, spanStackMock, mdcInfoMock); } })).isInstanceOf(IllegalArgumentException.class); assertThat(catchThrowable(new ThrowingCallable() { @Override public void call() throws Throwable { withTracing(null, spanStackMock, mdcInfoMock); } })).isInstanceOf(IllegalArgumentException.class); }