chore(spanner): fix OpenTelemetrySpanTest

Author: harshachinta

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

@@ -150,7 +150,7 @@ public class OpenTelemetrySpanTest {
           "Commit Done",
           "Transaction Attempt Succeeded");
 
-  private int expectedReadWriteTransactionCount = 7;
+  private int expectedReadWriteTransactionEventsCount = 7;
   private List<String> expectedReadWriteTransactionErrorWithBeginTransactionEvents =
       ImmutableList.of(
           "Acquiring session",
@@ -244,6 +244,7 @@ public void setUp() throws Exception {
                 SessionPoolOptions.newBuilder()
                     .setMinSessions(2)
                     .setWaitForMinSessionsDuration(Duration.ofSeconds(10))
+                    .setSkipVerifyingBeginTransactionForMuxRW(true)
                     .build());
 
     spanner = builder.build().getService();
@@ -458,6 +459,16 @@ public void transactionRunner() {
                 "CloudSpannerOperation.Commit",
                 "CloudSpannerOperation.BatchCreateSessions",
                 "CloudSpanner.ReadWriteTransaction");
+
+    if (isMultiplexedSessionsEnabledForRW()) {
+      expectedReadWriteTransactionEvents =
+          ImmutableList.of(
+              "Starting Transaction Attempt",
+              "Starting Commit",
+              "Commit Done",
+              "Transaction Attempt Succeeded");
+      expectedReadWriteTransactionEventsCount = 4;
+    }
     DatabaseClient client = getClient();
     TransactionRunner runner = client.readWriteTransaction();
     runner.run(transaction -> transaction.executeUpdate(UPDATE_STATEMENT));
@@ -502,7 +513,7 @@ public void transactionRunner() {
                   verifyRequestEvents(
                       spanItem,
                       expectedReadWriteTransactionEvents,
-                      expectedReadWriteTransactionCount);
+                      expectedReadWriteTransactionEventsCount);
                   verifyCommonAttributes(spanItem);
                   break;
                 default:
@@ -528,6 +539,14 @@ public void transactionRunnerWithError() {
                 "CloudSpannerOperation.BatchCreateSessions",
                 "CloudSpannerOperation.ExecuteUpdate",
                 "CloudSpanner.ReadWriteTransaction");
+    if (isMultiplexedSessionsEnabledForRW()) {
+      expectedReadWriteTransactionErrorEvents =
+          ImmutableList.of(
+              "Starting Transaction Attempt",
+              "Transaction Attempt Failed in user operation",
+              "exception");
+      expectedReadWriteTransactionErrorEventsCount = 3;
+    }
     DatabaseClient client = getClient();
     TransactionRunner runner = client.readWriteTransaction();
     SpannerException e =
@@ -589,6 +608,18 @@ public void transactionRunnerWithFailedAndBeginTransaction() {
             "CloudSpannerOperation.Commit",
             "CloudSpannerOperation.BatchCreateSessions",
             "CloudSpanner.ReadWriteTransaction");
+    if (isMultiplexedSessionsEnabledForRW()) {
+      expectedReadWriteTransactionErrorWithBeginTransactionEvents =
+          ImmutableList.of(
+              "Starting Transaction Attempt",
+              "Transaction Attempt Aborted in user operation. Retrying",
+              "Creating Transaction",
+              "Transaction Creation Done",
+              "Starting Commit",
+              "Commit Done",
+              "Transaction Attempt Succeeded");
+      expectedReadWriteTransactionErrorWithBeginTransactionEventsCount = 8;
+    }
     DatabaseClient client = getClient();
     assertEquals(
         Long.valueOf(1L),
@@ -892,4 +923,11 @@ private boolean isMultiplexedSessionsEnabled() {
     }
     return spanner.getOptions().getSessionPoolOptions().getUseMultiplexedSession();
   }
+
+  private boolean isMultiplexedSessionsEnabledForRW() {
+    if (spanner.getOptions() == null || spanner.getOptions().getSessionPoolOptions() == null) {
+      return false;
+    }
+    return spanner.getOptions().getSessionPoolOptions().getUseMultiplexedSessionForRW();
+  }
 }