Skip to content

Commit 7956c2f

Browse files
committed
test(bqstorage): Add test for default keep-alive settings in read client
1 parent 7e9bc92 commit 7956c2f

1 file changed

Lines changed: 14 additions & 0 deletions

File tree

java-bigquerystorage/google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1/BigQueryReadClientTest.java

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,12 +16,14 @@
1616
package com.google.cloud.bigquery.storage.v1;
1717

1818
import static org.junit.jupiter.api.Assertions.assertEquals;
19+
import static org.junit.jupiter.api.Assertions.assertInstanceOf;
1920
import static org.junit.jupiter.api.Assertions.assertThrows;
2021
import static org.junit.jupiter.api.Assertions.assertTrue;
2122

2223
import com.google.api.gax.core.NoCredentialsProvider;
2324
import com.google.api.gax.grpc.GaxGrpcProperties;
2425
import com.google.api.gax.grpc.GrpcStatusCode;
26+
import com.google.api.gax.grpc.InstantiatingGrpcChannelProvider;
2527
import com.google.api.gax.grpc.testing.LocalChannelProvider;
2628
import com.google.api.gax.grpc.testing.MockGrpcService;
2729
import com.google.api.gax.grpc.testing.MockServiceHelper;
@@ -33,6 +35,7 @@
3335
import com.google.api.gax.rpc.ResourceExhaustedException;
3436
import com.google.api.gax.rpc.ServerStreamingCallable;
3537
import com.google.api.gax.rpc.StatusCode;
38+
import com.google.api.gax.rpc.TransportChannelProvider;
3639
import com.google.protobuf.AbstractMessage;
3740
import com.google.protobuf.Duration;
3841
import com.google.protobuf.Parser;
@@ -139,6 +142,17 @@ void createReadSessionTest() {
139142
GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
140143
}
141144

145+
@Test
146+
void readSettingsGrpcChannelDefault() throws IOException {
147+
BigQueryReadSettings bigQueryReadSettings = BigQueryReadSettings.newBuilder().build();
148+
TransportChannelProvider transportChannelProvider = bigQueryReadSettings.getTransportChannelProvider();
149+
assertInstanceOf(InstantiatingGrpcChannelProvider.class, transportChannelProvider);
150+
InstantiatingGrpcChannelProvider grpcChannelProvider = (InstantiatingGrpcChannelProvider) transportChannelProvider;
151+
assertEquals(java.time.Duration.ofMinutes(1), grpcChannelProvider.getKeepAliveTimeDuration());
152+
assertEquals(java.time.Duration.ofMinutes(1), grpcChannelProvider.getKeepAliveTimeoutDuration());
153+
assertTrue(grpcChannelProvider.getKeepAliveWithoutCalls());
154+
}
155+
142156
@Test
143157
@SuppressWarnings("all")
144158
void createReadSessionExceptionTest() throws Exception {

0 commit comments

Comments
 (0)