SERVER-30620 Lookup collection by UUID in fetchAndInsertMissingDocument

1 files changed, 81 insertions, 0 deletions

diff --git a/jstests/replsets/initial_sync_update_missing_doc3.js b/jstests/replsets/initial_sync_update_missing_doc3.js
new file mode 100644
index 00000000000..27f06953b9a
--- /dev/null
+++ b/jstests/replsets/initial_sync_update_missing_doc3.js
@@ -0,0 +1,81 @@
+/**
+ * Initial sync runs in several phases - the first 3 are as follows:
+ * 1) fetches the last oplog entry (op_start1) on the source;
+ * 2) copies all non-local databases from the source; and
+ * 3) fetches and applies operations from the source after op_start1.
+ *
+ * This test updates and deletes a document on the source between phases 1 and 2. The
+ * secondary will initially fail to apply the update operation in phase 3 and subsequently have
+ * to attempt to check the source for a new copy of the document. Before the secondary checks the
+ * source, we insert a new copy of the document into this collection on the source and mark the
+ * collection on which the document to be fetched resides as drop pending, thus effectively
+ * renaming the collection but preserving the UUID. This ensures the secondary fetches the
+ * document by UUID rather than namespace.
+ */
+
+(function() {
+    load("jstests/libs/check_log.js");
+
+    var name = 'initial_sync_update_missing_doc3';
+    var replSet = new ReplSetTest({
+        name: name,
+        nodes: [{}, {rsConfig: {arbiterOnly: true}}],
+    });
+
+    replSet.startSet();
+    replSet.initiate();
+    var primary = replSet.getPrimary();
+
+    var coll = primary.getDB('test').getCollection(name);
+    assert.writeOK(coll.insert({_id: 0, x: 1}));
+
+    // Add a secondary node but make it hang after retrieving the last op on the source
+    // but before copying databases.
+    var secondary = replSet.add();
+    secondary.setSlaveOk();
+
+    assert.commandWorked(secondary.getDB('admin').runCommand(
+        {configureFailPoint: 'initialSyncHangBeforeCopyingDatabases', mode: 'alwaysOn'}));
+    assert.commandWorked(secondary.getDB('admin').runCommand(
+        {configureFailPoint: 'initialSyncHangBeforeGettingMissingDocument', mode: 'alwaysOn'}));
+    replSet.reInitiate();
+
+    // Wait for fail point message to be logged.
+    checkLog.contains(secondary,
+                      'initial sync - initialSyncHangBeforeCopyingDatabases fail point enabled');
+
+    assert.writeOK(coll.update({_id: 0}, {x: 2}, {upsert: false, writeConcern: {w: 1}}));
+    assert.writeOK(coll.remove({_id: 0}, {justOne: true, writeConcern: {w: 1}}));
+
+    assert.commandWorked(secondary.getDB('admin').runCommand(
+        {configureFailPoint: 'initialSyncHangBeforeCopyingDatabases', mode: 'off'}));
+
+    checkLog.contains(secondary, 'update of non-mod failed');
+    checkLog.contains(secondary, 'Fetching missing document');
+
+    checkLog.contains(
+        secondary, 'initial sync - initialSyncHangBeforeGettingMissingDocument fail point enabled');
+    var doc = {_id: 0, x: 3};
+    // Re-insert deleted document.
+    assert.writeOK(coll.insert(doc, {writeConcern: {w: 1}}));
+    // Mark the collection as drop pending so it gets renamed, but retains the UUID.
+    assert.commandWorked(primary.getDB('test').runCommand({"drop": name}));
+
+    var res = assert.commandWorked(secondary.adminCommand({replSetGetStatus: 1, initialSync: 1}));
+    assert.eq(res.initialSyncStatus.fetchedMissingDocs, 0);
+    var firstOplogEnd = res.initialSyncStatus.initialSyncOplogEnd;
+
+    secondary.getDB('test').setLogLevel(1, 'replication');
+    assert.commandWorked(secondary.getDB('admin').runCommand(
+        {configureFailPoint: 'initialSyncHangBeforeGettingMissingDocument', mode: 'off'}));
+
+    checkLog.contains(secondary, 'Inserted missing document');
+    secondary.getDB('test').setLogLevel(0, 'replication');
+
+    checkLog.contains(secondary, 'initial sync done');
+
+    replSet.awaitReplication();
+    replSet.awaitSecondaryNodes();
+
+    replSet.stopSet();
+})();