mongo/jstests/sharding/remove_shard_1.js
Anna Maria Nestorov da6ca6ce50 SERVER-112544 Rename addShard and removeShard test to have a consistent naming (#43132)
GitOrigin-RevId: 2d0dbb9cb6997c7918960feea9d81a5d0e64a6f3
2025-10-27 17:02:24 +00:00

51 lines
2.3 KiB
JavaScript

import {ShardTransitionUtil} from "jstests/libs/shard_transition_util.js";
import {ShardingTest} from "jstests/libs/shardingtest.js";
let s = new ShardingTest({shards: 2, other: {enableBalancer: true}});
let config = s.s0.getDB("config");
assert.commandWorked(s.s0.adminCommand({enableSharding: "needToMove", primaryShard: s.shard0.shardName}));
let topologyTime0 = config.shards.findOne({_id: s.shard0.shardName}).topologyTime;
let topologyTime1 = config.shards.findOne({_id: s.shard1.shardName}).topologyTime;
assert.gt(topologyTime1, topologyTime0);
// removeShard is not permited on shard0 (the configShard) if configShard is enabled, so we want
// to use transitionToDedicatedConfigServer instead
let removeShardOrTransitionToDedicated = TestData.configShard ? "transitionToDedicatedConfigServer" : "removeShard";
// First remove puts in draining mode, the second tells me a db needs to move, the third
// actually removes
assert.commandWorked(s.s0.adminCommand({[removeShardOrTransitionToDedicated]: s.shard0.shardName}));
// Can't make all shards in the cluster draining
assert.commandFailedWithCode(s.s0.adminCommand({removeshard: s.shard1.shardName}), ErrorCodes.IllegalOperation);
let removeResult = assert.commandWorked(s.s0.adminCommand({[removeShardOrTransitionToDedicated]: s.shard0.shardName}));
assert.eq(removeResult.dbsToMove, ["needToMove"], "didn't show db to move");
assert(removeResult.note !== undefined);
s.s0.getDB("needToMove").dropDatabase();
// Ensure the balancer moves the config.system.sessions collection chunks out of the shard being
// removed
s.awaitBalancerRound();
if (TestData.configShard) {
// A config shard can't be removed until all range deletions have finished.
ShardTransitionUtil.waitForRangeDeletions(s.s);
}
removeResult = assert.commandWorked(s.s0.adminCommand({[removeShardOrTransitionToDedicated]: s.shard0.shardName}));
assert.eq("completed", removeResult.state, "Shard was not removed: " + tojson(removeResult));
let existingShards = config.shards.find({}).toArray();
assert.eq(1, existingShards.length, "Removed server still appears in count: " + tojson(existingShards));
let topologyTime2 = existingShards[0].topologyTime;
assert.gt(topologyTime2, topologyTime1);
assert.commandFailed(s.s0.adminCommand({removeshard: s.shard1.shardName}));
s.stop();