1818import static com .google .common .truth .Truth .assertThat ;
1919
2020import com .google .api .gax .batching .Batcher ;
21+ import com .google .api .gax .rpc .FixedHeaderProvider ;
22+ import com .google .api .gax .rpc .HeaderProvider ;
2123import com .google .bigtable .v2 .BigtableGrpc ;
2224import com .google .bigtable .v2 .CheckAndMutateRowRequest ;
2325import com .google .bigtable .v2 .CheckAndMutateRowResponse ;
@@ -61,11 +63,14 @@ public class HeadersTest {
6163 private static final String TABLE_NAME =
6264 NameUtil .formatTableName (PROJECT_ID , INSTANCE_ID , TABLE_ID );
6365 private static final String APP_PROFILE_ID = "fake-profile" ;
66+ private static final String TEST_FIXED_HEADER_STRING = "test_fixed_header" ;
6467
6568 private static final Metadata .Key <String > X_GOOG_REQUEST_PARAMS_KEY =
6669 Metadata .Key .of ("x-goog-request-params" , Metadata .ASCII_STRING_MARSHALLER );
6770 private static final Metadata .Key <String > API_CLIENT_HEADER_KEY =
6871 Metadata .Key .of ("x-goog-api-client" , Metadata .ASCII_STRING_MARSHALLER );
72+ private static final Metadata .Key <String > TEST_FIXED_HEADER =
73+ Metadata .Key .of (TEST_FIXED_HEADER_STRING , Metadata .ASCII_STRING_MARSHALLER );
6974
7075 private Server server ;
7176 private BlockingQueue <Metadata > sentMetadata = new ArrayBlockingQueue <>(10 );
@@ -91,9 +96,13 @@ public void setUp() throws Exception {
9196 .setInstanceId (INSTANCE_ID )
9297 .setAppProfileId (APP_PROFILE_ID );
9398
99+ HeaderProvider headerProvider =
100+ FixedHeaderProvider .create (TEST_FIXED_HEADER_STRING , "test_header_value" );
101+
94102 // Force immediate flush
95103 settings
96104 .stubSettings ()
105+ .setHeaderProvider (headerProvider )
97106 .bulkMutateRowsSettings ()
98107 .setBatchingSettings (
99108 settings
@@ -172,6 +181,9 @@ private void verifyHeaderSent() {
172181 assertThat (apiClientValue ).containsMatch ("gl-java/[.\\ d_]+" );
173182 assertThat (apiClientValue ).containsMatch ("gax/[.\\ d_]+" );
174183 assertThat (apiClientValue ).containsMatch ("grpc/[.\\ d_]+" );
184+
185+ String fixedHeader = metadata .get (TEST_FIXED_HEADER );
186+ assertThat (fixedHeader ).isEqualTo ("test_header_value" );
175187 }
176188
177189 private class MetadataInterceptor implements ServerInterceptor {
0 commit comments