1 files changed, 104 insertions, 0 deletions
diff --git a/src/mongo/gotools/test/qa-tests/jstests/restore/users_and_roles_temp_collections.js b/src/mongo/gotools/test/qa-tests/jstests/restore/users_and_roles_temp_collections.js
new file mode 100644
index 00000000000..fdbf236e8f8
--- /dev/null
+++ b/src/mongo/gotools/test/qa-tests/jstests/restore/users_and_roles_temp_collections.js
@@ -0,0 +1,104 @@
+(function() {
+
+  if (typeof getToolTest === 'undefined') {
+    load('jstests/configs/plain_28.config.js');
+  }
+
+  // Tests running mongorestore with --restoreDbUsersAndRoles
+
+  jsTest.log('Testing running mongorestore with --restoreDbUsersAndRoles');
+
+  var toolTest = getToolTest('users_and_roles_temp_collections');
+  var commonToolArgs = getCommonToolArguments();
+
+  // where we'll put the dump
+  var dumpTarget = 'users_and_roles_temp_collections_dump';
+  resetDbpath(dumpTarget);
+
+  // the db we'll be using
+  var testDB = toolTest.db.getSiblingDB('test');
+
+  // create some users and roles on the database
+  testDB.createUser({
+    user: 'userOne',
+    pwd: 'pwdOne',
+    roles: [{role: 'read', db: 'test'}],
+  });
+  testDB.createRole({
+    role: 'roleOne',
+    privileges: [{
+      resource: {db: 'test', collection: ''},
+      actions: ['find'],
+    }],
+    roles: [],
+  });
+  testDB.createUser({
+    user: 'userTwo',
+    pwd: 'pwdTwo',
+    roles: [{role: 'roleOne', db: 'test'}],
+  });
+
+  // insert some data
+  for (var i = 0; i < 10; i++) {
+    testDB.data.insert({_id: i});
+  }
+  // sanity check the insertion worked
+  assert.eq(10, testDB.data.count());
+
+  // dump the data
+  var ret = toolTest.runTool.apply(toolTest, ['dump', '--db', 'test', '--dumpDbUsersAndRoles']
+    .concat(getDumpTarget(dumpTarget))
+    .concat(commonToolArgs));
+  assert.eq(0, ret);
+
+  // drop the database, users, and roles
+  testDB.dropDatabase();
+  testDB.dropAllUsers();
+  testDB.dropAllRoles();
+
+  // insert to the default temp collections so we hit them later
+  var adminDB = toolTest.db.getSiblingDB('admin');
+  adminDB.tempusers.insert({_id: 1});
+  adminDB.temproles.insert({_id: 1});
+
+  // try to restore the data
+  ret = toolTest.runTool.apply(toolTest, ['restore', '--db', 'test', '--restoreDbUsersAndRoles']
+    .concat(getRestoreTarget(dumpTarget+'/test'))
+    .concat(commonToolArgs));
+
+  // we should succeed with default temp collections
+  assert.eq(0, ret);
+
+  // try to restore the data with new temp collections
+  ret = toolTest.runTool.apply(toolTest, ['restore',
+      '--db', 'test',
+      '--tempUsersColl', 'tempU',
+      '--tempRolesColl', 'tempR',
+      '--restoreDbUsersAndRoles']
+    .concat(getRestoreTarget(dumpTarget+'/test'))
+    .concat(commonToolArgs));
+
+  // we should succeed with new temp collections
+  assert.eq(0, ret);
+
+  // make sure the data was restored
+  assert.eq(10, testDB.data.count());
+  for (i = 0; i < 10; i++) {
+    assert.eq(1, testDB.data.count({_id: i}));
+  }
+
+  // make sure the users were restored
+  var users = testDB.getUsers();
+  assert.eq(2, users.length);
+  assert(users[0].user === 'userOne' || users[1].user === 'userOne');
+  assert(users[0].user === 'userTwo' || users[1].user === 'userTwo');
+
+  // make sure the role was restored
+  var roles = testDB.getRoles();
+  assert.eq(1, roles.length);
+  assert.eq('roleOne', roles[0].role);
+
+  // success
+  toolTest.stop();
+
+}());