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
|
// SERVER-5124
// The puporse of this test is to test authentication when adding/removing a shard. The test sets
// up a sharded system, then adds/removes a shard.
(function() {
'use strict';
load('jstests/sharding/libs/remove_shard_util.js');
// TODO SERVER-50144 Remove this and allow orphan checking.
// This test calls removeShard which can leave docs in config.rangeDeletions in state "pending",
// therefore preventing orphans from being cleaned up.
TestData.skipCheckOrphans = true;
// login method to login into the database
function login(userObj) {
var authResult = mongos.getDB(userObj.db).auth(userObj.username, userObj.password);
printjson(authResult);
}
// admin user object
var adminUser = {db: "admin", username: "foo", password: "bar"};
// set up a 2 shard cluster with keyfile
var st = new ShardingTest({shards: 1, mongos: 1, other: {keyFile: 'jstests/libs/key1'}});
var mongos = st.s0;
var admin = mongos.getDB("admin");
print("1 shard system setup");
// add the admin user
print("adding user");
mongos.getDB(adminUser.db)
.createUser({user: adminUser.username, pwd: adminUser.password, roles: jsTest.adminUserRoles});
// login as admin user
login(adminUser);
assert.eq(1, st.config.shards.count(), "initial server count wrong");
// start a mongod with NO keyfile
var rst = new ReplSetTest({nodes: 1});
rst.startSet({shardsvr: ""});
rst.initiateWithAnyNodeAsPrimary();
// --------------- Test 1 --------------------
// Add shard to the existing cluster (should fail because it was added without a keyfile)
printjson(assert.commandFailed(admin.runCommand({addShard: rst.getURL()})));
// stop mongod
rst.stopSet();
//--------------- Test 2 --------------------
// start mongod again, this time with keyfile
rst = new ReplSetTest({nodes: 1});
rst.startSet({keyFile: "jstests/libs/key1", shardsvr: ""});
rst.initiateWithAnyNodeAsPrimary();
// try adding the new shard
var addShardRes = admin.runCommand({addShard: rst.getURL()});
assert.commandWorked(addShardRes);
// Add some data
var db = mongos.getDB("foo");
var collA = mongos.getCollection("foo.bar");
// enable sharding on a collection
assert.commandWorked(admin.runCommand({enableSharding: "" + collA.getDB()}));
st.ensurePrimaryShard("foo", st.shard0.shardName);
assert.commandWorked(admin.runCommand({shardCollection: "" + collA, key: {_id: 1}}));
// add data to the sharded collection
for (var i = 0; i < 4; i++) {
db.bar.save({_id: i});
assert.commandWorked(admin.runCommand({split: "" + collA, middle: {_id: i}}));
}
// move a chunk
// TODO (SERVER-60767): remove _waitForDelete param; removeShard() will sync on range deletion.
assert.commandWorked(admin.runCommand(
{moveChunk: "foo.bar", find: {_id: 1}, to: addShardRes.shardAdded, _waitForDelete: true}));
// verify the chunk was moved
admin.runCommand({flushRouterConfig: 1});
var config = mongos.getDB("config");
st.printShardingStatus(true);
// start balancer before removing the shard
st.startBalancer();
//--------------- Test 3 --------------------
// now drain the shard
removeShard(st, rst.getURL());
// create user directly on new shard to allow direct reads from config.migrationCoordinators
rst.getPrimary()
.getDB(adminUser.db)
.createUser({user: adminUser.username, pwd: adminUser.password, roles: jsTest.adminUserRoles});
rst.getPrimary().getDB(adminUser.db).auth(adminUser.username, adminUser.password);
// wait until migration coordinator is finished
assert.soon(function() {
let migrationCoordinatorDocs =
rst.getPrimary().getDB('config').migrationCoordinators.find().toArray();
return migrationCoordinatorDocs.length === 0;
}, "failed to remove migration coordinator", 5 * 60 * 1000);
assert.eq(1, st.config.shards.count(), "removed server still appears in count");
rst.stopSet();
st.stop();
})();
|
