diff options
Diffstat (limited to 'src/mongo/s/catalog_cache_test_fixture.h')
-rw-r--r-- | src/mongo/s/catalog_cache_test_fixture.h | 10 |
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}; }; |