2 files changed, 6 insertions, 2 deletions
diff --git a/src/mongo/db/query/query_test_service_context.cpp b/src/mongo/db/query/query_test_service_context.cpp
index d752db15664..407a0341792 100644
--- a/src/mongo/db/query/query_test_service_context.cpp
+++ b/src/mongo/db/query/query_test_service_context.cpp
@@ -46,9 +46,12 @@ ServiceContext::UniqueOperationContext QueryTestServiceContext::makeOperationCon
 }
 
 ServiceContext::UniqueOperationContext QueryTestServiceContext::makeOperationContext(
-    LogicalSessionId lsid) {
+    LogicalSessionId lsid, boost::optional<TxnNumber> txnNumber) {
     auto opCtx = makeOperationContext();
     opCtx->setLogicalSessionId(lsid);
+    if (txnNumber) {
+        opCtx->setTxnNumber(*txnNumber);
+    }
     return opCtx;
 }
 
diff --git a/src/mongo/db/query/query_test_service_context.h b/src/mongo/db/query/query_test_service_context.h
index c52944cf2f5..bbf265a06ac 100644
--- a/src/mongo/db/query/query_test_service_context.h
+++ b/src/mongo/db/query/query_test_service_context.h
@@ -45,7 +45,8 @@ public:
 
     ServiceContext::UniqueOperationContext makeOperationContext();
 
-    ServiceContext::UniqueOperationContext makeOperationContext(LogicalSessionId lsid);
+    ServiceContext::UniqueOperationContext makeOperationContext(
+        LogicalSessionId lsid, boost::optional<TxnNumber> txnNumber);
 
     Client* getClient() const;