|
1 | 1 | package datadog.opentracing; |
2 | 2 |
|
3 | | -import static datadog.trace.api.sampling.PrioritySampling.SAMPLER_DROP; |
4 | | -import static datadog.trace.api.sampling.PrioritySampling.SAMPLER_KEEP; |
5 | 3 | import static datadog.trace.api.sampling.PrioritySampling.UNSET; |
6 | | -import static datadog.trace.api.sampling.PrioritySampling.USER_DROP; |
7 | | -import static datadog.trace.api.sampling.PrioritySampling.USER_KEEP; |
8 | 4 | import static datadog.trace.api.sampling.SamplingMechanism.AGENT_RATE; |
9 | | -import static datadog.trace.api.sampling.SamplingMechanism.DEFAULT; |
10 | | -import static datadog.trace.api.sampling.SamplingMechanism.MANUAL; |
11 | 5 | import static org.junit.jupiter.api.Assertions.assertEquals; |
12 | 6 | import static org.junit.jupiter.api.Assertions.assertNotNull; |
13 | 7 | import static org.junit.jupiter.api.Assertions.assertTrue; |
14 | | -import static org.junit.jupiter.params.provider.Arguments.arguments; |
15 | 8 |
|
16 | 9 | import datadog.trace.api.DDSpanId; |
17 | 10 | import datadog.trace.api.DDTraceId; |
18 | 11 | import datadog.trace.api.internal.util.LongStringUtils; |
19 | 12 | import datadog.trace.common.writer.ListWriter; |
20 | 13 | import datadog.trace.core.DDSpan; |
21 | 14 | import datadog.trace.core.DDSpanContext; |
| 15 | +import datadog.trace.junit.utils.tabletest.PrioritySamplingConverter; |
| 16 | +import datadog.trace.junit.utils.tabletest.SamplingMechanismConverter; |
22 | 17 | import datadog.trace.test.util.DDJavaSpecification; |
23 | 18 | import io.opentracing.Scope; |
24 | 19 | import io.opentracing.Tracer; |
|
28 | 23 | import java.util.HashMap; |
29 | 24 | import java.util.Iterator; |
30 | 25 | import java.util.Map; |
31 | | -import java.util.stream.Stream; |
32 | 26 | import org.junit.jupiter.api.AfterEach; |
33 | 27 | import org.junit.jupiter.api.Test; |
34 | 28 | import org.junit.jupiter.params.ParameterizedTest; |
35 | | -import org.junit.jupiter.params.provider.MethodSource; |
| 29 | +import org.junit.jupiter.params.converter.ConvertWith; |
36 | 30 | import org.tabletest.junit.TableTest; |
37 | 31 |
|
38 | 32 | // This test focuses on things that are different between OpenTracing API 0.31.0 and 0.32.0 |
@@ -91,19 +85,20 @@ void testScopeManager(String scenario, boolean finishSpan) { |
91 | 85 | assertEquals(finishSpan, ((DDSpan) ((OTSpan) span).getDelegate()).isFinished()); |
92 | 86 | } |
93 | 87 |
|
94 | | - static Stream<org.junit.jupiter.params.provider.Arguments> testInjectExtractArguments() { |
95 | | - return Stream.of( |
96 | | - arguments("sampler drop", SAMPLER_DROP, DEFAULT, SAMPLER_DROP), |
97 | | - arguments("sampler keep", SAMPLER_KEEP, DEFAULT, SAMPLER_KEEP), |
98 | | - arguments("unset", UNSET, DEFAULT, SAMPLER_KEEP), |
99 | | - arguments("user keep", USER_KEEP, MANUAL, USER_KEEP), |
100 | | - arguments("user drop", USER_DROP, MANUAL, USER_DROP)); |
101 | | - } |
102 | | - |
103 | 88 | @ParameterizedTest |
104 | | - @MethodSource("testInjectExtractArguments") |
| 89 | + @TableTest({ |
| 90 | + "scenario | contextPriority | samplingMechanism | propagatedPriority ", |
| 91 | + "sampler drop | PrioritySampling.SAMPLER_DROP | SamplingMechanism.DEFAULT | PrioritySampling.SAMPLER_DROP", |
| 92 | + "sampler keep | PrioritySampling.SAMPLER_KEEP | SamplingMechanism.DEFAULT | PrioritySampling.SAMPLER_KEEP", |
| 93 | + "unset | PrioritySampling.UNSET | SamplingMechanism.DEFAULT | PrioritySampling.SAMPLER_KEEP", |
| 94 | + "user keep | PrioritySampling.USER_KEEP | SamplingMechanism.MANUAL | PrioritySampling.USER_KEEP ", |
| 95 | + "user drop | PrioritySampling.USER_DROP | SamplingMechanism.MANUAL | PrioritySampling.USER_DROP " |
| 96 | + }) |
105 | 97 | void testInjectExtract( |
106 | | - String scenario, int contextPriority, int samplingMechanism, int propagatedPriority) |
| 98 | + String scenario, |
| 99 | + @ConvertWith(PrioritySamplingConverter.class) int contextPriority, |
| 100 | + @ConvertWith(SamplingMechanismConverter.class) int samplingMechanism, |
| 101 | + @ConvertWith(PrioritySamplingConverter.class) int propagatedPriority) |
107 | 102 | throws Exception { |
108 | 103 | io.opentracing.Span span = tracer.buildSpan("some name").start(); |
109 | 104 | io.opentracing.SpanContext context = span.context(); |
|
0 commit comments