SERVER-13448 broadcast updates/deletes to all shards when more than one targeted

1 files changed, 88 insertions, 0 deletions

diff --git a/jstests/sharding/multi_write_target.js b/jstests/sharding/multi_write_target.js
new file mode 100644
index 00000000000..52ce36a83e0
--- /dev/null
+++ b/jstests/sharding/multi_write_target.js
@@ -0,0 +1,88 @@
+//
+// Tests that multi-writes (update/delete) target *all* shards and not just shards in the collection
+//
+
+var options = { separateConfig : true };
+
+var st = new ShardingTest({ shards : 3, mongos : 1, other : options });
+st.stopBalancer();
+
+var mongos = st.s0;
+var admin = mongos.getDB( "admin" );
+var shards = mongos.getCollection( "config.shards" ).find().toArray();
+var coll = mongos.getCollection( "foo.bar" );
+
+assert( admin.runCommand({ enableSharding : coll.getDB() + "" }).ok );
+printjson( admin.runCommand({ movePrimary : coll.getDB() + "", to : shards[0]._id }) );
+assert( admin.runCommand({ shardCollection : coll + "", key : { skey : 1 } }).ok );
+assert( admin.runCommand({ split : coll + "", middle : { skey : 0 } }).ok );
+assert( admin.runCommand({ moveChunk : coll + "",
+                           find : { skey : 0 },
+                           to : shards[1]._id }).ok );
+
+st.printShardingStatus();
+
+jsTest.log("Testing multi-update...");
+
+// Put data on all shards
+st.shard0.getCollection(coll.toString()).insert({ _id : 0, skey : -1, x : 1 });
+assert.gleOK(st.shard0.getCollection(coll.toString()).getDB().getLastErrorObj());
+st.shard1.getCollection(coll.toString()).insert({ _id : 1, skey : 1, x : 1 });
+assert.gleOK(st.shard1.getCollection(coll.toString()).getDB().getLastErrorObj());
+// Data not in chunks
+st.shard2.getCollection(coll.toString()).insert({ _id : 0, x : 1 });
+assert.gleOK(st.shard2.getCollection(coll.toString()).getDB().getLastErrorObj());
+
+// Non-multi-update doesn't work without shard key
+coll.update({ x : 1 }, { $set : { updated : true } }, { multi : false });
+assert.gleError(coll.getDB().getLastErrorObj());
+
+coll.update({ x : 1 }, { $set : { updated : true } }, { multi : true });
+assert.gleOK(coll.getDB().getLastErrorObj());
+
+// Ensure update goes to *all* shards
+assert.neq(null, st.shard0.getCollection(coll.toString()).findOne({ updated : true }));
+assert.neq(null, st.shard1.getCollection(coll.toString()).findOne({ updated : true }));
+assert.neq(null, st.shard2.getCollection(coll.toString()).findOne({ updated : true }));
+
+// _id update works, and goes to all shards
+coll.update({ _id : 0 }, { $set : { updatedById : true } }, { multi : false });
+assert.gleOK(coll.getDB().getLastErrorObj());
+
+// Ensure _id update goes to *all* shards
+assert.neq(null, st.shard0.getCollection(coll.toString()).findOne({ updatedById : true }));
+assert.neq(null, st.shard2.getCollection(coll.toString()).findOne({ updatedById : true }));
+
+jsTest.log("Testing multi-delete...");
+
+// non-multi-delete doesn't work without shard key
+coll.remove({ x : 1 }, { justOne : true });
+assert.gleError(coll.getDB().getLastErrorObj());
+
+coll.remove({ x : 1 }, { justOne : false });
+assert.gleOK(coll.getDB().getLastErrorObj());
+
+// Ensure delete goes to *all* shards
+assert.eq(null, st.shard0.getCollection(coll.toString()).findOne({ x : 1 }));
+assert.eq(null, st.shard1.getCollection(coll.toString()).findOne({ x : 1 }));
+assert.eq(null, st.shard2.getCollection(coll.toString()).findOne({ x : 1 }));
+
+// Put more on all shards
+st.shard0.getCollection(coll.toString()).insert({ _id : 0, skey : -1, x : 1 });
+assert.gleOK(st.shard0.getCollection(coll.toString()).getDB().getLastErrorObj());
+st.shard1.getCollection(coll.toString()).insert({ _id : 1, skey : 1, x : 1 });
+assert.gleOK(st.shard1.getCollection(coll.toString()).getDB().getLastErrorObj());
+// Data not in chunks
+st.shard2.getCollection(coll.toString()).insert({ _id : 0, x : 1 });
+assert.gleOK(st.shard2.getCollection(coll.toString()).getDB().getLastErrorObj());
+
+coll.remove({ _id : 0 }, { justOne : true });
+assert.gleOK(coll.getDB().getLastErrorObj());
+
+// Ensure _id delete goes to *all* shards
+assert.eq(null, st.shard0.getCollection(coll.toString()).findOne({ x : 1 }));
+assert.eq(null, st.shard2.getCollection(coll.toString()).findOne({ x : 1 }));
+
+jsTest.log( "DONE!" );
+
+st.stop();