---

olavloite · kolea2 · commit 6655c03ac4ef · 2019-03-28T13:52:39.000-04:00
yaml --- r: 24037 b: refs/heads/autosynth-kms c: 7a9f3fe h: refs/heads/master i: 24035: 630b6d6
diff --git a/[refs] b/[refs]
@@ -132,7 +132,7 @@ refs/heads/autosynth-dialogflow: cb1ceeb3ec1a132d3096fc88d44b930a52130e18
 refs/heads/autosynth-errorreporting: 3f176c20b55dfaaa8fc32f28d82b31784b93e636
 refs/heads/autosynth-firestore: d48d82caace227856b6cd85ac30ee474528733ea
 refs/heads/autosynth-iot: 4025d1804241e74d54950a324dc4f667aeaad4b3
-refs/heads/autosynth-kms: e706ba1d72589cb73ca3cdb6fcdf039b42698da6
+refs/heads/autosynth-kms: 7a9f3fe52a3838ad3b743f8c5664e2f27f8d7b5c
 refs/heads/autosynth-language: c3d990dd34d81e7e935041e7147fb9dd27f8a557
 refs/heads/autosynth-os-login: 092fdbed6d5317948f92b708e9f50dedd89fc666
 refs/heads/autosynth-redis: 9e1fe503973c7b4a9ba26afb1fcddc2a57ba795a
diff --git a/branches/autosynth-kms/google-cloud-clients/google-cloud-spanner/src/test/java/com/google/cloud/spanner/GceTestEnvConfig.java b/branches/autosynth-kms/google-cloud-clients/google-cloud-spanner/src/test/java/com/google/cloud/spanner/GceTestEnvConfig.java
@@ -18,6 +18,7 @@
 
 import static com.google.common.base.Preconditions.checkState;
 
+import com.google.auth.oauth2.GoogleCredentials;
 import com.google.cloud.spanner.spi.v1.SpannerInterceptorProvider;
 import io.grpc.CallOptions;
 import io.grpc.Channel;
@@ -28,13 +29,16 @@
 import io.grpc.Metadata;
 import io.grpc.MethodDescriptor;
 import io.grpc.Status;
+import java.io.FileInputStream;
+import java.io.IOException;
 import java.util.Random;
 import java.util.concurrent.atomic.AtomicBoolean;
 
 /** Configure TestEnv based on configuration provided. */
 public class GceTestEnvConfig implements TestEnvConfig {
   public static final String GCE_PROJECT_ID = "spanner.gce.config.project_id";
   public static final String GCE_SERVER_URL = "spanner.gce.config.server_url";
+  public static final String GCE_CREDENTIALS_FILE = "spanner.gce.config.credentials_file";
   public static final String GCE_STREAM_BROKEN_PROBABILITY =
       "spanner.gce.config.stream_broken_probability";
 
@@ -43,6 +47,7 @@ public class GceTestEnvConfig implements TestEnvConfig {
   public GceTestEnvConfig() {
     String projectId = System.getProperty(GCE_PROJECT_ID, "");
     String serverUrl = System.getProperty(GCE_SERVER_URL, "");
+    String credentialsFile = System.getProperty(GCE_CREDENTIALS_FILE, "");
     double errorProbability =
         Double.parseDouble(System.getProperty(GCE_STREAM_BROKEN_PROBABILITY, "0.0"));
     checkState(errorProbability <= 1.0);
@@ -53,6 +58,13 @@ public GceTestEnvConfig() {
     if (!serverUrl.isEmpty()) {
       builder.setHost(serverUrl);
     }
+    if (!credentialsFile.isEmpty()) {
+      try {
+        builder.setCredentials(GoogleCredentials.fromStream(new FileInputStream(credentialsFile)));
+      } catch (IOException e) {
+        throw new RuntimeException(e);
+      }
+    }
     options =
         builder
             .setInterceptorProvider(
