Skip to content

Commit 02f104f

Browse files
committed
revert unnecessary change to read test
1 parent 75678e1 commit 02f104f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

sdks/java/io/google-cloud-platform/src/test/java/org/apache/beam/sdk/io/gcp/firestore/BaseFirestoreV1ReadFnTest.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ abstract class BaseFirestoreV1ReadFnTest<InT, OutT>
4545
public final void attemptsExhaustedForRetryableError() throws Exception {
4646
BaseFirestoreV1ReadFn<InT, OutT> fn = getFn(clock, ff, rpcQosOptions);
4747
V1RpcFnTestCtx ctx = newCtx();
48-
when(ff.getFirestoreStub(any(), any(), any())).thenReturn(stub);
48+
when(ff.getFirestoreStub(any())).thenReturn(stub);
4949
when(ff.getRpcQos(any())).thenReturn(rpcQos);
5050
when(rpcQos.newReadAttempt(fn.getRpcAttemptContext())).thenReturn(attempt);
5151
ctx.mockRpcToCallable(stub);
@@ -79,7 +79,7 @@ public final void attemptsExhaustedForRetryableError() throws Exception {
7979
public final void noRequestIsSentIfNotSafeToProceed() throws Exception {
8080
BaseFirestoreV1ReadFn<InT, OutT> fn = getFn(clock, ff, rpcQosOptions);
8181
V1RpcFnTestCtx ctx = newCtx();
82-
when(ff.getFirestoreStub(any(), any(), any())).thenReturn(stub);
82+
when(ff.getFirestoreStub(any())).thenReturn(stub);
8383
when(ff.getRpcQos(any())).thenReturn(rpcQos);
8484
when(rpcQos.newReadAttempt(fn.getRpcAttemptContext())).thenReturn(attempt);
8585

0 commit comments

Comments
 (0)