//
// Tests writeConcerns with findAndModify command
//
(function() {
    'use strict';

    var nodeCount = 3;
    var rst = new ReplSetTest({nodes: nodeCount});
    rst.startSet({nojournal: ""});
    rst.initiate();

    var primary = rst.getPrimary();
    var coll = primary.getCollection("test.find_and_modify_wc");
    coll.remove({});

    // insert some documents
    var docs = [];
    for (var i = 1; i <= 5; ++i) {
        docs.push({i: i, j: 2 * i});
    }
    var res =
        coll.runCommand({insert: coll.getName(), documents: docs, writeConcern: {w: nodeCount}});
    assert(res.ok);
    assert.eq(5, coll.find().itcount());

    // use for updates in subsequent runCommand calls
    var reqUpdate = {
        findAndModify: coll.getName(),
        query: {i: 3},
        update: {$inc: {j: 1}},
        writeConcern: {w: 'majority'}
    };

    // Verify findAndModify returns old document new: false
    var res = coll.runCommand(reqUpdate);
    assert(res.ok);
    assert(res.value);
    // (2 * res.value.i) == 6 == res.value.j (old document)
    assert.eq(2 * res.value.i, res.value.j);
    assert(!res.writeConcernError);

    // Verify findAndModify returns new document with new: true
    reqUpdate.new = true;
    res = coll.runCommand(reqUpdate);
    assert(res.ok);
    assert(res.value);
    // (2 * res.value.i + 2) == 8 == res.value.j (new document after two updates)
    assert.eq(2 * res.value.i + 2, res.value.j);
    assert(!res.writeConcernError);

    // Verify findAndModify remove works
    res = coll.runCommand(
        {findAndModify: coll.getName(), sort: {i: 1}, remove: true, writeConcern: {w: nodeCount}});
    assert.eq(res.value.i, 1);
    assert.eq(coll.find().itcount(), 4);
    assert(!res.writeConcernError);

    // Verify findAndModify returns writeConcernError
    // when given invalid writeConcerns
    [{w: 'invalid'}, {w: nodeCount + 1}].forEach(function(wc) {
        reqUpdate.writeConcern = wc;
        res = coll.runCommand(reqUpdate);

        assert(res.writeConcernError);
        assert(res.writeConcernError.code);
        assert(res.writeConcernError.errmsg);
    });

    rst.stopSet();

})();