0
0
mirror of https://github.com/mongodb/mongo.git synced 2024-12-01 01:21:03 +01:00
mongodb/jstests/replsets/temp_namespace.js

106 lines
4.0 KiB
JavaScript

// SERVER-10927
// This is to make sure that temp collections get cleaned up on promotion to primary
// @tags: [requires_replication]
var replTest = new ReplSetTest({name: 'testSet', nodes: 3});
var nodes = replTest.nodeList();
printjson(nodes);
// We need an arbiter to ensure that the primary doesn't step down when we restart the secondary
replTest.startSet();
replTest.initiate({
"_id": "testSet",
"members": [
{"_id": 0, "host": nodes[0]},
{"_id": 1, "host": nodes[1]},
{"_id": 2, "host": nodes[2], "arbiterOnly": true}
]
});
var master = replTest.getPrimary();
var second = replTest.getSecondary();
var masterId = replTest.getNodeId(master);
var secondId = replTest.getNodeId(second);
var masterDB = master.getDB('test');
var secondDB = second.getDB('test');
// set up collections
assert.commandWorked(masterDB.runCommand(
{applyOps: [{op: "c", ns: masterDB.getName() + ".$cmd", o: {create: "temp1", temp: true}}]}));
masterDB.temp1.ensureIndex({x: 1});
assert.commandWorked(masterDB.runCommand(
{applyOps: [{op: "c", ns: masterDB.getName() + ".$cmd", o: {create: "temp2", temp: 1}}]}));
masterDB.temp2.ensureIndex({x: 1});
assert.commandWorked(masterDB.runCommand(
{applyOps: [{op: "c", ns: masterDB.getName() + ".$cmd", o: {create: "keep1", temp: false}}]}));
assert.commandWorked(masterDB.runCommand(
{applyOps: [{op: "c", ns: masterDB.getName() + ".$cmd", o: {create: "keep2", temp: 0}}]}));
masterDB.runCommand({create: 'keep3'});
assert.writeOK(masterDB.keep4.insert({}, {writeConcern: {w: 2}}));
// make sure they exist on primary and secondary
function countCollection(mydb, nameFilter) {
var result = mydb.runCommand("listCollections", {filter: {name: nameFilter}});
assert.commandWorked(result);
return new DBCommandCursor(mydb, result).itcount();
}
function countIndexesFor(mydb, nameFilter) {
var result = mydb.runCommand("listCollections", {filter: {name: nameFilter}});
assert.commandWorked(result);
var arr = new DBCommandCursor(mydb, result).toArray();
var total = 0;
for (var i = 0; i < arr.length; i++) {
var coll = arr[i];
total += mydb.getCollection(coll.name).getIndexes().length;
}
return total;
}
assert.eq(countCollection(masterDB, /temp\d$/), 2); // collections
assert.eq(countIndexesFor(masterDB, /temp\d$/), 4); // indexes (2 _id + 2 x)
assert.eq(countCollection(masterDB, /keep\d$/), 4);
assert.eq(countCollection(secondDB, /temp\d$/), 2); // collections
assert.eq(countIndexesFor(secondDB, /temp\d$/), 4); // indexes (2 _id + 2 x)
assert.eq(countCollection(secondDB, /keep\d$/), 4);
// restart secondary and reconnect
replTest.restart(secondId, {}, /*wait=*/true);
// wait for the secondary to achieve secondary status
assert.soon(function() {
try {
res = second.getDB("admin").runCommand({replSetGetStatus: 1});
return res.myState == 2;
} catch (e) {
return false;
}
}, "took more than a minute for the secondary to become secondary again", 60 * 1000);
// make sure restarting secondary didn't drop collections
assert.eq(countCollection(secondDB, /temp\d$/), 2); // collections
assert.eq(countIndexesFor(secondDB, /temp\d$/), 4); // indexes (2 _id + 2 x)
assert.eq(countCollection(secondDB, /keep\d$/), 4);
// step down primary and make sure former secondary (now primary) drops collections
assert.commandWorked(master.adminCommand({replSetStepDown: 50, force: true}));
assert.soon(function() {
return secondDB.isMaster().ismaster;
}, '', 75 * 1000); // must wait for secondary to be willing to promote self
assert.eq(countCollection(secondDB, /temp\d$/), 0); // collections
assert.eq(countIndexesFor(secondDB, /temp\d$/), 0); // indexes (2 _id + 2 x)
assert.eq(countCollection(secondDB, /keep\d$/), 4);
// check that former primary dropped collections
replTest.awaitReplication();
assert.eq(countCollection(masterDB, /temp\d$/), 0); // collections
assert.eq(countIndexesFor(masterDB, /temp\d$/), 0); // indexes (2 _id + 2 x)
assert.eq(countCollection(masterDB, /keep\d$/), 4);
replTest.stopSet();