Search in sources :

Example 1 with CLIENT_SPAN

use of zipkin2.TestObjects.CLIENT_SPAN in project zipkin by openzipkin.

the class ITSpanConsumer method insertEntry_niceToString.

/**
 * It is easier to use a real Cassandra connection than mock a prepared statement.
 */
@Test
public void insertEntry_niceToString() {
    // This test can use fake data as it is never written to cassandra
    Span clientSpan = CLIENT_SPAN;
    AggregateCall<?, ?> acceptCall = (AggregateCall<?, ?>) storage.spanConsumer().accept(asList(clientSpan));
    List<Call<?>> insertEntryCalls = acceptCall.delegate().stream().filter(c -> c instanceof InsertEntry).collect(Collectors.toList());
    assertThat(insertEntryCalls.get(0)).hasToString("INSERT INTO span_by_service (service, span) VALUES (frontend,get)");
    assertThat(insertEntryCalls.get(1)).hasToString("INSERT INTO remote_service_by_service (service, remote_service) VALUES (frontend,backend)");
}
Also used : AggregateCall(zipkin2.internal.AggregateCall) IntStream(java.util.stream.IntStream) TestObjects.spanBuilder(zipkin2.TestObjects.spanBuilder) TestObjects.newClientSpan(zipkin2.TestObjects.newClientSpan) AggregateCall(zipkin2.internal.AggregateCall) Assertions.assertThat(org.assertj.core.api.Assertions.assertThat) Span(zipkin2.Span) InsertEntry(zipkin2.storage.cassandra.internal.call.InsertEntry) CLIENT_SPAN(zipkin2.TestObjects.CLIENT_SPAN) Collectors(java.util.stream.Collectors) TestInfo(org.junit.jupiter.api.TestInfo) Test(org.junit.jupiter.api.Test) List(java.util.List) Arrays.asList(java.util.Arrays.asList) SERVER(zipkin2.Span.Kind.SERVER) StorageComponent(zipkin2.storage.StorageComponent) Call(zipkin2.Call) ITStorage(zipkin2.storage.ITStorage) AggregateCall(zipkin2.internal.AggregateCall) Call(zipkin2.Call) InsertEntry(zipkin2.storage.cassandra.internal.call.InsertEntry) TestObjects.newClientSpan(zipkin2.TestObjects.newClientSpan) Span(zipkin2.Span) Test(org.junit.jupiter.api.Test)

Example 2 with CLIENT_SPAN

use of zipkin2.TestObjects.CLIENT_SPAN in project zipkin by openzipkin.

the class BulkRequestBenchmarks method buildAndWriteRequest_singleSpan.

@Benchmark
public HttpRequest buildAndWriteRequest_singleSpan() {
    BulkCallBuilder builder = new BulkCallBuilder(es, V6_0, "index-span");
    builder.index(spanIndex, "span", CLIENT_SPAN, BulkIndexWriter.SPAN);
    HttpCall.RequestSupplier supplier = builder.build().request;
    HttpRequestWriter request = HttpRequest.streaming(supplier.headers());
    supplier.writeBody(request::tryWrite);
    return request;
}
Also used : HttpCall(zipkin2.elasticsearch.internal.client.HttpCall) HttpRequestWriter(com.linecorp.armeria.common.HttpRequestWriter) Benchmark(org.openjdk.jmh.annotations.Benchmark)

Example 3 with CLIENT_SPAN

use of zipkin2.TestObjects.CLIENT_SPAN in project zipkin by openzipkin.

the class BulkRequestBenchmarks method buildAndWriteRequest_tenSpans.

@Benchmark
public HttpRequest buildAndWriteRequest_tenSpans() {
    BulkCallBuilder builder = new BulkCallBuilder(es, V6_0, "index-span");
    for (int i = 0; i < 10; i++) {
        builder.index(spanIndex, "span", CLIENT_SPAN, BulkIndexWriter.SPAN);
    }
    HttpCall.RequestSupplier supplier = builder.build().request;
    HttpRequestWriter request = HttpRequest.streaming(supplier.headers());
    supplier.writeBody(request::tryWrite);
    return request;
}
Also used : HttpCall(zipkin2.elasticsearch.internal.client.HttpCall) HttpRequestWriter(com.linecorp.armeria.common.HttpRequestWriter) Benchmark(org.openjdk.jmh.annotations.Benchmark)

Example 4 with CLIENT_SPAN

use of zipkin2.TestObjects.CLIENT_SPAN in project zipkin by openzipkin.

the class CollectorTest method storeSpansCallback_toStringIncludesSpanIds.

@Test
public void storeSpansCallback_toStringIncludesSpanIds() {
    Span span2 = CLIENT_SPAN.toBuilder().id("3").build();
    when(collector.idString(span2)).thenReturn("3");
    assertThat(collector.new StoreSpans(asList(CLIENT_SPAN, span2))).hasToString("StoreSpans([1, 3])");
}
Also used : Span(zipkin2.Span) Test(org.junit.Test)

Aggregations

HttpRequestWriter (com.linecorp.armeria.common.HttpRequestWriter)2 Benchmark (org.openjdk.jmh.annotations.Benchmark)2 Span (zipkin2.Span)2 HttpCall (zipkin2.elasticsearch.internal.client.HttpCall)2 Arrays.asList (java.util.Arrays.asList)1 List (java.util.List)1 Collectors (java.util.stream.Collectors)1 IntStream (java.util.stream.IntStream)1 Assertions.assertThat (org.assertj.core.api.Assertions.assertThat)1 Test (org.junit.Test)1 Test (org.junit.jupiter.api.Test)1 TestInfo (org.junit.jupiter.api.TestInfo)1 Call (zipkin2.Call)1 SERVER (zipkin2.Span.Kind.SERVER)1 CLIENT_SPAN (zipkin2.TestObjects.CLIENT_SPAN)1 TestObjects.newClientSpan (zipkin2.TestObjects.newClientSpan)1 TestObjects.spanBuilder (zipkin2.TestObjects.spanBuilder)1 AggregateCall (zipkin2.internal.AggregateCall)1 ITStorage (zipkin2.storage.ITStorage)1 StorageComponent (zipkin2.storage.StorageComponent)1