diff --git a/google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/internal/session/SessionImplTest.java b/google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/internal/session/SessionImplTest.java index 852d908700..3796b04721 100644 --- a/google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/internal/session/SessionImplTest.java +++ b/google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/internal/session/SessionImplTest.java @@ -440,7 +440,15 @@ void testHeartbeat() throws Exception { SessionParametersResponse.newBuilder() .setKeepAlive(Durations.fromMillis(keepAliveDurationMs)) .build()) - .putVrpcActions(0, ActionList.getDefaultInstance()) + .putVrpcActions( + 0, + ActionList.newBuilder() + .addActions( + Action.newBuilder() + .setDelay(Durations.fromMillis(500)) + .setResponse(VirtualRpcResponse.getDefaultInstance()) + .build()) + .build()) .build() .toByteString()) .build();