mongo/jstests/sharding/collection_uuid_insert.js

/**
 * Tests the collectionUUID parameter of the insert command when one collection is sharded and the
 * other collection is unsharded.
 *
 * @tags: [
 *   requires_fcv_60,
 * ]
 */
import {ShardingTest} from "jstests/libs/shardingtest.js";

const st = new ShardingTest({shards: 2});
const mongos = st.s;

const db = mongos.getDB(jsTestName());
assert.commandWorked(mongos.adminCommand({enableSharding: db.getName(), primaryShard: st.shard0.shardName}));

const shardedColl = db.sharded;
const unshardedColl = db.unsharded;

assert.commandWorked(shardedColl.insert([{_id: 0}, {_id: 5}]));
assert.commandWorked(unshardedColl.insert({_id: 0}));

const uuid = function (coll) {
    return assert
        .commandWorked(db.runCommand({listCollections: 1}))
        .cursor.firstBatch.find((c) => c.name === coll.getName()).info.uuid;
};

assert.commandWorked(mongos.adminCommand({shardCollection: shardedColl.getFullName(), key: {_id: 1}}));

// Split at {_id: 5}, moving {_id: 0} to shard0 and {_id: 5} to shard1.
assert.commandWorked(st.splitAt(shardedColl.getFullName(), {_id: 5}));
assert.commandWorked(
    mongos.adminCommand({moveChunk: shardedColl.getFullName(), find: {_id: 0}, to: st.shard0.shardName}),
);
assert.commandWorked(
    mongos.adminCommand({moveChunk: shardedColl.getFullName(), find: {_id: 5}, to: st.shard1.shardName}),
);

// Run an insert which only targets shard1, while the unsharded collection only exists on shard0.
let res = assert.commandFailedWithCode(
    db.runCommand({
        insert: shardedColl.getName(),
        documents: [{_id: 6}],
        collectionUUID: uuid(unshardedColl),
    }),
    ErrorCodes.CollectionUUIDMismatch,
).writeErrors[0];
assert.eq(res.db, db.getName());
assert.eq(res.collectionUUID, uuid(unshardedColl));
assert.eq(res.expectedCollection, shardedColl.getName());
assert.eq(res.actualCollection, unshardedColl.getName());

// Run an insert on the sharded collection with multiple documents that only targets shard1.
res = assert.commandFailedWithCode(
    db.runCommand({
        insert: shardedColl.getName(),
        documents: [{_id: 6}, {_id: 7}, {_id: 8}],
        ordered: false,
        collectionUUID: uuid(unshardedColl),
    }),
    ErrorCodes.CollectionUUIDMismatch,
).writeErrors;
for (let writeError of res) {
    assert.eq(writeError.db, db.getName());
    assert.eq(writeError.collectionUUID, uuid(unshardedColl));
    assert.eq(writeError.expectedCollection, shardedColl.getName());
    assert.eq(writeError.actualCollection, unshardedColl.getName());
}

// Run an insert on the sharded collection with multiple documents that targets both shards.
res = assert.commandFailedWithCode(
    db.runCommand({
        insert: shardedColl.getName(),
        documents: [{_id: 6}, {_id: 7}, {_id: 1}],
        ordered: false,
        collectionUUID: uuid(unshardedColl),
    }),
    ErrorCodes.CollectionUUIDMismatch,
).writeErrors;
for (let writeError of res) {
    assert.eq(writeError.db, db.getName());
    assert.eq(writeError.collectionUUID, uuid(unshardedColl));
    assert.eq(writeError.expectedCollection, shardedColl.getName());
    assert.eq(writeError.actualCollection, unshardedColl.getName());
}

// Run an insert on the unsharded collection, which only exists on shard0.
res = assert.commandFailedWithCode(
    db.runCommand({
        insert: unshardedColl.getName(),
        documents: [{_id: 1}],
        collectionUUID: uuid(shardedColl),
    }),
    ErrorCodes.CollectionUUIDMismatch,
).writeErrors[0];
assert.eq(res.db, db.getName());
assert.eq(res.collectionUUID, uuid(shardedColl));
assert.eq(res.expectedCollection, unshardedColl.getName());
assert.eq(res.actualCollection, shardedColl.getName());

// Run an insert on the unsharded collection with multiple documents.
res = assert.commandFailedWithCode(
    db.runCommand({
        insert: unshardedColl.getName(),
        documents: [{_id: 6}, {_id: 7}, {_id: 8}],
        ordered: false,
        collectionUUID: uuid(shardedColl),
    }),
    ErrorCodes.CollectionUUIDMismatch,
).writeErrors;
for (let writeError of res) {
    assert.eq(writeError.db, db.getName());
    assert.eq(writeError.collectionUUID, uuid(shardedColl));
    assert.eq(writeError.expectedCollection, unshardedColl.getName());
    assert.eq(writeError.actualCollection, shardedColl.getName());
}

st.stop();