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
|
/**
* 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 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) {
adminDb.printShardingStatus();
assert.eq(outputDb.numbers_out.count(), 50, "map/reduce failed");
assert.eq(configDb.collections.findOne().dropped, false, "no sharded collections");
}
function assertFailure(configDb, outputDb) {
adminDb.printShardingStatus();
assert.eq(outputDb.numbers_out.count(), 0, "map/reduce should not have succeeded");
}
var st = new ShardingTest( testName = "mrShardedOutputAuth",
numShards = 1,
verboseLevel = 0,
numMongos = 1,
{ extraOptions : {"keyFile" : "jstests/libs/key1"} }
);
// setup the users to the input, output and admin databases
var mongos = st.s;
var adminDb = mongos.getDB("admin");
adminDb.addUser("user", "pass", false);
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.addUser("user", "pass", false, 1);
var outputDb = authenticatedConn.getDB("output");
outputDb.addUser("user", "pass");
// 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();
|
