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
|
// Test that a query with default read preference ("primary") will succeed even if the node being
// queried steps down before the final result batch has been delivered.
(function() {
"use strict";
var dbName = "test";
var collName = jsTest.name();
function runTest(host, rst) {
// We create a new connection to 'host' here instead of passing in the original connection.
// This to work around the fact that connections created by ReplSetTest already have slaveOk
// set on them, but we need a connection with slaveOk not set for this test.
var conn = new Mongo(host);
var coll = conn.getDB(dbName).getCollection(collName);
assert(!coll.exists());
assert.writeOK(coll.insert([{}, {}, {}, {}, {}]));
var cursor = coll.find().batchSize(2);
// Retrieve the first batch of results.
cursor.next();
cursor.next();
assert.eq(0, cursor.objsLeftInBatch());
var primary = rst.getPrimary();
assert.throws(function() {
primary.getDB("admin").runCommand({replSetStepDown: 60, force: true});
});
rst.waitForState(primary, ReplSetTest.State.SECONDARY, 60 * 1000);
// When the primary steps down, it closes all client connections. Since 'conn' may be a
// direct connection to the primary and the shell doesn't automatically retry operations on
// network errors, we run a dummy operation here to force the shell to reconnect.
try {
conn.getDB("admin").runCommand("ping");
} catch (e) {
}
// Even though our connection doesn't have slaveOk set, we should still be able to iterate
// our cursor and kill our cursor.
assert(cursor.hasNext());
assert.doesNotThrow(function() {
cursor.close();
});
}
// Test querying a replica set primary directly.
var rst = new ReplSetTest({nodes: 1});
rst.startSet();
rst.initiate();
runTest(rst.getPrimary().host, rst);
rst.stopSet();
// Test querying a replica set primary through mongos.
var st = new ShardingTest({shards: 1, rs: true});
rst = st.rs0;
runTest(st.s0.host, rst);
st.stop();
})();
|
