mongo/jstests/replsets/rollback_clustered_indexes.js

67 lines
2.3 KiB
JavaScript

/**
* Tests that writes on collections clustered by _id can be rolled back.
* @tags: [
* requires_replication,
* ]
*/
import {getUUIDFromListCollections} from "jstests/libs/uuid_util.js";
import {checkRollbackFiles} from "jstests/replsets/libs/rollback_files.js";
import {RollbackTest} from "jstests/replsets/libs/rollback_test.js";
// Operations that will be present on both nodes, before the common point.
const dbName = jsTestName();
const collName = "testColl";
const fullCollName = `${dbName}.${collName}`;
let commonOps = (node) => {
const db = node.getDB(dbName);
assert.commandWorked(db.createCollection(collName, {clusteredIndex: {key: {_id: 1}, unique: true}}));
const coll = node.getCollection(fullCollName);
assert.commandWorked(coll.createIndex({a: 1, b: -1}));
assert.commandWorked(coll.insert({a: 0, b: 0}));
};
// Operations that will be performed on the rollback node past the common point.
let rollbackDocs = [];
let rollbackOps = (node) => {
const coll = node.getCollection(fullCollName);
let doc;
doc = {_id: new ObjectId(), a: 1, b: 3};
assert.commandWorked(coll.insert(doc));
rollbackDocs.push(doc);
doc = {_id: new ObjectId(), a: 2, b: 2};
assert.commandWorked(coll.insert(doc));
rollbackDocs.push(doc);
doc = {_id: new ObjectId(), a: 3, b: 1};
assert.commandWorked(coll.insert(doc));
rollbackDocs.push(doc);
};
// Set up Rollback Test.
const rollbackTest = new RollbackTest();
commonOps(rollbackTest.getPrimary());
const rollbackNode = rollbackTest.transitionToRollbackOperations();
rollbackOps(rollbackNode);
// Wait for rollback to finish.
rollbackTest.transitionToSyncSourceOperationsBeforeRollback();
rollbackTest.transitionToSyncSourceOperationsDuringRollback();
rollbackTest.transitionToSteadyStateOperations();
// Check collection count.
const primary = rollbackTest.getPrimary();
const coll = primary.getCollection(fullCollName);
assert.eq(1, coll.find().itcount());
assert.eq(1, coll.count());
// Confirm that the rollback wrote deleted documents to a file.
const replTest = rollbackTest.getTestFixture();
const uuid = getUUIDFromListCollections(rollbackTest.getPrimary().getDB(dbName), collName);
checkRollbackFiles(replTest.getDbPath(rollbackNode), fullCollName, uuid, rollbackDocs);
rollbackTest.stop();