Skip to content

Commit 9d86d41

Browse files
committed
test: check for credentials for traces
Add an explicit check to the test whether there are credentials available.
1 parent 533ff9b commit 9d86d41

File tree

1 file changed

+14
-7
lines changed

1 file changed

+14
-7
lines changed

src/test/java/com/google/cloud/spanner/jdbc/it/ITOpenTelemetryTest.java

Lines changed: 14 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,8 @@
2323

2424
import com.google.api.gax.core.FixedCredentialsProvider;
2525
import com.google.api.gax.rpc.ResourceExhaustedException;
26+
import com.google.auth.Credentials;
27+
import com.google.auth.oauth2.GoogleCredentials;
2628
import com.google.cloud.opentelemetry.trace.TraceConfiguration;
2729
import com.google.cloud.opentelemetry.trace.TraceExporter;
2830
import com.google.cloud.spanner.Database;
@@ -282,14 +284,19 @@ private void assertUpdateResult(long updateCount, String sql)
282284
}
283285

284286
private void assertTrace(String sql) throws IOException, InterruptedException {
287+
Credentials credentials;
288+
if (env.getTestHelper().getOptions().getCredentials() != null) {
289+
credentials = env.getTestHelper().getOptions().getCredentials();
290+
} else {
291+
credentials = GoogleCredentials.getApplicationDefault();
292+
}
293+
if (credentials == null) {
294+
throw new IllegalStateException("No credentials found");
295+
}
285296
TraceServiceSettings settings =
286-
env.getTestHelper().getOptions().getCredentials() == null
287-
? TraceServiceSettings.newBuilder().build()
288-
: TraceServiceSettings.newBuilder()
289-
.setCredentialsProvider(
290-
FixedCredentialsProvider.create(
291-
env.getTestHelper().getOptions().getCredentials()))
292-
.build();
297+
TraceServiceSettings.newBuilder()
298+
.setCredentialsProvider(FixedCredentialsProvider.create(credentials))
299+
.build();
293300
try (TraceServiceClient client = TraceServiceClient.create(settings)) {
294301
// It can take a few seconds before the trace is visible.
295302
Thread.sleep(5000L);

0 commit comments

Comments
 (0)