// Test read after opTime functionality with maxTimeMS.

(function() {
    "use strict";

    var replTest = new ReplSetTest({nodes: 2});
    replTest.startSet();
    replTest.initiate();
    var config = replTest.getReplSetConfigFromNode();

    var runTest = function(testDB, primaryConn) {
        var dbName = testDB.getName();
        assert.writeOK(primaryConn.getDB(dbName).user.insert({x: 1}, {writeConcern: {w: 2}}));

        var localDB = primaryConn.getDB('local');

        var oplogTS = localDB.oplog.rs.find().sort({$natural: -1}).limit(1).next();
        var twoKSecTS = new Timestamp(oplogTS.ts.getTime() + 2000, 0);

        var term = -1;
        if (config.protocolVersion === 1) {
            term = oplogTS.t;
        }

        // Test timeout with maxTimeMS
        var runTimeoutTest = function() {
            assert.commandFailedWithCode(testDB.runCommand({
                find: 'user',
                filter: {x: 1},
                readConcern: {afterOpTime: {ts: twoKSecTS, t: term}},
                maxTimeMS: 5000,
            }),
                                         ErrorCodes.ExceededTimeLimit);
        };

        var countLogMessages = function(msg) {
            var total = 0;
            var logMessages = assert.commandWorked(testDB.adminCommand({getLog: 'global'})).log;
            for (var i = 0; i < logMessages.length; i++) {
                if (logMessages[i].indexOf(msg) != -1) {
                    total++;
                }
            }
            return total;
        };

        var checkLog = function(msg, expectedCount) {
            var count;
            assert.soon(
                function() {
                    count = countLogMessages(msg);
                    return expectedCount == count;
                },
                'Expected ' + expectedCount + ', but instead saw ' + count +
                    ' log entries containing the following message: ' + msg,
                60000,
                300);
        };

        // Run the time out test 3 times with replication debug log level increased to 2
        // for first and last run. The time out message should be logged twice.
        testDB.setLogLevel(2, 'command');
        runTimeoutTest();
        testDB.setLogLevel(0, 'command');

        var msg = 'Command on database ' + testDB.getName() +
            ' timed out waiting for read concern to be satisfied. Command:';
        checkLog(msg, 1);

        // Read concern timed out message should not be logged.
        runTimeoutTest();

        testDB.setLogLevel(2, 'command');
        runTimeoutTest();
        testDB.setLogLevel(0, 'command');

        checkLog(msg, 2);

        // Test read on future afterOpTime that will eventually occur.
        primaryConn.getDB(dbName).parallelShellStarted.drop();
        var insertFunc =
            startParallelShell('let testDB = db.getSiblingDB("' + dbName + '"); ' +
                                   'testDB.parallelShellStarted.insert({}); ' +
                                   'sleep(2100); ' +
                                   'testDB.user.insert({y: 1}, { writeConcern: { w: 2 }});',
                               primaryConn.port);
        assert.soon(function() {
            return primaryConn.getDB(dbName).parallelShellStarted.count();
        });

        oplogTS = localDB.oplog.rs.find().sort({$natural: -1}).limit(1).next();
        var twoSecTS = new Timestamp(oplogTS.ts.getTime() + 2, 0);
        var res = assert.commandWorked(testDB.runCommand({
            find: 'user',
            filter: {x: 1},
            readConcern: {
                afterOpTime: {ts: twoSecTS, t: term},
            },
            maxTimeMS: 10 * 1000,
        }));

        assert.eq(null, res.code);

        insertFunc();
    };

    var primary = replTest.getPrimary();
    runTest(primary.getDB('test1'), primary);
    runTest(replTest.getSecondary().getDB('test2'), primary);

    replTest.stopSet();
})();