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
|
/**
* mrShardedOutputAuth.js -- SERVER-7641
* Test that a mapReduce job can write sharded output to a database
* from a separate input database while authenticated to both.
*/
(function() {
// TODO SERVER-35447: Multiple users cannot be authenticated on one connection within a session.
TestData.disableImplicitSessions = true;
function doMapReduce(connection, outputDb) {
// clean output db and run m/r
outputDb.numbers_out.drop();
printjson(connection.getDB('input').runCommand({
mapreduce: "numbers",
map: function() {
emit(this.num, {count: 1});
},
reduce: function(k, values) {
var result = {};
values.forEach(function(value) {
result.count = 1;
});
return result;
},
out: {merge: "numbers_out", sharded: true, db: "output"},
verbose: true,
query: {}
}));
}
function assertSuccess(configDb, outputDb) {
assert.eq(outputDb.numbers_out.count(), 50, "map/reduce failed");
assert(!configDb.collections.findOne().dropped, "no sharded collections");
}
function assertFailure(configDb, outputDb) {
assert.eq(outputDb.numbers_out.count(), 0, "map/reduce should not have succeeded");
}
// TODO: Remove 'shardAsReplicaSet: false' when SERVER-32672 is fixed.
var st = new ShardingTest({
name: "mrShardedOutputAuth",
shards: 1,
mongos: 1,
other: {keyFile: 'jstests/libs/key1', shardAsReplicaSet: false}
});
// Setup the users to the input, output and admin databases
var mongos = st.s;
var adminDb = mongos.getDB("admin");
adminDb.createUser({user: "user", pwd: "pass", roles: jsTest.adminUserRoles});
var authenticatedConn = new Mongo(mongos.host);
authenticatedConn.getDB('admin').auth("user", "pass");
adminDb = authenticatedConn.getDB("admin");
var configDb = authenticatedConn.getDB("config");
var inputDb = authenticatedConn.getDB("input");
inputDb.createUser({user: "user", pwd: "pass", roles: jsTest.basicUserRoles});
var outputDb = authenticatedConn.getDB("output");
outputDb.createUser({user: "user", pwd: "pass", roles: jsTest.basicUserRoles});
// Setup the input db
inputDb.numbers.drop();
for (var i = 0; i < 50; i++) {
inputDb.numbers.insert({num: i});
}
assert.eq(inputDb.numbers.count(), 50);
// Setup a connection authenticated to both input and output db
var inputOutputAuthConn = new Mongo(mongos.host);
inputOutputAuthConn.getDB('input').auth("user", "pass");
inputOutputAuthConn.getDB('output').auth("user", "pass");
doMapReduce(inputOutputAuthConn, outputDb);
assertSuccess(configDb, outputDb);
// setup a connection authenticated to only input db
var inputAuthConn = new Mongo(mongos.host);
inputAuthConn.getDB('input').auth("user", "pass");
doMapReduce(inputAuthConn, outputDb);
assertFailure(configDb, outputDb);
// setup a connection authenticated to only output db
var outputAuthConn = new Mongo(mongos.host);
outputAuthConn.getDB('output').auth("user", "pass");
doMapReduce(outputAuthConn, outputDb);
assertFailure(configDb, outputDb);
st.stop();
})();
|
