SERVER-35179 Make NiTL actually refresh

1 files changed, 4 insertions, 0 deletions

diff --git a/src/mongo/executor/connection_pool_test_fixture.cpp b/src/mongo/executor/connection_pool_test_fixture.cpp
index 1d42d0445d1..560f184a025 100644
--- a/src/mongo/executor/connection_pool_test_fixture.cpp
+++ b/src/mongo/executor/connection_pool_test_fixture.cpp
@@ -122,6 +122,7 @@ void ConnectionImpl::pushSetup(PushSetupCallback status) {
         _pushSetupQueue.pop_front();
 
         auto cb = connPtr->_setupCallback;
+        connPtr->indicateUsed();
         cb(connPtr, callback());
     }
 }
@@ -145,6 +146,7 @@ void ConnectionImpl::pushRefresh(PushRefreshCallback status) {
         _pushRefreshQueue.pop_front();
 
         auto cb = connPtr->_refreshCallback;
+        connPtr->indicateUsed();
         cb(connPtr, callback());
     }
 }
@@ -189,6 +191,7 @@ void ConnectionImpl::setup(Milliseconds timeout, SetupCallback cb) {
         _pushSetupQueue.pop_front();
 
         auto refreshCb = connPtr->_setupCallback;
+        connPtr->indicateUsed();
         refreshCb(connPtr, callback());
     }
 }
@@ -210,6 +213,7 @@ void ConnectionImpl::refresh(Milliseconds timeout, RefreshCallback cb) {
         _pushRefreshQueue.pop_front();
 
         auto refreshCb = connPtr->_refreshCallback;
+        connPtr->indicateUsed();
         refreshCb(connPtr, callback());
     }
 }