diff options
Diffstat (limited to 'jstests/core/getmore_invalidated_cursors.js')
| -rw-r--r-- | jstests/core/getmore_invalidated_cursors.js | 189 |
1 files changed, 94 insertions, 95 deletions
diff --git a/jstests/core/getmore_invalidated_cursors.js b/jstests/core/getmore_invalidated_cursors.js index c244b071716..43f27ed5e49 100644 --- a/jstests/core/getmore_invalidated_cursors.js +++ b/jstests/core/getmore_invalidated_cursors.js @@ -4,117 +4,116 @@ // Tests that running a getMore on a cursor that has been invalidated by something like a collection // drop will return an appropriate error message. (function() { - 'use strict'; +'use strict'; - load('jstests/libs/fixture_helpers.js'); // For FixtureHelpers. +load('jstests/libs/fixture_helpers.js'); // For FixtureHelpers. - const testDB = db.getSiblingDB("getmore_invalidated_cursors"); - const coll = testDB.test; +const testDB = db.getSiblingDB("getmore_invalidated_cursors"); +const coll = testDB.test; - const nDocs = 100; +const nDocs = 100; - function setupCollection() { - coll.drop(); - const bulk = coll.initializeUnorderedBulkOp(); - for (let i = 0; i < nDocs; ++i) { - bulk.insert({_id: i, x: i}); - } - assert.writeOK(bulk.execute()); - assert.commandWorked(coll.createIndex({x: 1})); +function setupCollection() { + coll.drop(); + const bulk = coll.initializeUnorderedBulkOp(); + for (let i = 0; i < nDocs; ++i) { + bulk.insert({_id: i, x: i}); } + assert.writeOK(bulk.execute()); + assert.commandWorked(coll.createIndex({x: 1})); +} - // Test that dropping the database between a find and a getMore will return an appropriate error - // code and message. - setupCollection(); - - // Make sure the batch size is small enough to ensure a getMore will need to be sent to at least - // one shard. - const batchSize = (nDocs / FixtureHelpers.numberOfShardsForCollection(coll)) - 1; - - const isShardedCollection = coll.stats().sharded; - const shellReadMode = testDB.getMongo().readMode(); +// Test that dropping the database between a find and a getMore will return an appropriate error +// code and message. +setupCollection(); - let cursor = coll.find().batchSize(batchSize); - cursor.next(); // Send the query to the server. +// Make sure the batch size is small enough to ensure a getMore will need to be sent to at least +// one shard. +const batchSize = (nDocs / FixtureHelpers.numberOfShardsForCollection(coll)) - 1; - assert.commandWorked(testDB.dropDatabase()); +const isShardedCollection = coll.stats().sharded; +const shellReadMode = testDB.getMongo().readMode(); - let error = assert.throws(() => cursor.itcount()); +let cursor = coll.find().batchSize(batchSize); +cursor.next(); // Send the query to the server. - if (testDB.runCommand({isdbgrid: 1}).isdbgrid && shellReadMode == 'legacy') { - // The cursor will be invalidated on mongos, and we won't be able to find it. - assert.neq(-1, error.message.indexOf('didn\'t exist on server'), error.message); - } else { - assert.eq(error.code, ErrorCodes.QueryPlanKilled, tojson(error)); - assert.neq(-1, error.message.indexOf('collection dropped'), error.message); - } +assert.commandWorked(testDB.dropDatabase()); - // Test that dropping the collection between a find and a getMore will return an appropriate - // error code and message. - setupCollection(); - cursor = coll.find().batchSize(batchSize); - cursor.next(); // Send the query to the server. +let error = assert.throws(() => cursor.itcount()); - coll.drop(); - error = assert.throws(() => cursor.itcount()); +if (testDB.runCommand({isdbgrid: 1}).isdbgrid && shellReadMode == 'legacy') { + // The cursor will be invalidated on mongos, and we won't be able to find it. + assert.neq(-1, error.message.indexOf('didn\'t exist on server'), error.message); +} else { assert.eq(error.code, ErrorCodes.QueryPlanKilled, tojson(error)); - // In replica sets, collection drops are done in two phases, first renaming the collection to a - // "drop pending" namespace, and then later reaping the collection. Therefore, we expect to - // either see an error message related to a collection drop, or one related to a collection - // rename. - const droppedMsg = 'collection dropped'; - const renamedMsg = 'collection renamed'; - assert(-1 !== error.message.indexOf(droppedMsg) || -1 !== error.message.indexOf(renamedMsg), - error.message); - - // Test that dropping an index between a find and a getMore has no effect on the query if the - // query is not using the index. + assert.neq(-1, error.message.indexOf('collection dropped'), error.message); +} + +// Test that dropping the collection between a find and a getMore will return an appropriate +// error code and message. +setupCollection(); +cursor = coll.find().batchSize(batchSize); +cursor.next(); // Send the query to the server. + +coll.drop(); +error = assert.throws(() => cursor.itcount()); +assert.eq(error.code, ErrorCodes.QueryPlanKilled, tojson(error)); +// In replica sets, collection drops are done in two phases, first renaming the collection to a +// "drop pending" namespace, and then later reaping the collection. Therefore, we expect to +// either see an error message related to a collection drop, or one related to a collection +// rename. +const droppedMsg = 'collection dropped'; +const renamedMsg = 'collection renamed'; +assert(-1 !== error.message.indexOf(droppedMsg) || -1 !== error.message.indexOf(renamedMsg), + error.message); + +// Test that dropping an index between a find and a getMore has no effect on the query if the +// query is not using the index. +setupCollection(); +cursor = coll.find().batchSize(batchSize); +cursor.next(); // Send the query to the server. +assert.commandWorked(testDB.runCommand({dropIndexes: coll.getName(), index: {x: 1}})); +assert.eq(cursor.itcount(), nDocs - 1); + +// Test that dropping the index being scanned by a cursor between a find and a getMore kills the +// query with the appropriate code and message. +setupCollection(); +cursor = coll.find().hint({x: 1}).batchSize(batchSize); +cursor.next(); // Send the query to the server. +assert.commandWorked(testDB.runCommand({dropIndexes: coll.getName(), index: {x: 1}})); +error = assert.throws(() => cursor.itcount()); +assert.eq(error.code, ErrorCodes.QueryPlanKilled, tojson(error)); +assert.neq(-1, error.message.indexOf('index \'x_1\' dropped'), error.message); + +// Test that killing a cursor between a find and a getMore will return an appropriate error +// code and message. + +setupCollection(); +// Use the find command so that we can extract the cursor id to pass to the killCursors command. +let cursorId = + assert + .commandWorked(testDB.runCommand({find: coll.getName(), filter: {}, batchSize: batchSize})) + .cursor.id; +assert.commandWorked(testDB.runCommand({killCursors: coll.getName(), cursors: [cursorId]})); +assert.commandFailedWithCode(testDB.runCommand({getMore: cursorId, collection: coll.getName()}), + ErrorCodes.CursorNotFound); + +// Test that all cursors on collections to be renamed get invalidated. Note that we can't do +// renames on sharded collections. +if (!isShardedCollection) { setupCollection(); + const collRenamed = testDB.test_rename; + collRenamed.drop(); cursor = coll.find().batchSize(batchSize); - cursor.next(); // Send the query to the server. - assert.commandWorked(testDB.runCommand({dropIndexes: coll.getName(), index: {x: 1}})); - assert.eq(cursor.itcount(), nDocs - 1); + assert(cursor.hasNext(), "Expected more data from find call on " + coll.getName()); + assert.commandWorked(testDB.adminCommand({ + renameCollection: testDB.getName() + "." + coll.getName(), + to: testDB.getName() + "." + collRenamed.getName() + })); - // Test that dropping the index being scanned by a cursor between a find and a getMore kills the - // query with the appropriate code and message. - setupCollection(); - cursor = coll.find().hint({x: 1}).batchSize(batchSize); - cursor.next(); // Send the query to the server. - assert.commandWorked(testDB.runCommand({dropIndexes: coll.getName(), index: {x: 1}})); + // Ensure getMore fails with an appropriate error code and message. error = assert.throws(() => cursor.itcount()); assert.eq(error.code, ErrorCodes.QueryPlanKilled, tojson(error)); - assert.neq(-1, error.message.indexOf('index \'x_1\' dropped'), error.message); - - // Test that killing a cursor between a find and a getMore will return an appropriate error - // code and message. - - setupCollection(); - // Use the find command so that we can extract the cursor id to pass to the killCursors command. - let cursorId = assert - .commandWorked(testDB.runCommand( - {find: coll.getName(), filter: {}, batchSize: batchSize})) - .cursor.id; - assert.commandWorked(testDB.runCommand({killCursors: coll.getName(), cursors: [cursorId]})); - assert.commandFailedWithCode(testDB.runCommand({getMore: cursorId, collection: coll.getName()}), - ErrorCodes.CursorNotFound); - - // Test that all cursors on collections to be renamed get invalidated. Note that we can't do - // renames on sharded collections. - if (!isShardedCollection) { - setupCollection(); - const collRenamed = testDB.test_rename; - collRenamed.drop(); - cursor = coll.find().batchSize(batchSize); - assert(cursor.hasNext(), "Expected more data from find call on " + coll.getName()); - assert.commandWorked(testDB.adminCommand({ - renameCollection: testDB.getName() + "." + coll.getName(), - to: testDB.getName() + "." + collRenamed.getName() - })); - - // Ensure getMore fails with an appropriate error code and message. - error = assert.throws(() => cursor.itcount()); - assert.eq(error.code, ErrorCodes.QueryPlanKilled, tojson(error)); - assert.neq(-1, error.message.indexOf('collection renamed'), error.message); - } - + assert.neq(-1, error.message.indexOf('collection renamed'), error.message); +} }()); |