mongo/jstests/core/write/update/update_addToSet.js

// Cannot implicitly shard accessed collections because of following errmsg: A single
// update/delete on a sharded collection must contain an exact match on _id or contain the shard
// key.
//
// @tags: [
//   assumes_unsharded_collection,
// ]

(function() {
"use strict";

const collNamePrefix = 'update_addToSet_';
let collCount = 0;
let coll = db.getCollection(collNamePrefix + collCount++);
coll.drop();

let doc = {_id: 1, a: [2, 1]};
assert.commandWorked(coll.insert(doc));

assert.eq(doc, coll.findOne());

assert.commandWorked(coll.update({}, {$addToSet: {a: 3}}));
doc.a.push(3);
assert.eq(doc, coll.findOne());

coll.update({}, {$addToSet: {a: 3}});
assert.eq(doc, coll.findOne());

// SERVER-628
assert.commandWorked(coll.update({}, {$addToSet: {a: {$each: [3, 5, 6]}}}));
doc.a.push(5);
doc.a.push(6);
assert.eq(doc, coll.findOne());

coll = db.getCollection(collNamePrefix + collCount++);
coll.drop();
doc = {
    _id: 1,
    a: [3, 5, 6]
};
assert.commandWorked(coll.insert(doc));
assert.commandWorked(coll.update({}, {$addToSet: {a: {$each: [3, 5, 6]}}}));
assert.eq(doc, coll.findOne());

coll = db.getCollection(collNamePrefix + collCount++);
coll.drop();
assert.commandWorked(coll.update({_id: 1}, {$addToSet: {a: {$each: [3, 5, 6]}}}, true));
assert.eq(doc, coll.findOne());
assert.commandWorked(coll.update({_id: 1}, {$addToSet: {a: {$each: [3, 5, 6]}}}, true));
assert.eq(doc, coll.findOne());

// SERVER-630
coll = db.getCollection(collNamePrefix + collCount++);
coll.drop();
assert.commandWorked(coll.update({_id: 2}, {$addToSet: {a: 3}}, true));
assert.eq(1, coll.find({}).itcount());
assert.eq({_id: 2, a: [3]}, coll.findOne());

// SERVER-3245
doc = {
    _id: 1,
    a: [1, 2]
};
coll = db.getCollection(collNamePrefix + collCount++);
coll.drop();
assert.commandWorked(coll.update({_id: 1}, {$addToSet: {a: {$each: [1, 2]}}}, true));
assert.eq(doc, coll.findOne());

coll = db.getCollection(collNamePrefix + collCount++);
coll.drop();
assert.commandWorked(coll.update({_id: 1}, {$addToSet: {a: {$each: [1, 2, 1, 2]}}}, true));
assert.eq(doc, coll.findOne());

coll = db.getCollection(collNamePrefix + collCount++);
coll.drop();
assert.commandWorked(coll.insert({_id: 1}));
assert.commandWorked(coll.update({_id: 1}, {$addToSet: {a: {$each: [1, 2, 2, 1]}}}));
assert.eq(doc, coll.findOne());

assert.commandWorked(coll.update({_id: 1}, {$addToSet: {a: {$each: [3, 2, 2, 3, 3]}}}));
doc.a.push(3);
assert.eq(doc, coll.findOne());

// Test that dotted and '$' prefixed field names work when nested.
coll = db.getCollection(collNamePrefix + collCount++);
coll.drop();
doc = {
    _id: 1,
    a: [1, 2]
};
assert.commandWorked(coll.insert(doc));

assert.commandWorked(coll.update({}, {$addToSet: {a: {'x.$.y': 'bad'}}}));
assert.commandWorked(coll.update({}, {$addToSet: {a: {b: {'x.$.y': 'bad'}}}}));

assert.commandWorked(coll.update({}, {$addToSet: {a: {"$bad": "bad"}}}));
assert.commandWorked(coll.update({}, {$addToSet: {a: {b: {"$bad": "bad"}}}}));

assert.commandWorked(coll.update({}, {$addToSet: {a: {_id: {"x.y": 2}}}}));

assert.commandWorked(coll.update({}, {$addToSet: {a: {$each: [{'x.$.y': 'bad'}]}}}));
assert.commandWorked(coll.update({}, {$addToSet: {a: {$each: [{b: {'x.$.y': 'bad'}}]}}}));

assert.commandWorked(coll.update({}, {$addToSet: {a: {$each: [{'$bad': 'bad'}]}}}));
assert.commandWorked(coll.update({}, {$addToSet: {a: {$each: [{b: {'$bad': 'bad'}}]}}}));

// Test that nested _id fields are allowed.
coll = db.getCollection(collNamePrefix + collCount++);
coll.drop();
doc = {
    _id: 1,
    a: [1, 2]
};
assert.commandWorked(coll.insert(doc));

assert.commandWorked(coll.update({}, {$addToSet: {a: {_id: ["foo", "bar", "baz"]}}}));
assert.commandWorked(coll.update({}, {$addToSet: {a: {_id: /acme.*corp/}}}));

// Test that DBRefs are allowed.
coll = db.getCollection(collNamePrefix + collCount++);
coll.drop();
doc = {
    _id: 1,
    a: [1, 2]
};
assert.commandWorked(coll.insert(doc));

const foo = {
    "foo": "bar"
};
assert.commandWorked(coll.insert(foo));
const fooDoc = coll.findOne(foo);
assert.eq(fooDoc.foo, foo.foo);

const fooDocRef = {
    reference: new DBRef(coll.getName(), fooDoc._id, coll.getDB().getName())
};

assert.commandWorked(coll.update({_id: doc._id}, {$addToSet: {a: fooDocRef}}));
assert.eq(coll.findOne({_id: doc._id}).a[2], fooDocRef);

assert.commandWorked(coll.update({_id: doc._id}, {$addToSet: {a: {b: fooDocRef}}}));
assert.eq(coll.findOne({_id: doc._id}).a[3].b, fooDocRef);
}());