Update Function Signature

data-integrations · Oct 25, 2023 · 9054991 · 9054991
1 parent 5f24893
commit 9054991
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/test/java/io/cdap/plugin/gcp/bigquery/sink/BigQueryOutputFormatTest.java b/src/test/java/io/cdap/plugin/gcp/bigquery/sink/BigQueryOutputFormatTest.java
@@ -93,7 +93,8 @@ private BigQueryOutputFormat.BigQueryOutputCommitter initMocks(String operation)
                                   ArgumentMatchers.any(),
                                   ArgumentMatchers.any(JobId.class),
                                   ArgumentMatchers.any(),
-                                  ArgumentMatchers.any(Dataset.class));
+                                  ArgumentMatchers.any(Dataset.class),
+                                  ArgumentMatchers.any());
 
     FieldSetter.setField(spy, bigQueryOutputCommitter.getClass().getDeclaredField("bigQueryHelper"),
                          bigQueryHelperMock);
@@ -157,7 +158,8 @@ public void commitJobTestUpdateBQInvocations() throws Exception {
               ArgumentMatchers.any(),
               ArgumentMatchers.any(JobId.class),
               ArgumentMatchers.any(),
-              ArgumentMatchers.any(Dataset.class));
+              ArgumentMatchers.any(Dataset.class),
+              ArgumentMatchers.any());
   }
 
 }