64 lines
2.2 KiB
JavaScript
64 lines
2.2 KiB
JavaScript
/**
|
|
* Tests the cloning portion of a resharding operation as part of the reshardCollection command.
|
|
*
|
|
* @tags: [
|
|
* uses_atclustertime,
|
|
* ]
|
|
*/
|
|
import {DiscoverTopology} from "jstests/libs/discover_topology.js";
|
|
import {ReshardingTest} from "jstests/sharding/libs/resharding_test_fixture.js";
|
|
|
|
const reshardingTest = new ReshardingTest({numDonors: 2, numRecipients: 2, reshardInPlace: true});
|
|
|
|
reshardingTest.setup();
|
|
|
|
const donorShardNames = reshardingTest.donorShardNames;
|
|
const inputCollection = reshardingTest.createShardedCollection({
|
|
ns: "reshardingDb.coll",
|
|
shardKeyPattern: {oldKey: 1},
|
|
chunks: [
|
|
{min: {oldKey: MinKey}, max: {oldKey: 0}, shard: donorShardNames[0]},
|
|
{min: {oldKey: 0}, max: {oldKey: MaxKey}, shard: donorShardNames[1]},
|
|
],
|
|
});
|
|
|
|
assert.commandWorked(
|
|
inputCollection.insert([
|
|
{_id: "stays on shard0", oldKey: -10, newKey: -10},
|
|
{_id: "moves to shard0", oldKey: 10, newKey: -10},
|
|
{_id: "moves to shard1", oldKey: -10, newKey: 10},
|
|
{_id: "stays on shard1", oldKey: 10, newKey: 10},
|
|
]),
|
|
);
|
|
|
|
const recipientShardNames = reshardingTest.recipientShardNames;
|
|
reshardingTest.withReshardingInBackground({
|
|
newShardKeyPattern: {newKey: 1},
|
|
newChunks: [
|
|
{min: {newKey: MinKey}, max: {newKey: 0}, shard: recipientShardNames[0]},
|
|
{min: {newKey: 0}, max: {newKey: MaxKey}, shard: recipientShardNames[1]},
|
|
],
|
|
});
|
|
|
|
function assertClonedContents(shardConn, expectedDocs) {
|
|
// We sort by oldKey so the order of `expectedDocs` can be deterministic.
|
|
assert.eq(expectedDocs, shardConn.getCollection(inputCollection.getFullName()).find().sort({oldKey: 1}).toArray());
|
|
}
|
|
|
|
const mongos = inputCollection.getMongo();
|
|
const topology = DiscoverTopology.findConnectedNodes(mongos);
|
|
const recipient0 = new Mongo(topology.shards[recipientShardNames[0]].primary);
|
|
const recipient1 = new Mongo(topology.shards[recipientShardNames[1]].primary);
|
|
|
|
assertClonedContents(recipient0, [
|
|
{_id: "stays on shard0", oldKey: -10, newKey: -10},
|
|
{_id: "moves to shard0", oldKey: 10, newKey: -10},
|
|
]);
|
|
|
|
assertClonedContents(recipient1, [
|
|
{_id: "moves to shard1", oldKey: -10, newKey: 10},
|
|
{_id: "stays on shard1", oldKey: 10, newKey: 10},
|
|
]);
|
|
|
|
reshardingTest.teardown();
|