diff options
Diffstat (limited to 'jstests/sharding/find_and_modify_after_multi_write.js')
| -rw-r--r-- | jstests/sharding/find_and_modify_after_multi_write.js | 105 |
1 files changed, 52 insertions, 53 deletions
diff --git a/jstests/sharding/find_and_modify_after_multi_write.js b/jstests/sharding/find_and_modify_after_multi_write.js index 749f999c54c..81c6db44a3f 100644 --- a/jstests/sharding/find_and_modify_after_multi_write.js +++ b/jstests/sharding/find_and_modify_after_multi_write.js @@ -1,73 +1,72 @@ (function() { - "use strict"; +"use strict"; - /** - * Test that a targetted findAndModify will be properly routed after executing a write that - * does not perform any shard version checks. - */ - var runTest = function(writeFunc) { - var st = new ShardingTest({shards: 2, mongos: 2}); +/** + * Test that a targetted findAndModify will be properly routed after executing a write that + * does not perform any shard version checks. + */ +var runTest = function(writeFunc) { + var st = new ShardingTest({shards: 2, mongos: 2}); - var testDB = st.s.getDB('test'); + var testDB = st.s.getDB('test'); - assert.commandWorked(testDB.adminCommand({enableSharding: 'test'})); - st.ensurePrimaryShard('test', st.shard0.shardName); + assert.commandWorked(testDB.adminCommand({enableSharding: 'test'})); + st.ensurePrimaryShard('test', st.shard0.shardName); - assert.commandWorked(testDB.adminCommand({shardCollection: 'test.user', key: {x: 1}})); - assert.commandWorked(testDB.adminCommand({split: 'test.user', middle: {x: 0}})); - assert.commandWorked(testDB.adminCommand( - {moveChunk: 'test.user', find: {x: 0}, to: st.shard1.shardName, _waitForDelete: true})); + assert.commandWorked(testDB.adminCommand({shardCollection: 'test.user', key: {x: 1}})); + assert.commandWorked(testDB.adminCommand({split: 'test.user', middle: {x: 0}})); + assert.commandWorked(testDB.adminCommand( + {moveChunk: 'test.user', find: {x: 0}, to: st.shard1.shardName, _waitForDelete: true})); - var testDB2 = st.s1.getDB('test'); - testDB2.user.insert({x: 123456}); + var testDB2 = st.s1.getDB('test'); + testDB2.user.insert({x: 123456}); - // Move chunk to bump version on a different mongos. - assert.commandWorked(testDB.adminCommand( - {moveChunk: 'test.user', find: {x: 0}, to: st.shard0.shardName, _waitForDelete: true})); + // Move chunk to bump version on a different mongos. + assert.commandWorked(testDB.adminCommand( + {moveChunk: 'test.user', find: {x: 0}, to: st.shard0.shardName, _waitForDelete: true})); - // Issue a targetted findAndModify and check that it was upserted to the right shard. - assert.commandWorked(testDB2.runCommand( - {findAndModify: 'user', query: {x: 100}, update: {$set: {y: 1}}, upsert: true})); + // Issue a targetted findAndModify and check that it was upserted to the right shard. + assert.commandWorked(testDB2.runCommand( + {findAndModify: 'user', query: {x: 100}, update: {$set: {y: 1}}, upsert: true})); - assert.neq(null, st.rs0.getPrimary().getDB('test').user.findOne({x: 100})); - assert.eq(null, st.rs1.getPrimary().getDB('test').user.findOne({x: 100})); + assert.neq(null, st.rs0.getPrimary().getDB('test').user.findOne({x: 100})); + assert.eq(null, st.rs1.getPrimary().getDB('test').user.findOne({x: 100})); - // At this point, s1 thinks the version of 'test.user' is 2, bounce it again so it gets - // incremented to 3 - assert.commandWorked(testDB.adminCommand( - {moveChunk: 'test.user', find: {x: 0}, to: st.shard1.shardName, _waitForDelete: true})); + // At this point, s1 thinks the version of 'test.user' is 2, bounce it again so it gets + // incremented to 3 + assert.commandWorked(testDB.adminCommand( + {moveChunk: 'test.user', find: {x: 0}, to: st.shard1.shardName, _waitForDelete: true})); - assert.commandWorked(testDB2.runCommand( - {findAndModify: 'user', query: {x: 200}, update: {$set: {y: 1}}, upsert: true})); + assert.commandWorked(testDB2.runCommand( + {findAndModify: 'user', query: {x: 200}, update: {$set: {y: 1}}, upsert: true})); - assert.eq(null, st.rs0.getPrimary().getDB('test').user.findOne({x: 200})); - assert.neq(null, st.rs1.getPrimary().getDB('test').user.findOne({x: 200})); + assert.eq(null, st.rs0.getPrimary().getDB('test').user.findOne({x: 200})); + assert.neq(null, st.rs1.getPrimary().getDB('test').user.findOne({x: 200})); - // At this point, s0 thinks the version of 'test.user' is 3, bounce it again so it gets - // incremented to 4 - assert.commandWorked(testDB.adminCommand( - {moveChunk: 'test.user', find: {x: 0}, to: st.shard0.shardName, _waitForDelete: true})); + // At this point, s0 thinks the version of 'test.user' is 3, bounce it again so it gets + // incremented to 4 + assert.commandWorked(testDB.adminCommand( + {moveChunk: 'test.user', find: {x: 0}, to: st.shard0.shardName, _waitForDelete: true})); - // Ensure that write commands with multi version do not reset the connection shard version - // to - // ignored. - writeFunc(testDB2); + // Ensure that write commands with multi version do not reset the connection shard version + // to + // ignored. + writeFunc(testDB2); - assert.commandWorked(testDB2.runCommand( - {findAndModify: 'user', query: {x: 300}, update: {$set: {y: 1}}, upsert: true})); + assert.commandWorked(testDB2.runCommand( + {findAndModify: 'user', query: {x: 300}, update: {$set: {y: 1}}, upsert: true})); - assert.neq(null, st.rs0.getPrimary().getDB('test').user.findOne({x: 300})); - assert.eq(null, st.rs1.getPrimary().getDB('test').user.findOne({x: 300})); + assert.neq(null, st.rs0.getPrimary().getDB('test').user.findOne({x: 300})); + assert.eq(null, st.rs1.getPrimary().getDB('test').user.findOne({x: 300})); - st.stop(); - }; + st.stop(); +}; - runTest(function(db) { - db.user.update({}, {$inc: {y: 987654}}, false, true); - }); - - runTest(function(db) { - db.user.remove({y: 'noMatch'}, false); - }); +runTest(function(db) { + db.user.update({}, {$inc: {y: 987654}}, false, true); +}); +runTest(function(db) { + db.user.remove({y: 'noMatch'}, false); +}); })(); |