fix: fix broken tests due to the merge

Author: yifanzyifanz

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

@@ -375,7 +375,7 @@ public void testExecuteStreamingPartitionedUpdateRSTstream() {
     long count = tx.executeStreamingPartitionedUpdate(Statement.of(sql), Duration.ofMinutes(10));
 
     assertThat(count).isEqualTo(1000L);
-    verify(rpc).beginTransaction(any(BeginTransactionRequest.class), anyMap());
+    verify(rpc).beginTransaction(any(BeginTransactionRequest.class), anyMap(), eq(true));
     verify(rpc)
         .executeStreamingPartitionedDml(
             Mockito.eq(executeRequestWithoutResumeToken), anyMap(), any(Duration.class));

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

@@ -788,7 +788,7 @@ public void testSessionNotFoundReadWriteTransaction() {
           .thenReturn(ApiFutures.immediateFuture(Empty.getDefaultInstance()));
       when(closedSession.newTransaction(Options.fromTransactionOptions()))
           .thenReturn(closedTransactionContext);
-      when(closedSession.beginTransactionAsync(any(), true)).thenThrow(sessionNotFound);
+      when(closedSession.beginTransactionAsync(any(), eq(true))).thenThrow(sessionNotFound);
       TransactionRunnerImpl closedTransactionRunner = new TransactionRunnerImpl(closedSession);
       closedTransactionRunner.setSpan(mock(Span.class));
       when(closedSession.readWriteTransaction()).thenReturn(closedTransactionRunner);
@@ -801,7 +801,7 @@ public void testSessionNotFoundReadWriteTransaction() {
       final TransactionContextImpl openTransactionContext = mock(TransactionContextImpl.class);
       when(openSession.newTransaction(Options.fromTransactionOptions()))
           .thenReturn(openTransactionContext);
-      when(openSession.beginTransactionAsync(any(), true))
+      when(openSession.beginTransactionAsync(any(), eq(true)))
           .thenReturn(ApiFutures.immediateFuture(ByteString.copyFromUtf8("open-txn")));
       TransactionRunnerImpl openTransactionRunner = new TransactionRunnerImpl(openSession);
       openTransactionRunner.setSpan(mock(Span.class));