fix: shuffle servers when only two servers are selected

Author: baileympearson

src/sdam/topology.ts

@@ -904,9 +904,7 @@ function processWaitQueue(topology: Topology) {
     } else if (selectedDescriptions.length === 1) {
       selectedServer = topology.s.servers.get(selectedDescriptions[0].address);
     } else {
-      // don't shuffle the array if there are only two elements
-      const descriptions =
-        selectedDescriptions.length === 2 ? selectedDescriptions : shuffle(selectedDescriptions, 2);
+      const descriptions = shuffle(selectedDescriptions, 2);
       const server1 = topology.s.servers.get(descriptions[0].address);
       const server2 = topology.s.servers.get(descriptions[1].address);
 

test/integration/server-selection/server_selection.prose.operation_count.test.ts

@@ -166,15 +166,15 @@ describe('operationCount-based Selection Within Latency Window - Prose Test', fu
     expect(percentageToHost2).to.be.greaterThan(40).and.lessThan(60);
   });
 
-  it.only(
-    'equally distributes operations with both hosts are fine test 2',
+  it(
+    'equally distributes operations with both hosts when requests are in parallel',
     TEST_METADATA,
     async function () {
       const collection = client.db('test-db').collection('collection0');
 
       const { insertedId } = await collection.insertOne({ name: 'bumpy' });
 
-      const n = 2000;
+      const n = 1000;
 
       for (let i = 0; i < n; ++i) {
         await collection.findOne({ _id: insertedId });
@@ -184,7 +184,6 @@ describe('operationCount-based Selection Within Latency Window - Prose Test', fu
       const [host1, host2] = seeds.map(seed => seed.split(':')[1]);
       const percentageToHost1 = (counts[host1] / n) * 100;
       const percentageToHost2 = (counts[host2] / n) * 100;
-      console.error({ percentageToHost1, percentageToHost2 });
       expect(percentageToHost1).to.be.greaterThan(40).and.lessThan(60);
       expect(percentageToHost2).to.be.greaterThan(40).and.lessThan(60);
     }