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());

assert.commandWorked(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++);
assert(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++);
assert(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++);
assert(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++);
assert(coll.drop());
assert.commandWorked(coll.update({_id: 1}, {$addToSet: {a: {$each: [1, 2]}}}, true));
assert.eq(doc, coll.findOne());

coll = db.getCollection(collNamePrefix + collCount++);
assert(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++);
assert(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++);
assert(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++);
assert(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++);
assert(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);