Skip to content

Commit 602195f

Browse files
committed
fix it test
1 parent 450caba commit 602195f

File tree

1 file changed

+9
-26
lines changed

1 file changed

+9
-26
lines changed

google-cloud-bigquery/src/test/java/com/google/cloud/bigquery/it/ITBigQueryTest.java

Lines changed: 9 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -1030,9 +1030,9 @@ public class ITBigQueryTest {
10301030

10311031
private static BigQuery bigquery;
10321032
private static Storage storage;
1033+
private static OpenTelemetry otel;
10331034

1034-
private static class CustomSpanExporter
1035-
implements io.opentelemetry.sdk.trace.export.SpanExporter {
1035+
private static class TestSpanExporter implements io.opentelemetry.sdk.trace.export.SpanExporter {
10361036
@Override
10371037
public CompletableResultCode export(Collection<SpanData> collection) {
10381038
if (collection.isEmpty()) {
@@ -1064,6 +1064,13 @@ public static void beforeClass() throws InterruptedException, IOException {
10641064
RemoteBigQueryHelper bigqueryHelper = RemoteBigQueryHelper.create();
10651065
RemoteStorageHelper storageHelper = RemoteStorageHelper.create();
10661066
Map<String, String> labels = ImmutableMap.of("test-job-name", "test-load-job");
1067+
SdkTracerProvider tracerProvider =
1068+
SdkTracerProvider.builder()
1069+
.addSpanProcessor(SimpleSpanProcessor.create(new TestSpanExporter()))
1070+
.setSampler(Sampler.alwaysOn())
1071+
.build();
1072+
otel = OpenTelemetrySdk.builder().setTracerProvider(tracerProvider).buildAndRegisterGlobal();
1073+
10671074
bigquery = bigqueryHelper.getOptions().getService();
10681075
storage = storageHelper.getOptions().getService();
10691076
storage.create(BucketInfo.of(BUCKET));
@@ -7537,14 +7544,6 @@ public void testStatementType() throws InterruptedException {
75377544

75387545
@Test
75397546
public void testOpenTelemetryTracingDatasets() {
7540-
SdkTracerProvider tracerProvider =
7541-
SdkTracerProvider.builder()
7542-
.addSpanProcessor(SimpleSpanProcessor.create(new CustomSpanExporter()))
7543-
.setSampler(Sampler.alwaysOn())
7544-
.build();
7545-
7546-
OpenTelemetry otel =
7547-
OpenTelemetrySdk.builder().setTracerProvider(tracerProvider).buildAndRegisterGlobal();
75487547
Tracer tracer = otel.getTracer("Test Tracer");
75497548
BigQueryOptions otelOptions =
75507549
BigQueryOptions.newBuilder()
@@ -7618,14 +7617,6 @@ public void testOpenTelemetryTracingDatasets() {
76187617

76197618
@Test
76207619
public void TestOpenTelemetryTracingTables() {
7621-
SdkTracerProvider tracerProvider =
7622-
SdkTracerProvider.builder()
7623-
.addSpanProcessor(SimpleSpanProcessor.create(new CustomSpanExporter()))
7624-
.setSampler(Sampler.alwaysOn())
7625-
.build();
7626-
7627-
OpenTelemetry otel =
7628-
OpenTelemetrySdk.builder().setTracerProvider(tracerProvider).buildAndRegisterGlobal();
76297620
Tracer tracer = otel.getTracer("Test Tracer");
76307621
BigQueryOptions otelOptions =
76317622
BigQueryOptions.newBuilder()
@@ -7670,14 +7661,6 @@ public void TestOpenTelemetryTracingTables() {
76707661

76717662
@Test
76727663
public void testOpenTelemetryTracingQuery() throws InterruptedException {
7673-
SdkTracerProvider tracerProvider =
7674-
SdkTracerProvider.builder()
7675-
.addSpanProcessor(SimpleSpanProcessor.create(new CustomSpanExporter()))
7676-
.setSampler(Sampler.alwaysOn())
7677-
.build();
7678-
7679-
OpenTelemetry otel =
7680-
OpenTelemetrySdk.builder().setTracerProvider(tracerProvider).buildAndRegisterGlobal();
76817664
Tracer tracer = otel.getTracer("Test Tracer");
76827665
BigQueryOptions otelOptions =
76837666
BigQueryOptions.newBuilder()

0 commit comments

Comments
 (0)