diff options
Diffstat (limited to 'jstests/core/find_and_modify.js')
| -rw-r--r-- | jstests/core/find_and_modify.js | 35 |
1 files changed, 26 insertions, 9 deletions
diff --git a/jstests/core/find_and_modify.js b/jstests/core/find_and_modify.js index 83e1e7da79d..8f437726c51 100644 --- a/jstests/core/find_and_modify.js +++ b/jstests/core/find_and_modify.js @@ -33,6 +33,32 @@ out = t.findAndModify( {query: {inprogress: false}, sort: {priority: -1}, update: {$set: {inprogress: true}}}); assert.eq(out.priority, 9); +// Use expressions in the 'fields' argument with 'new' false. +out = t.findAndModify({ + query: {inprogress: false}, + sort: {priority: -1}, + 'new': false, + update: {$set: {inprogress: true}, $inc: {value: 1}}, + fields: {priority: 1, inprogress: 1, computedField: {$add: ["$value", 2]}} +}); +assert.eq(out.priority, 8); +assert.eq(out.inprogress, false); +// The projection should have been applied to the pre image of the update. +assert.eq(out.computedField, 2); + +// Use expressions in the 'fields' argument with 'new' true. +out = t.findAndModify({ + query: {inprogress: false}, + sort: {priority: -1}, + update: {$set: {inprogress: true}, $inc: {value: 1}}, + 'new': true, + fields: {priority: 1, inprogress: 1, computedField: {$add: ["$value", 2]}} +}); +assert.eq(out.priority, 7); +assert.eq(out.inprogress, true); +// The projection should have been applied to the update post image. +assert.eq(out.computedField, 3); + // remove lowest priority out = t.findAndModify({sort: {priority: 1}, remove: true}); assert.eq(out.priority, 1); @@ -122,15 +148,6 @@ cmdRes = db.runCommand({ }); assert.commandFailed(cmdRes); -// Cannot use expressions in the projection part of findAndModify. -cmdRes = db.runCommand({ - findAndModify: "coll", - update: {a: 1}, - fields: {b: {$ln: ['c']}}, - upsert: true, -}); -assert.commandFailedWithCode(cmdRes, ErrorCodes.BadValue); - // // SERVER-17372 // |