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
|
/**
* Tests 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.
* @tags: [requires_replication, requires_sharding]
*/
// Checking UUID consistency involves talking to a shard node, which in this test is shutdown
TestData.skipCheckingUUIDsConsistentAcrossCluster = true;
(function() {
'use strict';
// Set the refresh period to 10 min to rule out races
_setShellFailPoint({
configureFailPoint: "modifyReplicaSetMonitorDefaultRefreshPeriod",
mode: "alwaysOn",
data: {
period: 10 * 60,
},
});
var dbName = "test";
var collName = jsTest.name();
function runTest(host, rst, waitForPrimary) {
// 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();
var secondary = rst.getSecondary();
assert.commandWorked(primary.getDB("admin").runCommand({replSetStepDown: 60, force: true}));
rst.waitForState(primary, ReplSetTest.State.SECONDARY);
if (waitForPrimary) {
rst.waitForState(secondary, ReplSetTest.State.PRIMARY);
}
// 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, false);
rst.stopSet();
rst = new ReplSetTest({nodes: 2});
rst.startSet();
rst.initiate();
runTest(rst.getURL(), rst, true);
rst.stopSet();
// Test querying a replica set primary through mongos.
var st = new ShardingTest({shards: 1, rs: {nodes: 2}, config: 2});
rst = st.rs0;
runTest(st.s0.host, rst, true);
st.stop();
})();
|
