1 files changed, 48 insertions, 0 deletions
diff --git a/src/mongo/gotools/test/qa-tests/jstests/import/replset.js b/src/mongo/gotools/test/qa-tests/jstests/import/replset.js
new file mode 100644
index 00000000000..3ff32650d07
--- /dev/null
+++ b/src/mongo/gotools/test/qa-tests/jstests/import/replset.js
@@ -0,0 +1,48 @@
+(function() {
+  jsTest.log('Testing running import with upserts');
+
+  var toolTest = new ToolTest('import_repl');
+
+  var replset1 = new ReplSetTest({nodes: 3, name: 'importtest'});
+  replset1.startSet();
+  replset1.initiate();
+
+  var primary = replset1.getPrimary();
+  var secondary = replset1.getSecondary();
+
+  var db = primary.getDB('import_repl_test');
+
+  // trying to write to the secondary should fail
+  assert.neq(runMongoProgram.apply(this, ['mongoimport',
+        '--file', 'jstests/import/testdata/basic.json',
+        '--db', db.getName(),
+        '--collection', db.c.getName(),
+        '--host', secondary.host]), 0,
+      "writing to secondary should fail");
+
+  assert.eq(db.c.count(), 0, 'database not empty');
+
+  // now import using the primary
+  assert.eq(runMongoProgram.apply(this, ['mongoimport',
+        '--file', 'jstests/import/testdata/basic.json',
+        '--db', db.getName(),
+        '--collection', db.c.getName(),
+        '--host', primary.host]), 0,
+      "writing to primary should succeed");
+
+  assert.neq(db.c.count(), 0, 'database unexpectedly empty on primary');
+
+  db.dropDatabase();
+
+  // import using the secondary but include replset name, should succeed
+  assert.eq(runMongoProgram.apply(this, ['mongoimport',
+        '--file', 'jstests/import/testdata/basic.json',
+        '--db', db.getName(),
+        '--collection', db.c.getName(),
+        '--host', replset1.name + "/" + secondary.host]), 0,
+      "writing to secondary with replset name should succeed");
+
+  assert.neq(db.c.count(), 0, 'database unexpectedly empty on secondary');
+
+  toolTest.stop();
+}());