From ec73a1500d434b33bed3fba8ebdfe66a8f16bf6f Mon Sep 17 00:00:00 2001
From: Pratick Chokhani <pratick@google.com>
Date: Thu, 12 Dec 2024 16:14:11 +0530
Subject: [PATCH] test(spanner): Enabled multiplexed session for partitioned
 operations in systest.

---
 .../com/google/cloud/executor/spanner/CloudClientExecutor.java  | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/google-cloud-spanner-executor/src/main/java/com/google/cloud/executor/spanner/CloudClientExecutor.java b/google-cloud-spanner-executor/src/main/java/com/google/cloud/executor/spanner/CloudClientExecutor.java
index d3f5712646a..d0490be44a1 100644
--- a/google-cloud-spanner-executor/src/main/java/com/google/cloud/executor/spanner/CloudClientExecutor.java
+++ b/google-cloud-spanner-executor/src/main/java/com/google/cloud/executor/spanner/CloudClientExecutor.java
@@ -832,6 +832,8 @@ private synchronized Spanner getClient(long timeoutSeconds, boolean useMultiplex
     SessionPoolOptionsHelper.setUseMultiplexedSession(poolOptionsBuilder, useMultiplexedSession);
     SessionPoolOptionsHelper.setUseMultiplexedSessionForRW(
         poolOptionsBuilder, useMultiplexedSession);
+    SessionPoolOptionsHelper.setUseMultiplexedSessionForPartitionedOperations(
+        poolOptionsBuilder, useMultiplexedSession);
     LOGGER.log(
         Level.INFO,
         String.format(