/**
 * Test that we can use the $_resumeAfter and $_requestResumeToken options to resume a query.
 * @tags: [
 *  assumes_against_mongod_not_mongos,
 *  requires_find_command,
 *  requires_persistence,
 *  multiversion_incompatible,
 *  requires_getmore]
 */

(function() {
"use strict";
const testName = TestData.testName;

const testDb = db.getSiblingDB(testName);
assert.commandWorked(testDb.dropDatabase());

jsTestLog("Setting up the data.");
const testData = [{_id: 0, a: 1}, {_id: 1, b: 2}, {_id: 2, c: 3}, {_id: 3, d: 4}];
assert.commandWorked(testDb.test.insert(testData));

jsTestLog("Running the initial query.");
let res = assert.commandWorked(testDb.runCommand(
    {find: "test", hint: {$natural: 1}, batchSize: 1, $_requestResumeToken: true}));
assert.eq(1, res.cursor.firstBatch.length);
assert.contains(res.cursor.firstBatch[0], testData);
let queryData = res.cursor.firstBatch;
assert.hasFields(res.cursor, ["postBatchResumeToken"]);
let resumeToken = res.cursor.postBatchResumeToken;

// Kill the cursor before attempting to resume.
assert.commandWorked(testDb.runCommand({killCursors: "test", cursors: [res.cursor.id]}));

jsTestLog("Running the second query after killing the cursor.");
res = assert.commandWorked(testDb.runCommand({
    find: "test",
    hint: {$natural: 1},
    batchSize: 1,
    $_requestResumeToken: true,
    $_resumeAfter: resumeToken
}));
assert.eq(1, res.cursor.firstBatch.length);
// The return value should not be the same as the one before.
assert.neq(queryData[0], res.cursor.firstBatch[0]);
assert.contains(res.cursor.firstBatch[0], testData);
queryData.push(res.cursor.firstBatch[0]);
let cursorId = res.cursor.id;

jsTestLog("Running getMore.");
res =
    assert.commandWorked(testDb.runCommand({getMore: cursorId, collection: "test", batchSize: 1}));
queryData.push(res.cursor.nextBatch[0]);
assert.hasFields(res.cursor, ["postBatchResumeToken"]);
resumeToken = res.cursor.postBatchResumeToken;

// Kill the cursor before attempting to resume.
assert.commandWorked(testDb.runCommand({killCursors: "test", cursors: [res.cursor.id]}));

jsTestLog("Testing resume from getMore");
res = assert.commandWorked(testDb.runCommand({
    find: "test",
    hint: {$natural: 1},
    batchSize: 10,
    $_requestResumeToken: true,
    $_resumeAfter: resumeToken
}));
assert.eq(1, res.cursor.firstBatch.length);
// This should have exhausted the collection.
assert.eq(0, res.cursor.id);
queryData.push(res.cursor.firstBatch[0]);

assert.sameMembers(testData, queryData);
})();