Search in sources :

Example 1 with DefaultRiposteProxyRouterSpanNamingAndTaggingStrategy

use of com.nike.riposte.server.config.distributedtracing.DefaultRiposteProxyRouterSpanNamingAndTaggingStrategy in project riposte by Nike-Inc.

the class ProxyRouterProcessingStateTest method beforeMethod.

@Before
public void beforeMethod() {
    stateSpy = spy(new ProxyRouterProcessingState());
    initialSpanNameFromStrategy = new AtomicReference<>("span-name-from-strategy-" + UUID.randomUUID().toString());
    strategyInitialSpanNameMethodCalled = new AtomicBoolean(false);
    strategyRequestTaggingMethodCalled = new AtomicBoolean(false);
    strategyResponseTaggingAndFinalSpanNameMethodCalled = new AtomicBoolean(false);
    strategyInitialSpanNameArgs = new AtomicReference<>(null);
    strategyRequestTaggingArgs = new AtomicReference<>(null);
    strategyResponseTaggingArgs = new AtomicReference<>(null);
    wingtipsStrategy = new ArgCapturingHttpTagAndSpanNamingStrategy<>(initialSpanNameFromStrategy, strategyInitialSpanNameMethodCalled, strategyRequestTaggingMethodCalled, strategyResponseTaggingAndFinalSpanNameMethodCalled, strategyInitialSpanNameArgs, strategyRequestTaggingArgs, strategyResponseTaggingArgs);
    wingtipsAdapterMock = mock(HttpTagAndSpanNamingAdapter.class);
    proxyTaggingStrategy = new DefaultRiposteProxyRouterSpanNamingAndTaggingStrategy(wingtipsStrategy, wingtipsAdapterMock);
    requestMock = mock(HttpRequest.class);
    responseMock = mock(HttpResponse.class);
    errorMock = mock(Throwable.class);
    spanMock = mock(Span.class);
    distributedTracingConfigMock = mock(DistributedTracingConfig.class);
    doReturn(proxyTaggingStrategy).when(distributedTracingConfigMock).getProxyRouterSpanNamingAndTaggingStrategy();
}
Also used : HttpRequest(io.netty.handler.codec.http.HttpRequest) AtomicBoolean(java.util.concurrent.atomic.AtomicBoolean) DistributedTracingConfig(com.nike.riposte.server.config.distributedtracing.DistributedTracingConfig) DefaultRiposteProxyRouterSpanNamingAndTaggingStrategy(com.nike.riposte.server.config.distributedtracing.DefaultRiposteProxyRouterSpanNamingAndTaggingStrategy) HttpResponse(io.netty.handler.codec.http.HttpResponse) Assertions.catchThrowable(org.assertj.core.api.Assertions.catchThrowable) HttpTagAndSpanNamingAdapter(com.nike.wingtips.tags.HttpTagAndSpanNamingAdapter) Span(com.nike.wingtips.Span) Before(org.junit.Before)

Aggregations

DefaultRiposteProxyRouterSpanNamingAndTaggingStrategy (com.nike.riposte.server.config.distributedtracing.DefaultRiposteProxyRouterSpanNamingAndTaggingStrategy)1 DistributedTracingConfig (com.nike.riposte.server.config.distributedtracing.DistributedTracingConfig)1 Span (com.nike.wingtips.Span)1 HttpTagAndSpanNamingAdapter (com.nike.wingtips.tags.HttpTagAndSpanNamingAdapter)1 HttpRequest (io.netty.handler.codec.http.HttpRequest)1 HttpResponse (io.netty.handler.codec.http.HttpResponse)1 AtomicBoolean (java.util.concurrent.atomic.AtomicBoolean)1 Assertions.catchThrowable (org.assertj.core.api.Assertions.catchThrowable)1 Before (org.junit.Before)1