SERVER-15525 Split sharding map/reduce tests

1 files changed, 53 insertions, 0 deletions

diff --git a/jstests/sharding/mapReduce_outSharded.js b/jstests/sharding/mapReduce_outSharded.js
new file mode 100644
index 00000000000..0e8f68acf69
--- /dev/null
+++ b/jstests/sharding/mapReduce_outSharded.js
@@ -0,0 +1,53 @@
+var verifyOutput = function(out) {
+    printjson(out);
+    assert.eq(out.counts.input, 51200, "input count is wrong");
+    assert.eq(out.counts.emit, 51200, "emit count is wrong");
+    assert.gt(out.counts.reduce, 99, "reduce count is wrong");
+    assert.eq(out.counts.output, 512, "output count is wrong");
+}
+
+var st = new ShardingTest({ shards : 2, verbose : 1, mongos : 1, other : { chunksize : 1 } });
+st.startBalancer();
+
+st.adminCommand( { enablesharding : "mrShard" } )
+st.adminCommand( { shardcollection : "mrShard.srcSharded", key : { "_id" : 1 } } )
+
+var db = st.getDB( "mrShard" );
+
+var bulk = db.srcNonSharded.initializeUnorderedBulkOp();
+for (j = 0; j < 100; j++) {
+    for (i = 0; i < 512; i++) {
+        bulk.insert({ j: j, i: i });
+    }
+}
+assert.writeOK(bulk.execute());
+
+function map() { emit(this.i, 1); }
+function reduce(key, values) { return Array.sum(values) } 
+
+// non sharded src sharded dst
+var suffix = "OutSharded";
+
+out = db.srcNonSharded.mapReduce(map, reduce, { out: { replace: "mrReplace" + suffix, sharded: true } });
+verifyOutput(out);
+
+out = db.srcNonSharded.mapReduce(map, reduce, { out: { merge: "mrMerge" + suffix, sharded: true } });
+verifyOutput(out);
+
+out = db.srcNonSharded.mapReduce(map, reduce, { out: { reduce: "mrReduce" + suffix, sharded: true } });
+verifyOutput(out);
+
+out = db.srcNonSharded.mapReduce(map, reduce, { out: { inline: 1, sharded: true } });
+verifyOutput(out);
+assert(out.results != 'undefined', "no results for inline");
+
+out = db.srcNonSharded.mapReduce(map, reduce, { out: { replace: "mrReplace" + suffix, db: "mrShardOtherDB", sharded: true } });
+verifyOutput(out);
+
+out = db.runCommand({
+    mapReduce: "srcNonSharded", // use new name mapReduce rather than mapreduce
+    map: map,
+    reduce: reduce,
+    out: "mrBasic" + "srcNonSharded",
+  });
+verifyOutput(out);