SERVER-23610 CanonicalQuery should own a CollatorInterface

1 files changed, 5 insertions, 1 deletions

diff --git a/src/mongo/s/shard_key_pattern_test.cpp b/src/mongo/s/shard_key_pattern_test.cpp
index 57cb7bca521..bb54fc289b5 100644
--- a/src/mongo/s/shard_key_pattern_test.cpp
+++ b/src/mongo/s/shard_key_pattern_test.cpp
@@ -29,6 +29,7 @@
 
 #include "mongo/db/hasher.h"
 #include "mongo/db/json.h"
+#include "mongo/db/query/query_test_service_context.h"
 #include "mongo/unittest/unittest.h"
 
 namespace {
@@ -249,7 +250,10 @@ TEST(ShardKeyPattern, ExtractDocShardKeyHashed) {
 }
 
 static BSONObj queryKey(const ShardKeyPattern& pattern, const BSONObj& query) {
-    StatusWith<BSONObj> status = pattern.extractShardKeyFromQuery(query);
+    QueryTestServiceContext serviceContext;
+    auto txn = serviceContext.makeOperationContext();
+
+    StatusWith<BSONObj> status = pattern.extractShardKeyFromQuery(txn.get(), query);
     if (!status.isOK())
         return BSONObj();
     return status.getValue();