1 files changed, 5 insertions, 0 deletions
diff --git a/src/mongo/db/query/query_test_service_context.h b/src/mongo/db/query/query_test_service_context.h
index 7466116d354..658b73ef36a 100644
--- a/src/mongo/db/query/query_test_service_context.h
+++ b/src/mongo/db/query/query_test_service_context.h
@@ -29,6 +29,7 @@
 #pragma once
 
 #include "mongo/db/client.h"
+#include "mongo/db/logical_session_id.h"
 #include "mongo/db/service_context_noop.h"
 
 namespace mongo {
@@ -44,6 +45,10 @@ public:
 
     ServiceContext::UniqueOperationContext makeOperationContext();
 
+    ServiceContext::UniqueOperationContext makeOperationContext(LogicalSessionId lsid);
+
+    Client* getClient() const;
+
 private:
     ServiceContextNoop _serviceContext;
     ServiceContext::UniqueClient _uniqueClient;