---

rahulKQL · sduskis · commit 09513cf5914e · 2019-02-14T18:05:27.000-05:00
yaml --- r: 15291 b: refs/heads/autosynth-bigtable c: 228192e h: refs/heads/master i: 15289: f97d222 15287: 71d4fd5
diff --git a/[refs] b/[refs]
@@ -126,7 +126,7 @@ refs/heads/autosynth-asset: cd8251de8c40e239ad24dcf9ed93ea2708a3eed5
 refs/heads/autosynth-automl: cced2f56bbef0499609073edbca6253e1df5e535
 refs/heads/autosynth-bigquerydatatransfer: a6667617707608b1dbfb02d59c22b5152f208da7
 refs/heads/autosynth-bigquerystorage: 14ab055598b943ae3f33f484e9fb1653355d08e7
-refs/heads/autosynth-bigtable: e195b5f60b793a1d85f3cdc679cfb2fc58128887
+refs/heads/autosynth-bigtable: 228192ee97d6968acbc8fe9becfed23159f80720
 refs/heads/autosynth-bigtable-admin: 6379a2bc712f2736c83de0e009b4d26da4fa82ca
 refs/heads/autosynth-containeranalysis: 781fdb430a60f9a6491f116e31e4e10118157bdb
 refs/heads/autosynth-datastore: af1fb76aa3eee02fe6f31f8fa1c72a4f048d149b
diff --git a/branches/autosynth-bigtable/google-cloud-clients/google-cloud-bigtable/src/main/java/com/google/cloud/bigtable/data/v2/stub/EnhancedBigtableStub.java b/branches/autosynth-bigtable/google-cloud-clients/google-cloud-bigtable/src/main/java/com/google/cloud/bigtable/data/v2/stub/EnhancedBigtableStub.java
@@ -90,6 +90,7 @@ public static EnhancedBigtableStub create(EnhancedBigtableStubSettings settings)
             .setTransportChannelProvider(settings.getTransportChannelProvider())
             .setEndpoint(settings.getEndpoint())
             .setCredentialsProvider(settings.getCredentialsProvider())
+            .setHeaderProvider(settings.getHeaderProvider())
             .setStreamWatchdogProvider(settings.getStreamWatchdogProvider())
             .setStreamWatchdogCheckInterval(settings.getStreamWatchdogCheckInterval());
 
diff --git a/branches/autosynth-bigtable/google-cloud-clients/google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/stub/ResourceHeaderTest.java b/branches/autosynth-bigtable/google-cloud-clients/google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/stub/ResourceHeaderTest.java
@@ -20,6 +20,8 @@
 import com.google.api.gax.core.NoCredentialsProvider;
 import com.google.api.gax.grpc.testing.InProcessServer;
 import com.google.api.gax.grpc.testing.LocalChannelProvider;
+import com.google.api.gax.rpc.FixedHeaderProvider;
+import com.google.api.gax.rpc.HeaderProvider;
 import com.google.bigtable.v2.BigtableGrpc;
 import com.google.cloud.bigtable.data.v2.BigtableDataClient;
 import com.google.cloud.bigtable.data.v2.BigtableDataSettings;
@@ -48,6 +50,9 @@ public class ResourceHeaderTest {
   private static final Pattern EXPECTED_HEADER_PATTERN =
       Pattern.compile(".*" + NameUtil.formatTableName(PROJECT_ID, INSTANCE_ID, TABLE_ID) + ".*");
   private static final String HEADER_NAME = "x-goog-request-params";
+  private static final String TEST_HEADER_NAME = "simple-header-name";
+  private static final String TEST_HEADER_VALUE = "simple-header-value";
+  private static final Pattern TEST_PATTERN = Pattern.compile(".*" + TEST_HEADER_VALUE + ".*");
 
   private InProcessServer<?> server;
   private LocalChannelProvider channelProvider;
@@ -59,11 +64,13 @@ public void setUp() throws Exception {
     server.start();
     channelProvider = LocalChannelProvider.create(NAME);
 
+    HeaderProvider headerProvider = FixedHeaderProvider.create(TEST_HEADER_NAME, TEST_HEADER_VALUE);
     BigtableDataSettings.Builder settings =
         BigtableDataSettings.newBuilder()
             .setProjectId(PROJECT_ID)
             .setInstanceId(INSTANCE_ID)
             .setTransportChannelProvider(channelProvider)
+            .setHeaderProvider(headerProvider)
             .setCredentialsProvider(NoCredentialsProvider.create());
 
     // Force immediate flush
@@ -132,5 +139,7 @@ public void readModifyWriteTest() {
   private void verifyHeaderSent() {
     boolean headerSent = channelProvider.isHeaderSent(HEADER_NAME, EXPECTED_HEADER_PATTERN);
     assertWithMessage("Header was sent").that(headerSent).isTrue();
+    boolean testHeader = channelProvider.isHeaderSent(TEST_HEADER_NAME, TEST_PATTERN);
+    assertWithMessage("HeaderProvider's header received in Channel").that(testHeader).isTrue();
   }
 }
