chore(spanner): fix test

harshachinta · harshachinta · commit 8db9d831c7f9 · 2025-02-20T17:30:42.000Z
diff --git a/google-cloud-spanner/src/test/java/com/google/cloud/spanner/connection/PartitionedQueryMockServerTest.java b/google-cloud-spanner/src/test/java/com/google/cloud/spanner/connection/PartitionedQueryMockServerTest.java
@@ -695,6 +695,8 @@ public void testRunPartitionedQueryWithMaxParallelism() {
       }
       if (isMultiplexedSessionsEnabledForPartitionedOps(connection.getSpanner())) {
         assertEquals(2, mockSpanner.countRequestsOfType(CreateSessionRequest.class));
+      } else if (isMultiplexedSessionsEnabled(connection.getSpanner())) {
+        assertEquals(6, mockSpanner.countRequestsOfType(CreateSessionRequest.class));
       } else {
         assertEquals(5, mockSpanner.countRequestsOfType(CreateSessionRequest.class));
       }

Original file line number	Diff line number	Diff line change
`@@ -695,6 +695,8 @@ public void testRunPartitionedQueryWithMaxParallelism() {`
`695`	`695`	`}`
`696`	`696`	`if (isMultiplexedSessionsEnabledForPartitionedOps(connection.getSpanner())) {`
`697`	`697`	`assertEquals(2, mockSpanner.countRequestsOfType(CreateSessionRequest.class));`
	`698`	`+ } else if (isMultiplexedSessionsEnabled(connection.getSpanner())) {`
	`699`	`+ assertEquals(6, mockSpanner.countRequestsOfType(CreateSessionRequest.class));`
`698`	`700`	`} else {`
`699`	`701`	`assertEquals(5, mockSpanner.countRequestsOfType(CreateSessionRequest.class));`
`700`	`702`	`}`