1 files changed, 7 insertions, 3 deletions
diff --git a/src/mongo/s/catalog_cache_test_fixture.h b/src/mongo/s/catalog_cache_test_fixture.h
index 3d58f6a8557..8886c1e74bb 100644
--- a/src/mongo/s/catalog_cache_test_fixture.h
+++ b/src/mongo/s/catalog_cache_test_fixture.h
@@ -116,8 +116,11 @@ protected:
     /**
      * The common implementation for any shard key.
      */
-    ChunkManager loadRoutingTableWithTwoChunksAndTwoShardsImpl(NamespaceString nss,
-                                                               const BSONObj& shardKey);
+    ChunkManager loadRoutingTableWithTwoChunksAndTwoShardsImpl(
+        NamespaceString nss,
+        const BSONObj& shardKey,
+        boost::optional<std::string> primaryShardId = boost::none,
+        boost::optional<UUID> uuid = boost::none);
 
     /**
      * Mocks network responses for loading a sharded database and collection from the config server.
@@ -125,7 +128,8 @@ protected:
     void expectGetDatabase(NamespaceString nss, std::string primaryShard = "0");
     void expectGetCollection(NamespaceString nss,
                              OID epoch,
-                             const ShardKeyPattern& shardKeyPattern);
+                             const ShardKeyPattern& shardKeyPattern,
+                             boost::optional<UUID> uuid = boost::none);
 
     const HostAndPort kConfigHostAndPort{"DummyConfig", 1234};
 };