chore(spanner): fix tests

Author: harshachinta

google-cloud-spanner/src/test/java/com/google/cloud/spanner/SessionPoolTest.java

@@ -1495,9 +1495,10 @@ public void testSessionNotFoundReadWriteTransaction() {
               .build();
       when(closedSession.asyncClose())
           .thenReturn(ApiFutures.immediateFuture(Empty.getDefaultInstance()));
-      when(closedSession.newTransaction(Options.fromTransactionOptions()))
+      when(closedSession.newTransaction(eq(Options.fromTransactionOptions()), any()))
           .thenReturn(closedTransactionContext);
-      when(closedSession.beginTransactionAsync(any(), eq(true), any())).thenThrow(sessionNotFound);
+      when(closedSession.beginTransactionAsync(any(), eq(true), any(), any()))
+          .thenThrow(sessionNotFound);
       when(closedSession.getTracer()).thenReturn(tracer);
       TransactionRunnerImpl closedTransactionRunner = new TransactionRunnerImpl(closedSession);
       closedTransactionRunner.setSpan(span);
@@ -1510,9 +1511,9 @@ public void testSessionNotFoundReadWriteTransaction() {
       when(openSession.getName())
           .thenReturn("projects/dummy/instances/dummy/database/dummy/sessions/session-open");
       final TransactionContextImpl openTransactionContext = mock(TransactionContextImpl.class);
-      when(openSession.newTransaction(Options.fromTransactionOptions()))
+      when(openSession.newTransaction(eq(Options.fromTransactionOptions()), any()))
           .thenReturn(openTransactionContext);
-      when(openSession.beginTransactionAsync(any(), eq(true), any()))
+      when(openSession.beginTransactionAsync(any(), eq(true), any(), any()))
           .thenReturn(ApiFutures.immediateFuture(ByteString.copyFromUtf8("open-txn")));
       when(openSession.getTracer()).thenReturn(tracer);
       TransactionRunnerImpl openTransactionRunner = new TransactionRunnerImpl(openSession);

google-cloud-spanner/src/test/java/com/google/cloud/spanner/TransactionRunnerImplTest.java

@@ -20,6 +20,7 @@
 import static org.junit.Assert.assertArrayEquals;
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertThrows;
+import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.ArgumentMatchers.eq;
 import static org.mockito.Mockito.doThrow;
 import static org.mockito.Mockito.eq;
@@ -117,7 +118,7 @@ public void setUp() {
     tracer = new TraceWrapper(Tracing.getTracer(), OpenTelemetry.noop().getTracer(""), false);
     firstRun = true;
     when(session.getErrorHandler()).thenReturn(DefaultErrorHandler.INSTANCE);
-    when(session.newTransaction(Options.fromTransactionOptions())).thenReturn(txn);
+    when(session.newTransaction(eq(Options.fromTransactionOptions()), any())).thenReturn(txn);
     when(session.getTracer()).thenReturn(tracer);
     when(rpc.executeQuery(Mockito.any(ExecuteSqlRequest.class), Mockito.anyMap(), eq(true)))
         .thenAnswer(
@@ -343,7 +344,8 @@ private long[] batchDmlException(int status) {
             .setTracer(session.getTracer())
             .setSpan(session.getTracer().getCurrentSpan())
             .build();
-    when(session.newTransaction(Options.fromTransactionOptions())).thenReturn(transaction);
+    when(session.newTransaction(eq(Options.fromTransactionOptions()), any()))
+        .thenReturn(transaction);
     when(session.getName()).thenReturn(SessionId.of("p", "i", "d", "test").getName());
     TransactionRunnerImpl runner = new TransactionRunnerImpl(session);
     runner.setSpan(span);