1 files changed, 44 insertions, 0 deletions
diff --git a/src/mongo/gotools/test/qa-tests/jstests/restore/objcheck_valid_bson.js b/src/mongo/gotools/test/qa-tests/jstests/restore/objcheck_valid_bson.js
new file mode 100644
index 00000000000..60ed695b1cf
--- /dev/null
+++ b/src/mongo/gotools/test/qa-tests/jstests/restore/objcheck_valid_bson.js
@@ -0,0 +1,44 @@
+(function() {
+
+  load("jstests/configs/standard_dump_targets.config.js");
+  // Tests that running mongorestore with --objcheck on valid bson
+  // files restores the data successfully.
+
+  jsTest.log('Testing restoration with --objcheck');
+
+  var toolTest = new ToolTest('objcheck_valid_bson');
+  toolTest.startDB('foo');
+
+  // where we'll put the dump
+  var dumpTarget = 'objcheck_valid_bson_dump';
+  resetDbpath(dumpTarget);
+
+  // the db and collection we will use
+  var testDB = toolTest.db.getSiblingDB('test');
+  var testColl = testDB.coll;
+
+  // insert some data
+  for (var i = 0; i < 50; i++) {
+    testColl.insert({_id: i});
+  }
+  // sanity check the insert worked
+  assert.eq(50, testColl.count());
+
+  // dump the data
+  var ret = toolTest.runTool.apply(toolTest, ['dump'].concat(getDumpTarget(dumpTarget)));
+  assert.eq(0, ret);
+
+  // drop the data
+  testDB.dropDatabase();
+
+  // restore the data, with --objcheck
+  ret = toolTest.runTool.apply(toolTest, ['restore'].concat(getRestoreTarget(dumpTarget)));
+  assert.eq(0, ret);
+
+  // make sure the restore completed succesfully
+  assert.eq(50, testColl.count());
+
+  // success
+  toolTest.stop();
+
+}());