mirror of https://github.com/mongodb/mongo
50 lines
1.8 KiB
JavaScript
50 lines
1.8 KiB
JavaScript
/**
|
|
* Tests running aggregations within a client session. This test was designed to reproduce
|
|
* SERVER-33660.
|
|
* @tags: [
|
|
* # This test uses a new $_internalSplitPipeline syntax introduced in 8.0.
|
|
* requires_fcv_80,
|
|
* ]
|
|
*/
|
|
import {ShardingTest} from "jstests/libs/shardingtest.js";
|
|
|
|
const st = new ShardingTest({shards: 2});
|
|
|
|
// Gate this test to transaction supporting engines only as it uses txnNumber.
|
|
let shardDB = st.rs0.getPrimary().getDB("test");
|
|
if (!shardDB.serverStatus().storageEngine.supportsSnapshotReadConcern) {
|
|
jsTestLog("Do not run on storage engine that does not support transactions");
|
|
st.stop();
|
|
quit();
|
|
}
|
|
|
|
const mongosDB = st.s0.getDB("test");
|
|
const session = mongosDB.getMongo().startSession();
|
|
const mongosColl = session.getDatabase("test")[jsTestName()];
|
|
|
|
// Shard the collection, split it into two chunks, and move the [1, MaxKey] chunk to the other
|
|
// shard. We need chunks distributed across multiple shards in order to force a split pipeline
|
|
// merging on a mongod - otherwise the entire pipeline will be forwarded without a split and
|
|
// without a $mergeCursors stage.
|
|
st.shardColl(mongosColl, {_id: 1}, {_id: 1}, {_id: 1});
|
|
assert.commandWorked(mongosColl.insert([{_id: 0}, {_id: 1}, {_id: 2}]));
|
|
|
|
// This assertion will reproduce the hang described in SERVER-33660.
|
|
assert.eq(
|
|
[{_id: 0}, {_id: 1}, {_id: 2}],
|
|
mongosColl
|
|
.aggregate([{$_internalSplitPipeline: {mergeType: {"specificShard": st.shard0.shardName}}}, {$sort: {_id: 1}}])
|
|
.toArray(),
|
|
);
|
|
|
|
// Test a couple more aggregations to be sure.
|
|
assert.eq(
|
|
[{_id: 0}, {_id: 1}, {_id: 2}],
|
|
mongosColl
|
|
.aggregate([{$_internalSplitPipeline: {mergeType: st.getMergeType(mongosDB)}}, {$sort: {_id: 1}}])
|
|
.toArray(),
|
|
);
|
|
assert.eq(mongosColl.aggregate([{$sort: {_id: 1}}, {$out: "testing"}]).itcount(), 0);
|
|
|
|
st.stop();
|