34 lines
1.3 KiB
JavaScript
34 lines
1.3 KiB
JavaScript
// This test ensures that an aggregation pipeline with a $sample stage can be issued through a
|
|
// direct connection to a shard without failing, and $sample behaves as if we sampled an unsharded
|
|
// collection.
|
|
// @tags: [requires_fcv_51]
|
|
import {ShardingTest} from "jstests/libs/shardingtest.js";
|
|
|
|
let st = new ShardingTest({shards: 1});
|
|
|
|
assert.commandWorked(st.s.adminCommand({enableSharding: 'test'}));
|
|
assert.commandWorked(st.s.adminCommand({shardCollection: 'test.sharded', key: {x: 1}}));
|
|
const testDB = st.s.getDB('test');
|
|
|
|
// We must have >100 samples to attempt the storage optimized sample path.
|
|
for (let x = 0; x < 101; x++) {
|
|
assert.commandWorked(testDB.foo.insert({x: x}));
|
|
}
|
|
|
|
const shardDB = st.rs0.getPrimary().getDB('test');
|
|
|
|
const res = assert.commandWorked(
|
|
shardDB.runCommand({aggregate: 'foo', pipeline: [{$sample: {size: 3}}], cursor: {}}));
|
|
assert.eq(res.cursor.firstBatch.length, 3);
|
|
|
|
// TODO(SERVER-94154): Remove version check here.
|
|
const fcvDoc = st.s.adminCommand({getParameter: 1, featureCompatibilityVersion: 1});
|
|
if (MongoRunner.compareBinVersions(fcvDoc.featureCompatibilityVersion.version, "8.1") >= 0) {
|
|
// Using a sample of size zero is only disallowed in some newer versions.
|
|
assert.commandFailedWithCode(
|
|
shardDB.runCommand({aggregate: 'foo', pipeline: [{$sample: {size: 0}}], cursor: {}}),
|
|
28747);
|
|
}
|
|
|
|
st.stop();
|