Skip to content

Commit 611aa90

Browse files
committed
Moved API to internal directory
1 parent 798ba56 commit 611aa90

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

instrumentation-api/src/test/java/io/opentelemetry/instrumentation/api/internal/InstrumentationCustomizerTest.java

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,9 @@
77

88
import static io.opentelemetry.sdk.testing.assertj.OpenTelemetryAssertions.assertThat;
99

10+
import io.opentelemetry.instrumentation.api.instrumenter.AttributesExtractor;
11+
import io.opentelemetry.instrumentation.api.instrumenter.ContextCustomizer;
12+
import io.opentelemetry.instrumentation.api.instrumenter.OperationMetrics;
1013
import java.util.function.Predicate;
1114
import org.junit.jupiter.api.BeforeEach;
1215
import org.junit.jupiter.api.Test;
@@ -19,6 +22,7 @@ class InstrumentationCustomizerTest {
1922

2023
@Mock private AttributesExtractor<Object, Object> attributesExtractor;
2124
@Mock private OperationMetrics operationMetrics;
25+
@Mock private ContextCustomizer<Object> contextCustomizer;
2226

2327
private InstrumentationCustomizer customizer;
2428

@@ -27,6 +31,11 @@ class InstrumentationCustomizerTest {
2731
void setUp() {
2832
customizer =
2933
new InstrumentationCustomizer() {
34+
@Override
35+
public <REQUEST> ContextCustomizer<REQUEST> getContextCustomizer() {
36+
return (ContextCustomizer<REQUEST>) contextCustomizer;
37+
}
38+
3039
@Override
3140
public Predicate<String> instrumentationNamePredicate() {
3241
return name -> name.startsWith("test.instrumentation");
@@ -45,6 +54,11 @@ public OperationMetrics getOperationMetrics() {
4554
};
4655
}
4756

57+
@Test
58+
void testGetContextCustomizer() {
59+
assertThat(customizer.getContextCustomizer()).isSameAs(contextCustomizer);
60+
}
61+
4862
@Test
4963
void testInstrumentationNamePredicate() {
5064
assertThat(customizer.instrumentationNamePredicate().test("test.instrumentation.example"))

0 commit comments

Comments
 (0)