mirror of https://github.com/mongodb/mongo
80 lines
2.6 KiB
JavaScript
80 lines
2.6 KiB
JavaScript
// Test that interrupting a count returns an error code.
|
|
//
|
|
// @tags: [
|
|
// # This test attempts to perform a count command and find it using the currentOp command. The
|
|
// # former operation may be routed to a secondary in the replica set, whereas the latter must be
|
|
// # routed to the primary.
|
|
// assumes_read_preference_unchanged,
|
|
// # The aggregation stage $currentOp cannot run with a readConcern other than 'local'
|
|
// assumes_read_concern_unchanged,
|
|
// does_not_support_stepdowns,
|
|
// does_not_support_repeated_reads,
|
|
// # Uses $where operator
|
|
// requires_scripting,
|
|
// uses_multiple_connections,
|
|
// uses_parallel_shell,
|
|
// # The balancer can interrupt the count operation, entering in a race with the parallel shell.
|
|
// assumes_balancer_off,
|
|
// requires_getmore,
|
|
// # Time series collections (as views) generate different profiling/plan
|
|
// # summary entries (e.g. distinct from underlying bucket collection).
|
|
// exclude_from_timeseries_crud_passthrough,
|
|
// ]
|
|
|
|
let coll = db.count10;
|
|
coll.drop();
|
|
|
|
let bulk = coll.initializeUnorderedBulkOp();
|
|
for (let i = 0; i < 100; i++) {
|
|
coll.insertOne({x: i});
|
|
}
|
|
assert.commandWorked(bulk.execute());
|
|
|
|
// Start a parallel shell which repeatedly checks for a count
|
|
// query using db.currentOp(). As soon as the op is found,
|
|
// kill it via db.killOp().
|
|
let s = startParallelShell(function () {
|
|
assert.soon(function () {
|
|
let currentCountOps = db
|
|
.getSiblingDB("admin")
|
|
.aggregate([
|
|
{$currentOp: {}},
|
|
{
|
|
$match: {"ns": db.count10.getFullName(), "command.count": db.count10.getName()},
|
|
},
|
|
{$limit: 1},
|
|
])
|
|
.toArray();
|
|
|
|
// Check that we found the count op. If not, return false so
|
|
// that assert.soon will retry.
|
|
if (currentCountOps.length !== 1) {
|
|
jsTest.log("Still didn't find count operation in the currentOp log.");
|
|
return false;
|
|
}
|
|
|
|
let countOp = currentCountOps[0];
|
|
jsTest.log("Found count op:");
|
|
printjson(countOp);
|
|
// Found the count op. Try to kill it.
|
|
assert.commandWorked(db.killOp(countOp.opid));
|
|
return true;
|
|
}, "Could not find count op after retrying, gave up");
|
|
});
|
|
|
|
let res = assert.throws(
|
|
function () {
|
|
coll.find("sleep(1000)").count();
|
|
},
|
|
[],
|
|
"Count op completed without being killed",
|
|
);
|
|
|
|
jsTest.log("Killed count output start");
|
|
printjson(res);
|
|
jsTest.log("Killed count output end");
|
|
assert(res.message.match(/count failed/) !== null);
|
|
assert(res.message.match(/\"code\"/) !== null);
|
|
|
|
s();
|