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
|
// Basic tests for the $listLocalSessions aggregation stage.
//
// @tags: [
// # This test attempts to start a session and find it using the $listLocalSessions stage. The
// # former operation must be routed to the primary in a replica set, whereas the latter may be
// # routed to a secondary.
// assumes_read_preference_unchanged,
// # Sessions are asynchronously flushed to disk, so a stepdown immediately after calling
// # startSession may cause this test to fail to find the returned sessionId.
// does_not_support_stepdowns,
// requires_sharding,
// ]
(function() {
'use strict';
const st = new ShardingTest({
shards: 1,
mongos: 1,
other: {mongosOptions: {setParameter: {disableLogicalSessionCacheRefresh: true}}}
});
const admin = st.s.getDB("admin");
function listLocalSessions() {
return admin.aggregate([{'$listLocalSessions': {allUsers: false}}]);
}
// Get current log level.
let originalLogLevel = assert.commandWorked(admin.setLogLevel(1)).was.verbosity;
try {
// Start a new session and capture its sessionId.
const myid = assert.commandWorked(st.s.adminCommand({startSession: 1})).id.id;
assert(myid !== undefined);
// Ensure that the cache now contains the session and is visible.
const resultArray = assert.doesNotThrow(listLocalSessions).toArray();
assert.gte(resultArray.length, 1);
const resultArrayMine = resultArray
.map(function(sess) {
return sess._id.id;
})
.filter(function(id) {
return 0 == bsonWoCompare({x: id}, {x: myid});
});
assert.eq(resultArrayMine.length, 1);
// Try asking for the session by username.
const myusername = (function() {
if (0 == bsonWoCompare({x: resultArray[0]._id.uid}, {x: computeSHA256Block("")})) {
// Code for "we're running in no-auth mode"
return {user: "", db: ""};
}
const connstats = assert.commandWorked(db.runCommand({connectionStatus: 1}));
const authUsers = connstats.authInfo.authenticatedUsers;
assert(authUsers !== undefined);
assert.eq(authUsers.length, 1);
assert(authUsers[0].user !== undefined);
assert(authUsers[0].db !== undefined);
return {user: authUsers[0].user, db: authUsers[0].db};
})();
const listMyLocalSessions = function() {
return admin.aggregate([{'$listLocalSessions': {users: [myusername]}}]);
};
const myArray = assert.doesNotThrow(listMyLocalSessions)
.toArray()
.map(function(sess) {
return sess._id.id;
})
.filter(function(id) {
return 0 == bsonWoCompare({x: id}, {x: myid});
});
assert.eq(myArray.length, 1);
print("sessions returned from $listLocalSessions filtered by user: [ " + myArray +
" ]");
print("sessions returned from un-filtered $listLocalSessions for this user: [ " +
resultArrayMine + " ]");
assert.eq(
0,
bsonWoCompare(myArray, resultArrayMine),
"set of listed sessions for user contains different sessions from prior $listLocalSessions run");
} finally {
admin.setLogLevel(originalLogLevel);
}
st.stop();
})();
|