SERVER-54120 Move away from getLog to check if a command is done in our tests

1 files changed, 12 insertions, 0 deletions

diff --git a/src/mongo/db/curop.cpp b/src/mongo/db/curop.cpp
index a0e60fa3d1a..3898a2f98ab 100644
--- a/src/mongo/db/curop.cpp
+++ b/src/mongo/db/curop.cpp
@@ -599,6 +599,10 @@ bool CurOp::completeAndLogOperation(OperationContext* opCtx,
 // Failpoints after commands are logged.
 constexpr auto kPrepareTransactionCmdName = "prepareTransaction"_sd;
 MONGO_FAIL_POINT_DEFINE(waitForPrepareTransactionCommandLogged);
+constexpr auto kHelloCmdName = "hello"_sd;
+MONGO_FAIL_POINT_DEFINE(waitForHelloCommandLogged);
+constexpr auto kIsMasterCmdName = "isMaster"_sd;
+MONGO_FAIL_POINT_DEFINE(waitForIsMasterCommandLogged);
 
 void CurOp::_checkForFailpointsAfterCommandLogged() {
     if (!isCommand() || !getCommand()) {
@@ -610,6 +614,14 @@ void CurOp::_checkForFailpointsAfterCommandLogged() {
         if (MONGO_unlikely(waitForPrepareTransactionCommandLogged.shouldFail())) {
             LOGV2(31481, "waitForPrepareTransactionCommandLogged failpoint enabled");
         }
+    } else if (cmdName == kHelloCmdName) {
+        if (MONGO_unlikely(waitForHelloCommandLogged.shouldFail())) {
+            LOGV2(31482, "waitForHelloCommandLogged failpoint enabled");
+        }
+    } else if (cmdName == kIsMasterCmdName) {
+        if (MONGO_unlikely(waitForIsMasterCommandLogged.shouldFail())) {
+            LOGV2(31483, "waitForIsMasterCommandLogged failpoint enabled");
+        }
     }
 }