SERVER-36113 Add ability to perform $exchange when shard key is hashed

1 files changed, 15 insertions, 1 deletions

diff --git a/src/mongo/s/catalog_cache_test_fixture.cpp b/src/mongo/s/catalog_cache_test_fixture.cpp
index c7df14e96b1..cc53b998540 100644
--- a/src/mongo/s/catalog_cache_test_fixture.cpp
+++ b/src/mongo/s/catalog_cache_test_fixture.cpp
@@ -183,8 +183,22 @@ void CatalogCacheTestFixture::expectGetCollection(NamespaceString nss,
 
 CachedCollectionRoutingInfo CatalogCacheTestFixture::loadRoutingTableWithTwoChunksAndTwoShards(
     NamespaceString nss) {
+
+    return loadRoutingTableWithTwoChunksAndTwoShardsImpl(nss, BSON("_id" << 1));
+}
+
+CachedCollectionRoutingInfo CatalogCacheTestFixture::loadRoutingTableWithTwoChunksAndTwoShardsHash(
+    NamespaceString nss) {
+
+    return loadRoutingTableWithTwoChunksAndTwoShardsImpl(nss,
+                                                         BSON("_id"
+                                                              << "hashed"));
+}
+
+CachedCollectionRoutingInfo CatalogCacheTestFixture::loadRoutingTableWithTwoChunksAndTwoShardsImpl(
+    NamespaceString nss, const BSONObj& shardKey) {
     const OID epoch = OID::gen();
-    const ShardKeyPattern shardKeyPattern(BSON("_id" << 1));
+    const ShardKeyPattern shardKeyPattern(shardKey);
 
     auto future = scheduleRoutingInfoRefresh(nss);