summaryrefslogtreecommitdiff
path: root/jstests/sharding/find_and_modify_after_multi_write.js
diff options
context:
space:
mode:
Diffstat (limited to 'jstests/sharding/find_and_modify_after_multi_write.js')
-rw-r--r--jstests/sharding/find_and_modify_after_multi_write.js161
1 files changed, 71 insertions, 90 deletions
diff --git a/jstests/sharding/find_and_modify_after_multi_write.js b/jstests/sharding/find_and_modify_after_multi_write.js
index c8081ce9119..004fe8d8ead 100644
--- a/jstests/sharding/find_and_modify_after_multi_write.js
+++ b/jstests/sharding/find_and_modify_after_multi_write.js
@@ -1,93 +1,74 @@
(function() {
-"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 });
-
- var testDB = st.s.getDB('test');
- testDB.dropDatabase();
-
- assert.commandWorked(testDB.adminCommand({ enableSharding: 'test' }));
- st.ensurePrimaryShard('test', 'shard0000');
-
- 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: 'shard0001',
- _waitForDelete: true }));
-
- 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: 'shard0000',
- _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
- }));
-
- assert.neq(null, st.d0.getDB('test').user.findOne({ x: 100 }));
- assert.eq(null, st.d1.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: 'shard0001',
- _waitForDelete: true }));
-
- assert.commandWorked(testDB2.runCommand({
- findAndModify: 'user',
- query: { x: 200 },
- update: { $set: { y: 1 }},
- upsert: true
- }));
-
- assert.eq(null, st.d0.getDB('test').user.findOne({ x: 200 }));
- assert.neq(null, st.d1.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: 'shard0000',
- _waitForDelete: true }));
-
- // 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.neq(null, st.d0.getDB('test').user.findOne({ x: 300 }));
- assert.eq(null, st.d1.getDB('test').user.findOne({ x: 300 }));
-
- st.stop();
-};
-
-runTest(function(db) {
- db.user.update({}, { $inc: { y: 987654 }}, false, true);
-});
-
-runTest(function(db) {
- db.user.remove({ y: 'noMatch' }, false);
-});
+ "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});
+
+ var testDB = st.s.getDB('test');
+ testDB.dropDatabase();
+
+ assert.commandWorked(testDB.adminCommand({enableSharding: 'test'}));
+ st.ensurePrimaryShard('test', 'shard0000');
+
+ 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: 'shard0001', _waitForDelete: true}));
+
+ 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: 'shard0000', _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}));
+
+ assert.neq(null, st.d0.getDB('test').user.findOne({x: 100}));
+ assert.eq(null, st.d1.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: 'shard0001', _waitForDelete: true}));
+
+ assert.commandWorked(testDB2.runCommand(
+ {findAndModify: 'user', query: {x: 200}, update: {$set: {y: 1}}, upsert: true}));
+
+ assert.eq(null, st.d0.getDB('test').user.findOne({x: 200}));
+ assert.neq(null, st.d1.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: 'shard0000', _waitForDelete: true}));
+
+ // 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.neq(null, st.d0.getDB('test').user.findOne({x: 300}));
+ assert.eq(null, st.d1.getDB('test').user.findOne({x: 300}));
+
+ st.stop();
+ };
+
+ runTest(function(db) {
+ db.user.update({}, {$inc: {y: 987654}}, false, true);
+ });
+
+ runTest(function(db) {
+ db.user.remove({y: 'noMatch'}, false);
+ });
})();