1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
/**
* Tests the collectionUUID parameter of the shardCollection command.
*
* @tags: [
* requires_fcv_60,
* ]
*/
(function() {
'use strict';
const st = new ShardingTest({shards: 2});
const mongos = st.s0;
const db = mongos.getDB(jsTestName());
assert.commandWorked(mongos.adminCommand({enableSharding: db.getName()}));
const coll = db['coll'];
const resetColl = function() {
coll.drop();
assert.commandWorked(coll.insert({_id: 0}));
};
const uuid = function() {
return assert.commandWorked(db.runCommand({listCollections: 1}))
.cursor.firstBatch.find(c => c.name === coll.getName())
.info.uuid;
};
resetColl();
// The command succeeds when the correct UUID is provided.
assert.commandWorked(mongos.adminCommand(
{shardCollection: coll.getFullName(), key: {_id: 1}, collectionUUID: uuid()}));
// The command fails when the provided UUID does not correspond to an existing collection.
resetColl();
const nonexistentUUID = UUID();
let res = assert.commandFailedWithCode(
mongos.adminCommand(
{shardCollection: coll.getFullName(), key: {_id: 1}, collectionUUID: nonexistentUUID}),
ErrorCodes.CollectionUUIDMismatch);
assert.eq(res.db, db.getName());
assert.eq(res.collectionUUID, nonexistentUUID);
assert.eq(res.expectedCollection, coll.getName());
assert.eq(res.actualCollection, null);
// The command fails when the provided UUID corresponds to a different collection.
const coll2 = db['coll_2'];
assert.commandWorked(coll2.insert({_id: 1}));
res = assert.commandFailedWithCode(
mongos.adminCommand(
{shardCollection: coll2.getFullName(), key: {_id: 1}, collectionUUID: uuid()}),
ErrorCodes.CollectionUUIDMismatch);
assert.eq(res.db, db.getName());
assert.eq(res.collectionUUID, uuid());
assert.eq(res.expectedCollection, coll2.getName());
assert.eq(res.actualCollection, coll.getName());
// Only collections in the same database are specified by actualCollection.
const otherDB = db.getSiblingDB(db.getName() + '_2');
const coll3 = otherDB['coll_3'];
assert.commandWorked(mongos.adminCommand({enableSharding: otherDB.getName()}));
resetColl(coll3);
res = assert.commandFailedWithCode(mongos.adminCommand({
shardCollection: coll3.getFullName(),
key: {_id: 1},
collectionUUID: uuid(),
}),
ErrorCodes.CollectionUUIDMismatch);
assert.eq(res.db, otherDB.getName());
assert.eq(res.collectionUUID, uuid());
assert.eq(res.expectedCollection, coll3.getName());
assert.eq(res.actualCollection, null);
// The command fails when the collection is already sharded and the provided UUID corresponds to a
// different collection.
assert.commandWorked(mongos.adminCommand({shardCollection: coll2.getFullName(), key: {_id: 1}}));
res = assert.commandFailedWithCode(
mongos.adminCommand(
{shardCollection: coll2.getFullName(), key: {_id: 1}, collectionUUID: uuid()}),
ErrorCodes.CollectionUUIDMismatch);
assert.eq(res.db, db.getName());
assert.eq(res.collectionUUID, uuid());
assert.eq(res.expectedCollection, coll2.getName());
assert.eq(res.actualCollection, coll.getName());
// The command fails when the provided UUID corresponds to a different collection, even if the
// provided namespace does not exist.
coll2.drop();
res = assert.commandFailedWithCode(
mongos.adminCommand(
{shardCollection: coll2.getFullName(), key: {_id: 1}, collectionUUID: uuid()}),
ErrorCodes.CollectionUUIDMismatch);
assert.eq(res.db, db.getName());
assert.eq(res.collectionUUID, uuid());
assert.eq(res.expectedCollection, coll2.getName());
assert.eq(res.actualCollection, coll.getName());
st.stop();
})();
|
