From 1f49e7f7cdfa60af6d5d3c32d49ecbf51a2d1daf Mon Sep 17 00:00:00 2001
From: Pratick Chokhani <pratick@google.com>
Date: Tue, 10 Dec 2024 15:40:43 +0530
Subject: [PATCH] fix(spanner): Fixed unit test for multiplexed session.

---
 .../google/cloud/spanner/RetryOnInvalidatedSessionTest.java    | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/google-cloud-spanner/src/test/java/com/google/cloud/spanner/RetryOnInvalidatedSessionTest.java b/google-cloud-spanner/src/test/java/com/google/cloud/spanner/RetryOnInvalidatedSessionTest.java
index 3032a1cae40..a6c3f9fa7c5 100644
--- a/google-cloud-spanner/src/test/java/com/google/cloud/spanner/RetryOnInvalidatedSessionTest.java
+++ b/google-cloud-spanner/src/test/java/com/google/cloud/spanner/RetryOnInvalidatedSessionTest.java
@@ -1288,6 +1288,9 @@ public void write() throws InterruptedException {
 
   @Test
   public void writeAtLeastOnce() throws InterruptedException {
+    assumeFalse(
+        "Multiplexed session do not throw a SessionNotFound errors. ",
+        spanner.getOptions().getSessionPoolOptions().getUseMultiplexedSession());
     assertThrowsSessionNotFoundIfShouldFail(
         () ->
             client.writeAtLeastOnce(