diff options
Diffstat (limited to 'jstests/libs/override_methods/validate_collections_on_shutdown.js')
| -rw-r--r-- | jstests/libs/override_methods/validate_collections_on_shutdown.js | 190 |
1 files changed, 92 insertions, 98 deletions
diff --git a/jstests/libs/override_methods/validate_collections_on_shutdown.js b/jstests/libs/override_methods/validate_collections_on_shutdown.js index 49036790739..130404dfe6f 100644 --- a/jstests/libs/override_methods/validate_collections_on_shutdown.js +++ b/jstests/libs/override_methods/validate_collections_on_shutdown.js @@ -4,116 +4,110 @@ */ (function() { - "use strict"; +"use strict"; - load("jstests/libs/command_sequence_with_retries.js"); // for CommandSequenceWithRetries +load("jstests/libs/command_sequence_with_retries.js"); // for CommandSequenceWithRetries - MongoRunner.validateCollectionsCallback = function(port) { - if (jsTest.options().skipCollectionAndIndexValidation) { - print("Skipping collection validation during mongod shutdown"); - return; - } +MongoRunner.validateCollectionsCallback = function(port) { + if (jsTest.options().skipCollectionAndIndexValidation) { + print("Skipping collection validation during mongod shutdown"); + return; + } - let conn; - try { - conn = new Mongo("localhost:" + port); - } catch (e) { - print( - "Skipping collection validation because we couldn't establish a connection to the" + - " server on port " + port); - return; - } + let conn; + try { + conn = new Mongo("localhost:" + port); + } catch (e) { + print("Skipping collection validation because we couldn't establish a connection to the" + + " server on port " + port); + return; + } - // Set slaveOk=true so that we can run commands against any secondaries. - conn.setSlaveOk(); + // Set slaveOk=true so that we can run commands against any secondaries. + conn.setSlaveOk(); - let dbNames; - let result = - new CommandSequenceWithRetries(conn) - .then("running the isMaster command", - function(conn) { - const res = assert.commandWorked(conn.adminCommand({isMaster: 1})); - if (res.msg === "isdbgrid") { - return { - shouldStop: true, - reason: "not running validate against mongos" - }; - } else if (!res.ismaster && !res.secondary) { - return { - shouldStop: true, - reason: "not running validate since mongod isn't in the PRIMARY" + - " or SECONDARY states" - }; - } - }) - .then("authenticating", - function(conn) { - if (jsTest.options().keyFile) { - jsTest.authenticate(conn); - } - }) - .then( - "best effort to step down node forever", - function(conn) { - if (conn.isReplicaSetMember()) { - // This node should never run for election again. If the node has not - // been initialized yet, then it cannot get elected. - const kFreezeTimeSecs = 24 * 60 * 60; // 24 hours. + let dbNames; + let result = + new CommandSequenceWithRetries(conn) + .then("running the isMaster command", + function(conn) { + const res = assert.commandWorked(conn.adminCommand({isMaster: 1})); + if (res.msg === "isdbgrid") { + return {shouldStop: true, reason: "not running validate against mongos"}; + } else if (!res.ismaster && !res.secondary) { + return { + shouldStop: true, + reason: "not running validate since mongod isn't in the PRIMARY" + + " or SECONDARY states" + }; + } + }) + .then("authenticating", + function(conn) { + if (jsTest.options().keyFile) { + jsTest.authenticate(conn); + } + }) + .then("best effort to step down node forever", + function(conn) { + if (conn.isReplicaSetMember()) { + // This node should never run for election again. If the node has not + // been initialized yet, then it cannot get elected. + const kFreezeTimeSecs = 24 * 60 * 60; // 24 hours. - assert.commandWorkedOrFailedWithCode( - conn.adminCommand({replSetStepDown: kFreezeTimeSecs, force: true}), - [ + assert.commandWorkedOrFailedWithCode( + conn.adminCommand({replSetStepDown: kFreezeTimeSecs, force: true}), [ ErrorCodes.NotMaster, ErrorCodes.NotYetInitialized, ErrorCodes.Unauthorized - ]); + ]); - assert.commandWorkedOrFailedWithCode( - conn.adminCommand({replSetFreeze: kFreezeTimeSecs}), [ - ErrorCodes.NotYetInitialized, - ErrorCodes.Unauthorized, - // We include "NotSecondary" because if replSetStepDown receives - // "NotYetInitialized", then this command will fail with - // "NotSecondary". This is why this is a "best-effort". - ErrorCodes.NotSecondary - ]); - } - }) - .then("getting the list of databases", - function(conn) { - const res = conn.adminCommand({listDatabases: 1}); - if (!res.ok) { - // TODO: SERVER-31916 for the KeyNotFound error - assert.commandFailedWithCode( - res, [ErrorCodes.Unauthorized, ErrorCodes.KeyNotFound]); - return {shouldStop: true, reason: "cannot run listDatabases"}; - } - assert.commandWorked(res); - dbNames = res.databases.map(dbInfo => dbInfo.name); - }) - .execute(); + assert.commandWorkedOrFailedWithCode( + conn.adminCommand({replSetFreeze: kFreezeTimeSecs}), [ + ErrorCodes.NotYetInitialized, + ErrorCodes.Unauthorized, + // We include "NotSecondary" because if replSetStepDown receives + // "NotYetInitialized", then this command will fail with + // "NotSecondary". This is why this is a "best-effort". + ErrorCodes.NotSecondary + ]); + } + }) + .then("getting the list of databases", + function(conn) { + const res = conn.adminCommand({listDatabases: 1}); + if (!res.ok) { + // TODO: SERVER-31916 for the KeyNotFound error + assert.commandFailedWithCode( + res, [ErrorCodes.Unauthorized, ErrorCodes.KeyNotFound]); + return {shouldStop: true, reason: "cannot run listDatabases"}; + } + assert.commandWorked(res); + dbNames = res.databases.map(dbInfo => dbInfo.name); + }) + .execute(); - if (!result.ok) { - print("Skipping collection validation: " + result.msg); - return; - } + if (!result.ok) { + print("Skipping collection validation: " + result.msg); + return; + } - load('jstests/hooks/validate_collections.js'); // for validateCollections + load('jstests/hooks/validate_collections.js'); // for validateCollections - const cmds = new CommandSequenceWithRetries(conn); - for (let i = 0; i < dbNames.length; ++i) { - const dbName = dbNames[i]; - cmds.then("validating " + dbName, function(conn) { - const validate_res = validateCollections(conn.getDB(dbName), {full: true}); - if (!validate_res.ok) { - return { - shouldStop: true, - reason: "collection validation failed " + tojson(validate_res) - }; - } - }); - } + const cmds = new CommandSequenceWithRetries(conn); + for (let i = 0; i < dbNames.length; ++i) { + const dbName = dbNames[i]; + cmds.then("validating " + dbName, function(conn) { + const validate_res = validateCollections(conn.getDB(dbName), {full: true}); + if (!validate_res.ok) { + return { + shouldStop: true, + reason: "collection validation failed " + tojson(validate_res) + }; + } + }); + } - assert.commandWorked(cmds.execute()); - }; + assert.commandWorked(cmds.execute()); +}; })(); |