SERVER-8335 test for autosplit heuristics

1 files changed, 72 insertions, 0 deletions

diff --git a/jstests/slowNightly/autosplit_heuristics.js b/jstests/slowNightly/autosplit_heuristics.js
new file mode 100644
index 00000000000..51476a893cc
--- /dev/null
+++ b/jstests/slowNightly/autosplit_heuristics.js
@@ -0,0 +1,72 @@
+//
+// Tests autosplitting heuristics, and that the heuristic counting of chunk sizes
+// works as expected even after splitting.
+//
+
+var st = new ShardingTest({ shards : 1, 
+                            mongos : 1, 
+                            other : { mongosOptions : { chunkSize : 1, verbose : 1 }, 
+                            separateConfig : true } });
+
+// The balancer may interfere unpredictably with the chunk moves/splits depending on timing.
+st.stopBalancer();
+
+var mongos = st.s0;
+var config = mongos.getDB("config");
+var admin = mongos.getDB("admin");
+var coll = mongos.getCollection("foo.hashBar");
+
+printjson(admin.runCommand({ enableSharding : coll.getDB() + "" }));
+printjson(admin.runCommand({ shardCollection : coll + "", key : { _id : 1 } }));
+
+var numChunks = 10;
+
+// Split off the low and high chunks, to get non-special-case behavior
+printjson(admin.runCommand({ split : coll + "", middle : { _id : 0 } }));
+printjson(admin.runCommand({ split : coll + "", middle : { _id : numChunks } }));
+
+// Split all the other chunks
+for (var i = 1; i < numChunks; i++) {
+    printjson(admin.runCommand({ split : coll + "", middle : { _id : i } }));
+}
+
+jsTest.log("Setup collection...");
+st.printShardingStatus(true);
+
+var approxSize = Object.bsonsize({ _id : 0.0 });
+
+jsTest.log("Starting inserts of approx size: " + approxSize + "...");
+
+var chunkSizeBytes = 1024 * 1024;
+
+// We insert slightly more than the max number of docs per chunk, to test
+// if resetting the chunk size happens during reloads.  If the size is 
+// reset, we'd expect to split less, since the first split would then
+// disable further splits (statistically, since the decision is randomized).
+var insertsForSplit = Math.ceil((chunkSizeBytes * 1.25) / approxSize);
+var totalInserts = insertsForSplit * numChunks;
+
+printjson({ chunkSizeBytes : chunkSizeBytes,
+            insertsForSplit : insertsForSplit,
+            totalInserts : totalInserts });
+
+// Insert enough docs to trigger splits into all chunks
+for (var i = 0; i < totalInserts; i++) {
+    coll.insert({ _id : i % numChunks + (i / totalInserts) });
+}
+
+assert.eq(null, coll.getDB().getLastError());
+
+jsTest.log("Inserts completed...");
+
+st.printShardingStatus(true);
+printjson(coll.stats());
+
+// Check that all chunks (except the two extreme chunks)
+// have been split at least once.
+assert.gte(config.chunks.count(), numChunks * 2 + 2);
+
+jsTest.log("DONE!");
+
+st.stop();
+