diff options
author | Marcos José Grillo Ramírez <marcos.grillo@mongodb.com> | 2021-02-04 14:02:05 +0100 |
---|---|---|
committer | Evergreen Agent <no-reply@evergreen.mongodb.com> | 2021-02-05 02:11:18 +0000 |
commit | 83da96e969cd2b06677e441b2cb6e9cde229f026 (patch) | |
tree | a93f94840da8a737b272c142ffff1268f728b1c2 /src/mongo/db/s/config/sharding_catalog_manager_shard_collection_test.cpp | |
parent | 915296e7f09ee9934a5d6794bc70f3c78104f81e (diff) | |
download | mongo-83da96e969cd2b06677e441b2cb6e9cde229f026.tar.gz |
SERVER-52779 Added new coordinator to create collection with the new shard collection path
Diffstat (limited to 'src/mongo/db/s/config/sharding_catalog_manager_shard_collection_test.cpp')
-rw-r--r-- | src/mongo/db/s/config/sharding_catalog_manager_shard_collection_test.cpp | 80 |
1 files changed, 55 insertions, 25 deletions
diff --git a/src/mongo/db/s/config/sharding_catalog_manager_shard_collection_test.cpp b/src/mongo/db/s/config/sharding_catalog_manager_shard_collection_test.cpp index 85067a04d22..c3871c2a109 100644 --- a/src/mongo/db/s/config/sharding_catalog_manager_shard_collection_test.cpp +++ b/src/mongo/db/s/config/sharding_catalog_manager_shard_collection_test.cpp @@ -46,6 +46,7 @@ #include "mongo/s/catalog/type_shard.h" #include "mongo/s/client/shard_registry.h" #include "mongo/s/grid.h" +#include "mongo/s/request_types/shard_collection_gen.h" #include "mongo/s/shard_key_pattern.h" #include "mongo/util/scopeguard.h" #include "mongo/util/time_support.h" @@ -118,7 +119,9 @@ TEST_F(CreateFirstChunksTest, Split_Disallowed_With_Both_SplitPoints_And_Zones) ASSERT_THROWS_CODE( InitialSplitPolicy::calculateOptimizationStrategy(operationContext(), kShardKeyPattern, - request, + request.getNumInitialChunks(), + request.getPresplitHashedZones(), + request.getInitialSplitPoints(), tags, 2 /* numShards */, true /* collectionIsEmpty */), @@ -128,7 +131,9 @@ TEST_F(CreateFirstChunksTest, Split_Disallowed_With_Both_SplitPoints_And_Zones) ASSERT_THROWS_CODE( InitialSplitPolicy::calculateOptimizationStrategy(operationContext(), kShardKeyPattern, - request, + request.getNumInitialChunks(), + request.getPresplitHashedZones(), + request.getInitialSplitPoints(), tags, 2 /* numShards */, false /* collectionIsEmpty */), @@ -160,7 +165,9 @@ TEST_F(CreateFirstChunksTest, NonEmptyCollection_SplitPoints_FromSplitVector_Man auto optimization = InitialSplitPolicy::calculateOptimizationStrategy(operationContext(), kShardKeyPattern, - request, + request.getNumInitialChunks(), + request.getPresplitHashedZones(), + request.getInitialSplitPoints(), {}, /* tags */ 3 /* numShards */, false /* collectionIsEmpty */); @@ -203,12 +210,15 @@ TEST_F(CreateFirstChunksTest, NonEmptyCollection_SplitPoints_FromClient_ManyChun ShardsvrShardCollectionRequest request; request.setInitialSplitPoints(splitPoints); - auto optimization = InitialSplitPolicy::calculateOptimizationStrategy(operationContext(), - kShardKeyPattern, - request, - zones, - 3 /* numShards */, - collectionIsEmpty); + auto optimization = + InitialSplitPolicy::calculateOptimizationStrategy(operationContext(), + kShardKeyPattern, + request.getNumInitialChunks(), + request.getPresplitHashedZones(), + request.getInitialSplitPoints(), + zones, + 3 /* numShards */, + collectionIsEmpty); ASSERT(optimization->isOptimized()); @@ -236,8 +246,15 @@ TEST_F(CreateFirstChunksTest, NonEmptyCollection_WithZones_OneChunkToPrimary) { bool collectionIsEmpty = false; ShardsvrShardCollectionRequest request; - auto optimization = InitialSplitPolicy::calculateOptimizationStrategy( - operationContext(), kShardKeyPattern, request, zones, 3 /* numShards */, collectionIsEmpty); + auto optimization = + InitialSplitPolicy::calculateOptimizationStrategy(operationContext(), + kShardKeyPattern, + request.getNumInitialChunks(), + request.getPresplitHashedZones(), + request.getInitialSplitPoints(), + zones, + 3 /* numShards */, + collectionIsEmpty); ASSERT(optimization->isOptimized()); @@ -275,12 +292,15 @@ TEST_F(CreateFirstChunksTest, EmptyCollection_SplitPoints_FromClient_ManyChunksD ShardsvrShardCollectionRequest request; request.setInitialSplitPoints(splitPoints); - auto optimization = InitialSplitPolicy::calculateOptimizationStrategy(operationContext(), - kShardKeyPattern, - request, - zones, - 3 /* numShards */, - collectionIsEmpty); + auto optimization = + InitialSplitPolicy::calculateOptimizationStrategy(operationContext(), + kShardKeyPattern, + request.getNumInitialChunks(), + request.getPresplitHashedZones(), + request.getInitialSplitPoints(), + zones, + 3 /* numShards */, + collectionIsEmpty); ASSERT(optimization->isOptimized()); @@ -322,12 +342,15 @@ TEST_F(CreateFirstChunksTest, EmptyCollection_NoSplitPoints_OneChunkToPrimary) { ShardsvrShardCollectionRequest request; request.setInitialSplitPoints(splitPoints); - auto optimization = InitialSplitPolicy::calculateOptimizationStrategy(operationContext(), - kShardKeyPattern, - request, - zones, - 3 /* numShards */, - collectionIsEmpty); + auto optimization = + InitialSplitPolicy::calculateOptimizationStrategy(operationContext(), + kShardKeyPattern, + request.getNumInitialChunks(), + request.getPresplitHashedZones(), + request.getInitialSplitPoints(), + zones, + 3 /* numShards */, + collectionIsEmpty); ASSERT(optimization->isOptimized()); @@ -355,8 +378,15 @@ TEST_F(CreateFirstChunksTest, EmptyCollection_WithZones_ManyChunksOnFirstZoneSha ChunkRange(kShardKeyPattern.getKeyPattern().globalMin(), BSON("x" << 0)))}; bool collectionIsEmpty = true; ShardsvrShardCollectionRequest request; - auto optimization = InitialSplitPolicy::calculateOptimizationStrategy( - operationContext(), kShardKeyPattern, request, zones, 3 /* numShards */, collectionIsEmpty); + auto optimization = + InitialSplitPolicy::calculateOptimizationStrategy(operationContext(), + kShardKeyPattern, + request.getNumInitialChunks(), + request.getPresplitHashedZones(), + request.getInitialSplitPoints(), + zones, + 3 /* numShards */, + collectionIsEmpty); ASSERT(optimization->isOptimized()); |