/**
 * Tests the operationTime API for the following topologies:
 *   - mongos talking to a sharded replica set (sharded and unsharded collections)
 *   - mongod from a sharded replica set
 *   - mongod from a non-sharded replica set
 *   - standalone mongod
 */
import {ReplSetTest} from "jstests/libs/replsettest.js";
import {ShardingTest} from "jstests/libs/shardingtest.js";
import {isTimestamp} from "jstests/libs/timestamp_util.js";

function responseContainsTimestampOperationTime(res) {
    return res.operationTime !== undefined && isTimestamp(res.operationTime);
}

// A mongos that talks to a non-sharded collection on a sharded replica set returns an
// operationTime that is a Timestamp.
let st = new ShardingTest({name: "operation_time_api", shards: {rs0: {nodes: 1}}});

let testDB = st.s.getDB("test");
let res = assert.commandWorked(testDB.runCommand({insert: "foo", documents: [{x: 1}]}));
assert(
    responseContainsTimestampOperationTime(res),
    "Expected response from a mongos talking to a non-sharded collection on a sharded " +
        "replica set to contain an operationTime, received: " +
        tojson(res),
);

// A mongos that talks to a sharded collection on a sharded replica set returns an operationTime
// that is a Timestamp.
assert.commandWorked(st.s.adminCommand({enableSharding: "test"}));
assert.commandWorked(st.s.adminCommand({shardCollection: "test.bar", key: {x: 1}}));

res = assert.commandWorked(testDB.runCommand({insert: "bar", documents: [{x: 2}]}));
assert(
    responseContainsTimestampOperationTime(res),
    "Expected response from a mongos inserting to a sharded collection on a sharded " +
        "replica set to contain an operationTime, received: " +
        tojson(res),
);

// A mongod in a sharded replica set returns an operationTime that is a Timestamp.
testDB = st.rs0.getPrimary().getDB("test");
res = assert.commandWorked(testDB.runCommand({insert: "foo", documents: [{x: 3}]}));
assert(
    responseContainsTimestampOperationTime(res),
    "Expected response from a mongod in a sharded replica set to contain an " +
        "operationTime, received: " +
        tojson(res),
);

st.stop();

if (jsTestOptions().useAutoBootstrapProcedure) {
    // TODO: SERVER-80318 Delete tests below
    quit();
}

// A mongod from a non-sharded replica set returns an operationTime that is a Timestamp.
let replTest = new ReplSetTest({name: "operation_time_api_non_sharded_replset", nodes: 1});
replTest.startSet();
replTest.initiate();

testDB = replTest.getPrimary().getDB("test");
res = assert.commandWorked(testDB.runCommand({insert: "foo", documents: [{x: 4}]}));
assert(
    responseContainsTimestampOperationTime(res),
    "Expected response from a non-sharded replica set to contain an operationTime, " + "received: " + tojson(res),
);

replTest.stopSet();

// A standalone mongod does not return an operationTime.
let standalone = MongoRunner.runMongod();

testDB = standalone.getDB("test");
res = assert.commandWorked(testDB.runCommand({insert: "foo", documents: [{x: 5}]}));
assert(
    !responseContainsTimestampOperationTime(res),
    "Expected response from a standalone mongod to not contain an operationTime, " + "received: " + tojson(res),
);

MongoRunner.stopMongod(standalone);