SERVER-25125 Report initial sync progress in ReplSetGetStatus

1 files changed, 109 insertions, 0 deletions

diff --git a/jstests/noPassthroughWithMongod/initial_sync_replSetGetStatus.js b/jstests/noPassthroughWithMongod/initial_sync_replSetGetStatus.js
new file mode 100644
index 00000000000..df5600dc5a6
--- /dev/null
+++ b/jstests/noPassthroughWithMongod/initial_sync_replSetGetStatus.js
@@ -0,0 +1,109 @@
+/**
+ * This test tests that replSetGetStatus returns initial sync stats while initial sync is in
+ * progress.
+ */
+
+(function() {
+    "use strict";
+    // If the parameter is already set, don't run this test.
+    var parameters = db.adminCommand({getCmdLineOpts: 1}).parsed.setParameter;
+    if (parameters.use3dot2InitialSync || parameters.initialSyncOplogBuffer) {
+        jsTest.log("Skipping initial_sync_parameters.js because use3dot2InitialSync or " +
+                   "initialSyncOplogBuffer was already provided.");
+        return;
+    }
+
+    var name = 'initial_sync_replSetGetStatus';
+    var replSet = new ReplSetTest({
+        name: name,
+        nodes: 1,
+    });
+
+    replSet.startSet();
+    replSet.initiate();
+    var primary = replSet.getPrimary();
+
+    var coll = primary.getDB('test').foo;
+    assert.writeOK(coll.insert({a: 1}));
+    assert.writeOK(coll.insert({a: 2}));
+
+    // Add a secondary node but make it hang before copying databases.
+    var secondary = replSet.add(
+        {setParameter: {use3dot2InitialSync: false, initialSyncOplogBuffer: "collection"}});
+    secondary.setSlaveOk();
+
+    assert.commandWorked(secondary.getDB('admin').runCommand(
+        {configureFailPoint: 'initialSyncHangBeforeCopyingDatabases', mode: 'alwaysOn'}));
+    assert.commandWorked(secondary.getDB('admin').runCommand(
+        {configureFailPoint: 'initialSyncHangBeforeFinish', mode: 'alwaysOn'}));
+    replSet.reInitiate();
+
+    // Wait for fail point message to be logged.
+    var checkLog = function(node, msg) {
+        assert.soon(function() {
+            var logMessages = assert.commandWorked(node.adminCommand({getLog: 'global'})).log;
+            for (var i = 0; i < logMessages.length; i++) {
+                if (logMessages[i].indexOf(msg) != -1) {
+                    return true;
+                }
+            }
+            return false;
+        }, 'Did not see a log entry containing the following message: ' + msg, 10000, 1000);
+    };
+
+    // Wait for initial sync to pause before it copies the databases.
+    checkLog(secondary, 'initial sync - initialSyncHangBeforeCopyingDatabases fail point enabled');
+
+    // Test that replSetGetStatus returns the correct results while initial sync is in progress.
+    var res = assert.commandWorked(secondary.adminCommand({replSetGetStatus: 1}));
+    assert(!res.initialSyncStatus,
+           "Response should not have an 'initialSyncStatus' field: " + tojson(res));
+
+    res = assert.commandWorked(secondary.adminCommand({replSetGetStatus: 1, initialSync: 1}));
+    assert(res.initialSyncStatus,
+           "Response should have an 'initialSyncStatus' field: " + tojson(res));
+
+    assert.commandFailed(secondary.adminCommand({replSetGetStatus: 1, initialSync: "t"}),
+                         ErrorCodes.TypeMismatch);
+
+    assert.writeOK(coll.insert({a: 3}));
+    assert.writeOK(coll.insert({a: 4}));
+
+    // Let initial sync continue working.
+    assert.commandWorked(secondary.getDB('admin').runCommand(
+        {configureFailPoint: 'initialSyncHangBeforeCopyingDatabases', mode: 'off'}));
+
+    // Wait for initial sync to pause right before it finishes.
+    checkLog(secondary, 'initial sync - initialSyncHangBeforeFinish fail point enabled');
+
+    // Test that replSetGetStatus returns the correct results when initial sync is at the very end.
+    res = assert.commandWorked(secondary.adminCommand({replSetGetStatus: 1, initialSync: 1}));
+    assert(res.initialSyncStatus, "Response should have an 'initialSyncStatus' field.");
+    assert.eq(res.initialSyncStatus.fetchedMissingDocs, 0);
+    assert.eq(res.initialSyncStatus.appliedOps, 2);
+    assert.eq(res.initialSyncStatus.failedInitialSyncAttempts, 0);
+    assert.eq(res.initialSyncStatus.maxFailedInitialSyncAttempts, 10);
+    assert.eq(res.initialSyncStatus.databases.databasesCloned, 1);
+    assert.eq(res.initialSyncStatus.databases.test.collections, 1);
+    assert.eq(res.initialSyncStatus.databases.test.clonedCollections, 1);
+    assert.eq(res.initialSyncStatus.databases.test["test.foo"].documents, 4);
+    assert.eq(res.initialSyncStatus.databases.test["test.foo"].indexes, 1);
+    assert.eq(res.initialSyncStatus.databases.test["test.foo"].fetchedBatches, 1);
+
+    // Let initial sync finish and get into secondary state.
+    assert.commandWorked(secondary.getDB('admin').runCommand(
+        {configureFailPoint: 'initialSyncHangBeforeFinish', mode: 'off'}));
+    replSet.awaitSecondaryNodes(60 * 1000);
+
+    // Test that replSetGetStatus returns the correct results after initial sync is finished.
+    res = assert.commandWorked(secondary.adminCommand({replSetGetStatus: 1}));
+    assert(!res.initialSyncStatus,
+           "Response should not have an 'initialSyncStatus' field: " + tojson(res));
+
+    res = assert.commandWorked(secondary.adminCommand({replSetGetStatus: 1, initialSync: 1}));
+    assert(!res.initialSyncStatus,
+           "Response should not have an 'initialSyncStatus' field: " + tojson(res));
+
+    assert.commandFailedWithCode(secondary.adminCommand({replSetGetStatus: 1, initialSync: "m"}),
+                                 ErrorCodes.TypeMismatch);
+})();