1 files changed, 8 insertions, 0 deletions
diff --git a/src/mongo/s/query/results_merger_test_fixture.h b/src/mongo/s/query/results_merger_test_fixture.h
index 7b714c568ae..7a0f46c9058 100644
--- a/src/mongo/s/query/results_merger_test_fixture.h
+++ b/src/mongo/s/query/results_merger_test_fixture.h
@@ -32,6 +32,7 @@
 
 #include "mongo/s/query/async_results_merger.h"
 #include "mongo/s/sharding_router_test_fixture.h"
+#include "mongo/util/clock_source_mock.h"
 
 namespace mongo {
 
@@ -241,6 +242,13 @@ protected:
         remoteCursor.setCursorResponse(std::move(response));
         return remoteCursor;
     }
+
+    ClockSourceMock* getMockClockSource() {
+        ClockSourceMock* mockClock = dynamic_cast<ClockSourceMock*>(
+            operationContext()->getServiceContext()->getPreciseClockSource());
+        invariant(mockClock);
+        return mockClock;
+    }
 };
 
 }  // namespace mongo