Skip multiplex session tests in regular session

Author: sakthivelmanii

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

@@ -70,10 +70,7 @@ public class MultiplexedSessionDatabaseClientMockServerTest extends AbstractMock
 
   @BeforeClass
   public static void setupResults() {
-    assumeFalse(
-        System.getenv()
-            .getOrDefault("GOOGLE_CLOUD_SPANNER_MULTIPLEXED_SESSIONS", "")
-            .equalsIgnoreCase("false"));
+    assumeFalse(TestHelper.isMultiplexSessionDisabled());
     mockSpanner.putStatementResults(
         StatementResult.query(STATEMENT, new RandomResultSetGenerator(1).generate()));
     mockSpanner.putStatementResult(StatementResult.update(UPDATE_STATEMENT, UPDATE_COUNT));

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

@@ -24,6 +24,7 @@
 import static org.junit.Assert.assertThrows;
 import static org.junit.Assert.assertTrue;
 import static org.junit.Assert.fail;
+import static org.junit.Assume.assumeFalse;
 
 import com.google.api.gax.longrunning.OperationTimedPollAlgorithm;
 import com.google.api.gax.retrying.RetrySettings;
@@ -255,6 +256,7 @@ public void testMetricsWithGaxRetryUnaryRpc() {
 
   @Test
   public void testNoNetworkConnection() {
+    assumeFalse(TestHelper.isMultiplexSessionDisabled());
     // Create a Spanner instance that tries to connect to a server that does not exist.
     // This simulates a bad network connection.
     SpannerOptions.Builder builder = SpannerOptions.newBuilder();

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

@@ -283,6 +283,7 @@ public void testRandomizePositionQPSThreshold() {
 
   @Test
   public void testUseMultiplexedSession() {
+    assumeFalse(TestHelper.isMultiplexSessionDisabled());
     // skip these tests since this configuration can have dual behaviour in different test-runners
     assumeFalse(SessionPoolOptions.newBuilder().build().getUseMultiplexedSession());
     assertEquals(false, SessionPoolOptions.newBuilder().build().getUseMultiplexedSession());
@@ -304,6 +305,7 @@ public void testUseMultiplexedSession() {
   @Test
   public void testUseMultiplexedSessionForRW() {
     // skip these tests since this configuration can have dual behaviour in different test-runners
+    assumeFalse(TestHelper.isMultiplexSessionDisabled());
     assumeFalse(
         Boolean.parseBoolean(System.getenv("GOOGLE_CLOUD_SPANNER_MULTIPLEXED_SESSIONS_FOR_RW")));
     assumeFalse(SessionPoolOptions.newBuilder().build().getUseMultiplexedSession());

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

@@ -0,0 +1,10 @@
+package com.google.cloud.spanner;
+
+class TestHelper {
+
+  static boolean isMultiplexSessionDisabled() {
+    return System.getenv()
+        .getOrDefault("GOOGLE_CLOUD_SPANNER_MULTIPLEXED_SESSIONS", "")
+        .equalsIgnoreCase("false");
+  }
+}