mongo/jstests/sharding/initial_split_validate_shar...

66 lines
2.8 KiB
JavaScript

/**
* Explicitly validates that all shards' collections have the correct UUIDs after an initial split
* which spreads the collection across all available shards.
*/
import {ShardingTest} from "jstests/libs/shardingtest.js";
import {getUUIDFromListCollections} from "jstests/libs/uuid_util.js";
import {findChunksUtil} from "jstests/sharding/libs/find_chunks_util.js";
let st = new ShardingTest({shards: 2});
let mongos = st.s0;
assert.commandWorked(mongos.adminCommand({enableSharding: "test", primaryShard: st.shard1.shardName}));
assert.commandWorked(mongos.adminCommand({shardCollection: "test.user", key: {x: "hashed"}}));
// Ensure that all the pending (received chunks) have been incorporated in the shard's filtering
// metadata so they will show up in the getShardVersion command
assert.eq(0, mongos.getDB("test").user.find({}).itcount());
const expectedChunksOnConfigCount = 2;
const expectedChunksPerShardCount = 1;
st.printShardingStatus();
function checkMetadata(metadata) {
jsTestLog(tojson(metadata));
assert.eq(expectedChunksPerShardCount, metadata.chunks.length);
// Check that the single chunk on the shard's metadata is a valid chunk (getShardVersion
// represents chunks as an array of [min, max])
let chunks = metadata.chunks;
assert(bsonWoCompare(chunks[0][0], chunks[0][1]) < 0);
}
// Check that the shards' in-memory catalog caches were refreshed
checkMetadata(
assert.commandWorked(st.rs0.getPrimary().adminCommand({getShardVersion: "test.user", fullMetadata: true})).metadata,
);
checkMetadata(
assert.commandWorked(st.rs1.getPrimary().adminCommand({getShardVersion: "test.user", fullMetadata: true})).metadata,
);
// Check that the shards' catalogs have the correct UUIDs
const collEntry = mongos.getDB("config").collections.findOne({_id: "test.user"});
const shard0UUID = getUUIDFromListCollections(st.shard0.getDB("test"), "user");
const shard1UUID = getUUIDFromListCollections(st.shard1.getDB("test"), "user");
assert.eq(collEntry.uuid, shard0UUID);
assert.eq(collEntry.uuid, shard1UUID);
// Check that the shards' on-disk caches have the correct number of chunks
assert.commandWorked(st.shard0.adminCommand({_flushRoutingTableCacheUpdates: "test.user", syncFromConfig: false}));
assert.commandWorked(st.shard1.adminCommand({_flushRoutingTableCacheUpdates: "test.user", syncFromConfig: false}));
const chunksOnConfigCount = findChunksUtil.countChunksForNs(st.config, "test.user");
assert.eq(expectedChunksOnConfigCount, chunksOnConfigCount);
const chunksCollName = "cache.chunks.test.user";
const chunksOnShard0 = st.shard0.getDB("config").getCollection(chunksCollName).find().toArray();
const chunksOnShard1 = st.shard1.getDB("config").getCollection(chunksCollName).find().toArray();
assert.eq(chunksOnConfigCount, chunksOnShard0.length);
assert.eq(chunksOnConfigCount, chunksOnShard1.length);
st.stop();