tests: test round robin cluster order for overflow

Author: dougwilson

test/unit/pool-cluster/test-connection-rr.js

@@ -1,33 +1,40 @@
+var after   = require('after');
 var assert  = require('assert');
 var common  = require('../../common');
 var cluster = common.createPoolCluster();
 var server  = common.createFakeServer();
 
-var connCount  = 0;
+var count      = 5;
+var order      = [];
 var poolConfig = common.getTestConfig({port: common.fakeServerPort});
 cluster.add('SLAVE1', poolConfig);
 cluster.add('SLAVE2', poolConfig);
 
+var done = after(count, function () {
+  assert.deepEqual(order, [
+    'SLAVE1',
+    'SLAVE2',
+    'SLAVE1',
+    'SLAVE2',
+    'SLAVE1'
+  ]);
+  cluster.end(function (err) {
+    assert.ifError(err);
+    server.destroy();
+  });
+});
+
 server.listen(common.fakeServerPort, function(err) {
   assert.ifError(err);
 
   var pool = cluster.of('SLAVE*', 'RR');
 
-  pool.getConnection(function (err, conn1) {
-    assert.ifError(err);
-    assert.strictEqual(conn1._clusterId, 'SLAVE1');
-
-    pool.getConnection(function (err, conn2) {
+  for (var i = 0; i < count; i++) {
+    pool.getConnection(function (err, conn) {
       assert.ifError(err);
-      assert.strictEqual(conn2._clusterId, 'SLAVE2');
-
-      conn1.release();
-      conn2.release();
-
-      cluster.end(function (err) {
-        assert.ifError(err);
-        server.destroy();
-      });
+      order.push(conn._clusterId);
+      conn.release();
+      done();
     });
-  });
+  }
 });