SERVER-22079 Make sharding_rs1.js more compact

1 files changed, 60 insertions, 0 deletions

diff --git a/jstests/sharding/sharding_rs1.js b/jstests/sharding/sharding_rs1.js
new file mode 100644
index 00000000000..353f9c532cc
--- /dev/null
+++ b/jstests/sharding/sharding_rs1.js
@@ -0,0 +1,60 @@
+// tests sharding with replica sets
+(function() {
+'use strict';
+
+var s = new ShardingTest({ shards: 3,
+                           other: { rs: true, chunkSize: 1, enableBalancer: true }});
+
+s.adminCommand( { enablesharding : "test" } );
+s.ensurePrimaryShard('test', 'test-rs0');
+s.config.settings.update( { _id: "balancer" }, { $set : { _waitForDelete : true } } , true );
+
+var db = s.getDB("test");
+
+var bigString = "X".repeat(256 * 1024);
+
+var insertedBytes = 0;
+var num = 0;
+
+// Insert 10 MB of data to result in 10+ chunks
+var bulk = db.foo.initializeUnorderedBulkOp();
+while (insertedBytes < (10 * 1024 * 1024)) {
+    bulk.insert({ _id: num++, s: bigString, x: Math.random() });
+    insertedBytes += bigString.length;
+}
+assert.writeOK(bulk.execute({w: 3}));
+
+assert.commandWorked(s.s.adminCommand({ shardcollection: "test.foo" , key: { _id: 1 } }));
+
+jsTest.log("Waiting for balance to complete");
+s.awaitBalance('foo', 'test', 2 * 60 * 1000);
+
+jsTest.log("Stopping balancer");
+s.stopBalancer();
+
+jsTest.log("Balancer stopped, checking dbhashes");
+s._rs.forEach(function(rsNode) {
+    rsNode.test.awaitReplication();
+
+    var dbHashes = rsNode.test.getHashes("test");
+    print(rsNode.url + ': ' + tojson(dbHashes));
+
+    for (var j = 0; j < dbHashes.slaves.length; j++) {
+        assert.eq(dbHashes.master.md5,
+                  dbHashes.slaves[j].md5,
+                  "hashes not same for: " + rsNode.url + " slave: " + j);
+    }
+});
+
+assert.eq( num , db.foo.find().count() , "C1" )
+assert.eq( num , db.foo.find().itcount() , "C2" )
+assert.eq( num , db.foo.find().sort( { _id : 1 } ).itcount() , "C3" )
+assert.eq( num , db.foo.find().sort( { _id : -1 } ).itcount() , "C4" )
+
+db.foo.ensureIndex( { x : 1 } );
+assert.eq( num , db.foo.find().sort( { x : 1 } ).itcount() , "C5" )
+assert.eq( num , db.foo.find().sort( { x : -1 } ).itcount() , "C6" )
+
+s.stop();
+
+})();