Fix write unit test

pacoavila808 · pacoavila808 · commit d5ba7c80f973 · 2025-11-25T23:37:31.000Z
diff --git a/sdks/java/io/google-cloud-platform/src/test/java/org/apache/beam/sdk/io/gcp/firestore/BaseFirestoreV1WriteFnTest.java b/sdks/java/io/google-cloud-platform/src/test/java/org/apache/beam/sdk/io/gcp/firestore/BaseFirestoreV1WriteFnTest.java
@@ -111,7 +111,7 @@ public final void setUp() {
     when(rpcQos.newWriteAttempt(any())).thenReturn(attempt, attempt2);
 
     when(ff.getRpcQos(any())).thenReturn(rpcQos);
-    when(ff.getFirestoreStub(pipelineOptions)).thenReturn(stub);
+    when(ff.getFirestoreStub(any(), any(), any())).thenReturn(stub);
     when(stub.batchWriteCallable()).thenReturn(callable);
     metricsFixture = new MetricsFixture();
   }

Original file line number	Diff line number	Diff line change
`@@ -111,7 +111,7 @@ public final void setUp() {`
`111`	`111`	`when(rpcQos.newWriteAttempt(any())).thenReturn(attempt, attempt2);`
`112`	`112`
`113`	`113`	`when(ff.getRpcQos(any())).thenReturn(rpcQos);`
`114`		`- when(ff.getFirestoreStub(pipelineOptions)).thenReturn(stub);`
	`114`	`+ when(ff.getFirestoreStub(any(), any(), any())).thenReturn(stub);`
`115`	`115`	`when(stub.batchWriteCallable()).thenReturn(callable);`
`116`	`116`	`metricsFixture = new MetricsFixture();`
`117`	`117`	`}`