@Override protected @Nullable String doGetInitialSpanName( @NotNull REQ request, @NotNull HttpTagAndSpanNamingAdapter<REQ, ?> adapter ) { initialSpanNameMethodCalled.set(true); initialSpanNameArgs.set(new InitialSpanNameArgs(request, adapter)); return initialSpanName.get(); }
@Test public void doGetInitialSpanName_delegates_to_wingtips_strategy() { // when String result = impl.doGetInitialSpanName(requestMock); // then assertThat(result).isEqualTo(initialSpanNameFromStrategy.get()); strategyInitialSpanNameArgs.get().verifyArgs(requestMock, wingtipsAdapterMock); }
strategyInitialSpanNameArgs.get().verifyArgs(requestInfoMock, tagAndNamingAdapterMock); strategyRequestTaggingArgs.get().verifyArgs(span, requestInfoMock, tagAndNamingAdapterMock);
assertThat(span.isSampleable(), is(Boolean.valueOf(parentTraceEnabled))); strategyInitialSpanNameArgs.get().verifyArgs(requestInfoMock, tagAndNamingAdapterMock); strategyRequestTaggingArgs.get().verifyArgs(span, requestInfoMock, tagAndNamingAdapterMock);
@UseDataProvider("getSpanNameScenarioDataProvider") @Test public void getSpanName_works_as_expected(GetSpanNameScenario scenario) { // given DTraceStartHandler handlerSpy = spy(handler); initialSpanNameFromStrategy.set(scenario.strategySpanName); doReturn(scenario.fallbackSpanName).when(handlerSpy).getFallbackSpanName(any()); if (scenario.riposteRequestInfoIsNull) { requestInfoMock = null; } String expectedResult = (scenario.expectStrategyResult) ? scenario.strategySpanName : scenario.fallbackSpanName; // when String result = handlerSpy.getSpanName( httpRequest, requestInfoMock, distributedTracingConfig.getServerSpanNamingAndTaggingStrategy() ); // then Assertions.assertThat(result).isEqualTo(expectedResult); if (!scenario.riposteRequestInfoIsNull) { strategyInitialSpanNameArgs.get().verifyArgs(requestInfoMock, tagAndNamingAdapterMock); } if (!scenario.expectStrategyResult) { verify(handlerSpy).getFallbackSpanName(httpRequest); } }
@Test public void doGetInitialSpanName_delegates_to_wingtips_strategy() { // when String result = impl.doGetInitialSpanName(requestMock); // then assertThat(result).isEqualTo(initialSpanNameFromStrategy.get()); strategyInitialSpanNameArgs.get().verifyArgs(requestMock, wingtipsAdapterMock); }