/
HttpSpanCollectorTest.java
120 lines (90 loc) · 3.7 KB
/
HttpSpanCollectorTest.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
package com.github.kristofa.brave.http;
import com.github.kristofa.brave.SpanCollectorMetricsHandler;
import com.twitter.zipkin.gen.Span;
import java.util.List;
import java.util.concurrent.atomic.AtomicInteger;
import okhttp3.mockwebserver.MockResponse;
import okhttp3.mockwebserver.MockWebServer;
import okhttp3.mockwebserver.RecordedRequest;
import okhttp3.mockwebserver.SocketPolicy;
import org.junit.Rule;
import org.junit.Test;
import zipkin.Codec;
import static org.assertj.core.api.Assertions.assertThat;
public class HttpSpanCollectorTest {
@Rule
public final MockWebServer server = new MockWebServer();
TestMetricsHander metrics = new TestMetricsHander();
// set flush interval to 0 so that tests can drive flushing explicitly
HttpSpanCollector.Config config = HttpSpanCollector.Config.builder().flushInterval(0).build();
HttpSpanCollector collector = new HttpSpanCollector(server.url("").toString(), config, metrics);
@Test
public void collectDoesntDoIO() throws Exception {
collector.collect(span(1L, "foo"));
assertThat(server.getRequestCount()).isZero();
}
@Test
public void collectIncrementsAcceptedMetrics() throws Exception {
collector.collect(span(1L, "foo"));
assertThat(metrics.acceptedSpans.get()).isEqualTo(1);
assertThat(metrics.droppedSpans.get()).isZero();
}
@Test
public void dropsWhenQueueIsFull() throws Exception {
for (int i = 0; i < 1001; i++)
collector.collect(span(1L, "foo"));
assertThat(metrics.acceptedSpans.get()).isEqualTo(1001);
assertThat(metrics.droppedSpans.get()).isEqualTo(1);
}
@Test
public void postsSpans() throws Exception {
server.enqueue(new MockResponse());
collector.collect(span(1L, "foo"));
collector.collect(span(2L, "bar"));
collector.flush(); // manually flush the spans
// Ensure a proper request was sent
RecordedRequest request = server.takeRequest();
assertThat(request.getRequestLine()).isEqualTo("POST /api/v1/spans HTTP/1.1");
assertThat(request.getHeader("Content-Type")).isEqualTo("application/x-thrift");
// Now, let's read back the spans we sent!
List<zipkin.Span> zipkinSpans = Codec.THRIFT.readSpans(request.getBody().readByteArray());
assertThat(zipkinSpans).containsExactly(
zipkinSpan(1L, "foo"),
zipkinSpan(2L, "bar")
);
}
@Test
public void incrementsDroppedSpansWhenServerErrors() throws Exception {
server.enqueue(new MockResponse().setResponseCode(500));
collector.collect(span(1L, "foo"));
collector.collect(span(2L, "bar"));
collector.flush(); // manually flush the spans
assertThat(metrics.droppedSpans.get()).isEqualTo(2);
}
@Test
public void incrementsDroppedSpansWhenServerDisconnects() throws Exception {
server.enqueue(new MockResponse().setSocketPolicy(SocketPolicy.DISCONNECT_AFTER_REQUEST));
collector.collect(span(1L, "foo"));
collector.collect(span(2L, "bar"));
collector.flush(); // manually flush the spans
assertThat(metrics.droppedSpans.get()).isEqualTo(2);
}
class TestMetricsHander implements SpanCollectorMetricsHandler {
final AtomicInteger acceptedSpans = new AtomicInteger();
final AtomicInteger droppedSpans = new AtomicInteger();
@Override
public void incrementAcceptedSpans(int quantity) {
acceptedSpans.addAndGet(quantity);
}
@Override
public void incrementDroppedSpans(int quantity) {
droppedSpans.addAndGet(quantity);
}
}
static Span span(long traceId, String spanName) {
return new Span().setTrace_id(traceId).setId(traceId).setName(spanName);
}
static zipkin.Span zipkinSpan(long traceId, String spanName) {
return new zipkin.Span.Builder().traceId(traceId).id(traceId).name(spanName).build();
}
}