diff options
Diffstat (limited to 'src/mongo/db/s/balancer/balancer_commands_scheduler_test.cpp')
| -rw-r--r-- | src/mongo/db/s/balancer/balancer_commands_scheduler_test.cpp | 21 |
1 files changed, 12 insertions, 9 deletions
diff --git a/src/mongo/db/s/balancer/balancer_commands_scheduler_test.cpp b/src/mongo/db/s/balancer/balancer_commands_scheduler_test.cpp index aab01274138..fdf01b9eebd 100644 --- a/src/mongo/db/s/balancer/balancer_commands_scheduler_test.cpp +++ b/src/mongo/db/s/balancer/balancer_commands_scheduler_test.cpp @@ -319,13 +319,15 @@ TEST_F(BalancerCommandsSchedulerTest, SuccessfulRequestChunkDataSizeCommand) { _scheduler.start(operationContext(), getMigrationRecoveryDefaultValues()); ChunkType chunk = makeChunk(0, kShardId0); - auto futureResponse = _scheduler.requestDataSize(operationContext(), - kNss, - chunk.getShard(), - chunk.getRange(), - chunk.getVersion(), - KeyPattern(BSON("x" << 1)), - false /* issuedByRemoteUser */); + auto futureResponse = + _scheduler.requestDataSize(operationContext(), + kNss, + chunk.getShard(), + chunk.getRange(), + chunk.getVersion(), + KeyPattern(BSON("x" << 1)), + false /* issuedByRemoteUser */, + (kDefaultMaxChunkSizeBytes / 100) * 25 /* maxSize */); auto swReceivedDataSize = futureResponse.getNoThrow(); ASSERT_OK(swReceivedDataSize.getStatus()); auto receivedDataSize = swReceivedDataSize.getValue(); @@ -434,11 +436,12 @@ TEST_F(BalancerCommandsSchedulerTest, MoveChunkCommandGetsPersistedOnDiskWhenReq ASSERT_EQ(kNss, recoveredCommand->getNameSpace()); ASSERT_EQ(migrateInfo.from, recoveredCommand->getTarget()); ASSERT_TRUE(recoveredCommand->requiresDistributedLock()); + MoveChunkCommandInfo originalCommandInfo(migrateInfo.nss, migrateInfo.from, migrateInfo.to, migrateInfo.minKey, - migrateInfo.maxKey, + *migrateInfo.maxKey, requestSettings.maxChunkSizeBytes, requestSettings.secondaryThrottle, requestSettings.waitForDelete, @@ -464,7 +467,7 @@ TEST_F(BalancerCommandsSchedulerTest, PersistedCommandsAreReissuedWhenRecovering auto requestSettings = getMoveChunkSettings(kCustomizedMaxChunkSizeBytes); MigrationType recoveryInfo(migrateInfo.nss, migrateInfo.minKey, - migrateInfo.maxKey, + *migrateInfo.maxKey, migrateInfo.from, migrateInfo.to, migrateInfo.version, |