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
|
// Tests that the read preference set on the connection is used when we call the count helper.
(function() {
"use strict";
var commandsRan = [];
// Create a new DB object backed by a mock connection.
function MockMongo() {
this.getMinWireVersion = function getMinWireVersion() {
return 0;
};
this.getMaxWireVersion = function getMaxWireVersion() {
return 0;
};
}
MockMongo.prototype = Mongo.prototype;
MockMongo.prototype.runCommand = function(db, cmd, opts) {
commandsRan.push({db: db, cmd: cmd, opts: opts});
return {ok: 1, n: 100};
};
const mockMongo = new MockMongo();
var db = new DB(mockMongo, "test");
// Attach a dummy implicit session because the mock connection cannot create sessions.
db._session = new _DummyDriverSession(mockMongo);
assert.eq(commandsRan.length, 0);
// Run a count with no readPref.
db.getMongo().setReadPref(null);
db.foo.count();
// Check that there is no readPref on the command document.
assert.eq(commandsRan.length, 1);
assert.docEq(commandsRan[0].cmd, {count: "foo", query: {}});
commandsRan = [];
// Run with readPref secondary.
db.getMongo().setReadPref("secondary");
db.foo.count();
// Check that we have wrapped the command and attached the read preference.
assert.eq(commandsRan.length, 1);
assert.docEq(commandsRan[0].cmd,
{query: {count: "foo", query: {}}, $readPreference: {mode: "secondary"}});
})();
|
