SERVER-22105 run the validation hook on secondary nodes

1 files changed, 26 insertions, 5 deletions

diff --git a/jstests/hooks/run_validate_collections.js b/jstests/hooks/run_validate_collections.js
index 1e910b3e6cf..28198136774 100644
--- a/jstests/hooks/run_validate_collections.js
+++ b/jstests/hooks/run_validate_collections.js
@@ -6,11 +6,32 @@
     assert.eq(typeof db, 'object', 'Invalid `db` object, is the shell connected to a mongod?');
     load('jstests/hooks/validate_collections.js');  // For validateCollections
 
-    var dbNames = db.getMongo().getDBNames();
+    var serverList = [];
+    serverList.push(db.getMongo());
 
-    for (var dbName of dbNames) {
-        if (!validateCollections(db.getSiblingDB(dbName), {full: true})) {
-            throw new Error('Collection validation failed');
+    var addSecondaryNodes = function() {
+        var cmdLineOpts = db.adminCommand('getCmdLineOpts');
+        assert.commandWorked(cmdLineOpts);
+
+        if (cmdLineOpts.parsed.hasOwnProperty('replication') &&
+            cmdLineOpts.parsed.replication.hasOwnProperty('replSet')) {
+            var rst = new ReplSetTest(db.getMongo().host);
+            // Call getPrimary to populate rst with information about the nodes.
+            var primary = rst.getPrimary();
+            assert(primary, 'calling getPrimary() failed');
+            serverList.push(...rst.getSecondaries());
+        }
+    };
+
+    addSecondaryNodes();
+
+    for (var server of serverList) {
+        print('Running validate() on ' + server.host);
+        var dbNames = server.getDBNames();
+        for (var dbName of dbNames) {
+            if (!validateCollections(db.getSiblingDB(dbName), {full: true})) {
+                throw new Error('Collection validation failed');
+            }
         }
     }
-})();
\ No newline at end of file
+})();