diff --git a/java-bigtable/google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/BigtableDataClientTests.java b/java-bigtable/google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/BigtableDataClientTests.java index ec82f1c12c8a..c8a999a625ab 100644 --- a/java-bigtable/google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/BigtableDataClientTests.java +++ b/java-bigtable/google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/BigtableDataClientTests.java @@ -58,7 +58,6 @@ import org.junit.Test; import org.junit.runner.RunWith; import org.junit.runners.JUnit4; -import org.mockito.Answers; import org.mockito.ArgumentMatchers; import org.mockito.Mock; import org.mockito.Mockito; @@ -79,8 +78,7 @@ public class BigtableDataClientTests { @Mock private EnhancedBigtableStub mockStub; - @Mock(answer = Answers.RETURNS_DEEP_STUBS) - private ServerStreamingCallable mockReadRowsCallable; + @Mock private ServerStreamingCallable mockReadRowsCallable; @Mock private UnaryCallable mockReadRowCallable; @Mock private UnaryCallable> mockSampleRowKeysCallable; @@ -96,11 +94,11 @@ public class BigtableDataClientTests { @Mock private Batcher mockBulkReadRowsBatcher; @Mock private UnaryCallable mockPrepareQueryCallable; - @Mock(answer = Answers.RETURNS_DEEP_STUBS) + @Mock private ServerStreamingCallable mockGenerateInitialChangeStreamPartitionsCallable; - @Mock(answer = Answers.RETURNS_DEEP_STUBS) + @Mock private ServerStreamingCallable mockReadChangeStreamCallable;