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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
|
/**
* Tests adding shard to sharded cluster will fail if CWWC on shard disagrees with existing CWWC on
* cluster.
* @tags: [requires_fcv_50, requires_majority_read_concern, requires_persistence]
*/
(function() {
"use strict";
const st = new ShardingTest({shards: 1, mongos: 1});
const admin = st.getDB('admin');
let shardServer;
function createNewShard() {
if (shardServer) {
shardServer.stopSet();
}
shardServer = new ReplSetTest({name: "shardServer", nodes: 3, useHostName: true});
shardServer.startSet();
shardServer.initiate();
}
function convertShardToRS() {
jsTestLog("Converting shard server to replicaSet.");
shardServer.nodes.forEach(function(node) {
delete node.fullOptions.shardsvr;
});
shardServer.restart(shardServer.nodes);
shardServer.awaitNodesAgreeOnPrimary();
}
function convertRSToShard() {
jsTestLog("Converting replicaSet server to shardServer.");
shardServer.restart(shardServer.nodes, {shardsvr: ""});
shardServer.awaitNodesAgreeOnPrimary();
}
function removeShardAndWait() {
jsTestLog("Removing the shard from the cluster should succeed.");
const removeShardCmd = {removeShard: shardServer.getURL()};
const res = st.s.adminCommand(removeShardCmd);
assert.commandWorked(res);
assert(res.state === "started");
assert.soon(function() {
let res = st.s.adminCommand(removeShardCmd);
if (res.state === "completed") {
return true;
} else {
jsTestLog("Still waiting for shard removal to complete:");
printjson(res);
return false;
}
});
jsTestLog("Shard removal completed.");
}
function testAddShard(cwwcOnShard, cwwcOnCluster, shouldSucceed, fixCWWCOnShard) {
jsTestLog("Running addShard test with CWWCOnShard: " + tojson(cwwcOnShard) +
" and CWWCOnCluster: " + tojson(cwwcOnCluster));
let cwwcOnShardCmd, cwwcOnClusterCmd;
if (cwwcOnShard) {
jsTestLog("Setting the CWWC on shard before adding it.");
cwwcOnShardCmd = {
setDefaultRWConcern: 1,
defaultWriteConcern: cwwcOnShard,
writeConcern: {w: "majority"}
};
assert.commandWorked(shardServer.getPrimary().adminCommand(cwwcOnShardCmd));
shardServer.awaitReplication();
}
if (cwwcOnCluster) {
jsTestLog("Setting the CWWC on cluster before adding shard.");
cwwcOnClusterCmd = {
setDefaultRWConcern: 1,
defaultWriteConcern: cwwcOnCluster,
writeConcern: {w: "majority"}
};
assert.commandWorked(st.s.adminCommand(cwwcOnClusterCmd));
}
jsTestLog("Attempting to add shard to the cluster");
convertRSToShard();
if (!shouldSucceed) {
jsTestLog("Adding shard to the cluster should fail.");
assert.commandFailed(admin.runCommand({addshard: shardServer.getURL()}));
if (fixCWWCOnShard) {
convertShardToRS();
jsTestLog("Setting the CWWC on shard to match CWWC on cluster.");
assert.commandWorked(shardServer.getPrimary().adminCommand(cwwcOnClusterCmd));
shardServer.awaitReplication();
convertRSToShard();
} else {
jsTestLog("Setting the CWWC on cluster to match CWWC on shard.");
assert.commandWorked(st.s.adminCommand(cwwcOnShardCmd));
}
}
jsTestLog("Adding shard to the cluster should succeed.");
assert.commandWorked(admin.runCommand({addshard: shardServer.getURL()}));
// Cleanup.
removeShardAndWait();
convertShardToRS();
}
const cwwc = [
{w: 1},
{w: "majority"},
{w: "majority", wtimeout: 0, j: false},
{w: "majority", wtimeout: 0, j: true}
];
createNewShard();
// No CWWC set neither on shard nor cluster should succeed.
testAddShard(undefined /* cwwcOnShard */, undefined /* cwwcOnCluster */, true /* shouldSucceed */);
// No CWWC set on cluster while shard has CWWC should fail.
testAddShard(cwwc[0] /* cwwcOnShard */,
undefined /* cwwcOnCluster */,
false /* shouldSucceed */,
false /* fixCWWCOnShard */);
createNewShard();
// No CWWC set on shard while cluster has CWWC should succeed.
testAddShard(undefined /* cwwcOnShard */, cwwc[0] /* cwwcOnCluster */, true /* shouldSucceed */);
for (var i = 0; i < cwwc.length; i++) {
// Setting the same CWWC on shard and cluster should succeed.
testAddShard(cwwc[i] /* cwwcOnShard */, cwwc[i] /* cwwcOnCluster */, true /* shouldSucceed */);
for (var j = i + 1; j < cwwc.length; j++) {
for (const fixCWWCOnShard of [true, false]) {
// Setting different CWWC on shard and cluster should fail.
testAddShard(cwwc[i] /* cwwcOnShard */,
cwwc[j] /* cwwcOnCluster */,
false /* shouldSucceed */,
fixCWWCOnShard);
}
}
}
st.stop();
shardServer.stopSet();
})();
|
