Skip to content

Commit dd8189b

Browse files
committed
test(bqstorage): Add test for default keep alive settings in the Read Client (#12042)
1 parent 8ec1ba9 commit dd8189b

File tree

1 file changed

+17
-0
lines changed

1 file changed

+17
-0
lines changed

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

Lines changed: 17 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,20 @@ void createReadSessionTest() {
139142
GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
140143
}
141144

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

0 commit comments

Comments
 (0)