// Verify that a user with read and write access to database "test" cannot access database "test2"
// via a mapper, reducer or finalizer.

import {ShardingTest} from "jstests/libs/shardingtest.js";

//
// User document declarations.  All users in this test are added to the admin database.
//

let adminUser = {
    user: "admin",
    pwd: "a",
    roles: ["readWriteAnyDatabase", "dbAdminAnyDatabase", "userAdminAnyDatabase", "clusterAdmin"],
};

let test1User = {
    user: "test",
    pwd: "a",
    roles: [{role: "readWrite", db: "test1", hasRole: true, canDelegate: false}],
};

function assertRemove(collection, pattern) {
    assert.commandWorked(collection.remove(pattern));
}

function assertInsert(collection, obj) {
    assert.commandWorked(collection.insert(obj));
}

let cluster = new ShardingTest({name: "authmr", shards: 1, mongos: 1, other: {keyFile: "jstests/libs/key1"}});

// Set up the test data.
(function () {
    let adminDB = cluster.getDB("admin");
    let test1DB = adminDB.getSiblingDB("test1");
    let test2DB = adminDB.getSiblingDB("test2");
    let ex;
    try {
        adminDB.createUser(adminUser);
        assert(adminDB.auth(adminUser.user, adminUser.pwd));

        adminDB.dropUser(test1User.user);
        adminDB.createUser(test1User);

        assertInsert(test1DB.foo, {a: 1});
        assertInsert(test1DB.foo, {a: 2});
        assertInsert(test1DB.foo, {a: 3});
        assertInsert(test1DB.foo, {a: 4});
        assertInsert(test2DB.foo, {x: 1});
    } finally {
        adminDB.logout();
    }
})();

assert.throws(function () {
    let adminDB = cluster.getDB("admin");
    let test1DB;
    let test2DB;
    assert(adminDB.auth(test1User.user, test1User.pwd));
    try {
        test1DB = adminDB.getSiblingDB("test1");
        test2DB = adminDB.getSiblingDB("test2");

        // Sanity check.  test1User can count (read) test1, but not test2.
        assert.eq(test1DB.foo.count(), 4);
        assert.throws(test2DB.foo.count);

        test1DB.foo.mapReduce(
            function () {
                emit(0, this.a);
                let t2 = new Mongo().getDB("test2");
                t2.ad.insert(this);
            },
            function (k, vs) {
                let t2 = new Mongo().getDB("test2");
                t2.reductio.insert(this);

                return Array.sum(vs);
            },
            {
                out: "bar",
                finalize: function (k, v) {
                    for (k in this) {
                        if (this.hasOwnProperty(k)) print(k + "=" + v);
                    }
                    let t2 = new Mongo().getDB("test2");
                    t2.absurdum.insert({key: k, value: v});
                },
            },
        );
    } finally {
        adminDB.logout();
    }
});

(function () {
    let adminDB = cluster.getDB("admin");
    assert(adminDB.auth(adminUser.user, adminUser.pwd));
    try {
        let test2DB = cluster.getDB("test2");
        assert.eq(test2DB.reductio.count(), 0, "reductio");
        assert.eq(test2DB.ad.count(), 0, "ad");
        assert.eq(test2DB.absurdum.count(), 0, "absurdum");
    } finally {
        adminDB.logout();
    }
})();

cluster.stop();