SERVER-27741 fixed race in DataReplicatorTest::DataReplicatorResetsOnCompletionCallbackFunctionPointerUponCompletion

(cherry picked from commit 6a2af712f71774dae250c4cbe5f8ac695cfec3ea)

1 files changed, 7 insertions, 0 deletions

diff --git a/src/mongo/db/repl/data_replicator_test.cpp b/src/mongo/db/repl/data_replicator_test.cpp
index 524c95125b7..7fec2ca9862 100644
--- a/src/mongo/db/repl/data_replicator_test.cpp
+++ b/src/mongo/db/repl/data_replicator_test.cpp
@@ -833,6 +833,13 @@ TEST_F(DataReplicatorTest, DataReplicatorResetsOnCompletionCallbackFunctionPoint
     ASSERT_FALSE(sharedCallbackStateDestroyed);
 
     ASSERT_OK(dr->shutdown());
+
+    // Depending on which DataReplicator stage (_chooseSyncSource or _rollbackCheckerResetCallback)
+    // was interrupted by shutdown(), we may have to request the network interface to deliver
+    // cancellation signals to the DataReplicator callbacks in for DataReplicator to run to
+    // completion.
+    executor::NetworkInterfaceMock::InNetworkGuard(getNet())->runReadyNetworkOperations();
+
     dr->join();
 
     ASSERT_EQUALS(ErrorCodes::CallbackCanceled, lastApplied);