1 files changed, 67 insertions, 0 deletions
diff --git a/src/mongo/gotools/test/qa-tests/jstests/restore/oplog_replay_local_rs.js b/src/mongo/gotools/test/qa-tests/jstests/restore/oplog_replay_local_rs.js
new file mode 100644
index 00000000000..03a16c4a745
--- /dev/null
+++ b/src/mongo/gotools/test/qa-tests/jstests/restore/oplog_replay_local_rs.js
@@ -0,0 +1,67 @@
+/**
+ * oplog_replay_local_rs.js
+ *
+ * This file tests mongorestore with --oplogReplay where the oplog file is in the 'oplog.rs'
+ * collection of the 'local' database. This occurs when using a replica-set for replication.
+ */
+(function() {
+  'use strict';
+  if (typeof getToolTest === 'undefined') {
+    load('jstests/configs/plain_28.config.js');
+  }
+
+  var commonToolArgs = getCommonToolArguments();
+  var dumpTarget = 'oplog_replay_local_rs';
+
+  var toolTest = getToolTest('oplog_replay_local_rs');
+
+  // Set the test db to 'local' and collection to 'oplog.rs' to fake a replica set oplog
+  var testDB = toolTest.db.getSiblingDB('local');
+  var testColl = testDB['oplog.rs'];
+  var testRestoreDB = toolTest.db.getSiblingDB('test');
+  var testRestoreColl = testRestoreDB.op;
+  resetDbpath(dumpTarget);
+
+  var oplogSize = 100;
+  testDB.createCollection('oplog.rs', {capped: true, size: 100000});
+
+  // Create a fake oplog consisting of 100 inserts.
+  for (var i = 0; i < oplogSize; i++) {
+    var r = testColl.insert({
+      ts: new Timestamp(0, i),
+      op: "i",
+      o: {_id: i, x: 'a' + i},
+      ns: "test.op",
+    });
+    assert.eq(1, r.nInserted, "insert failed");
+  }
+
+  // Dump the fake oplog.
+  var ret = toolTest.runTool.apply(toolTest, ['dump',
+      '--db', 'local',
+      '-c', 'oplog.rs',
+      '--out', dumpTarget]
+    .concat(commonToolArgs));
+  assert.eq(0, ret, "dump operation failed");
+
+  // Dump original data.
+  testColl.drop();
+  assert.eq(0, testColl.count(), "all original entries should be dropped");
+
+
+  // Create the test.op collection.
+  testRestoreColl.drop();
+  testRestoreDB.createCollection("op");
+  assert.eq(0, testRestoreColl.count());
+
+  // Replay the oplog from the provided oplog
+  ret = toolTest.runTool.apply(toolTest, ['restore',
+      '--oplogReplay',
+      dumpTarget]
+    .concat(commonToolArgs));
+  assert.eq(0, ret, "restore operation failed");
+
+  assert.eq(oplogSize, testRestoreColl.count(),
+      "all oplog entries should be inserted");
+  toolTest.stop();
+}());