blob: 5c9fd65d8f25e5cbad8bcfca7edebc75df93a4dc (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
// Tests that an aggregation error which occurs on a sharded collection will send an error message
// containing the host and port of the shard where the error occurred.
(function() {
"use strict";
load("jstests/aggregation/extras/utils.js"); // For assertErrCodeAndErrMsgContains.
const st = new ShardingTest({shards: 2, config: 1});
const mongosDb = st.s.getDB(jsTestName());
const coll = mongosDb.getCollection("foo");
// Enable sharding on the test DB and ensure its primary is shard 0.
assert.commandWorked(mongosDb.adminCommand({enableSharding: mongosDb.getName()}));
st.ensurePrimaryShard(mongosDb.getName(), st.rs0.getURL());
// Shard the collection.
coll.drop();
st.shardColl(coll, {_id: 1}, {_id: 0}, {_id: 1});
assert.commandWorked(coll.insert({_id: 0}));
// Run an aggregation which will fail on shard 1, and verify that the error message contains
// the host and port of the shard that failed.
// We need to be careful here to involve some data in the computation that is actually
// sent to the shard before failing (i.e. "$_id") so that mongos doesn't short-curcuit and
// fail during optimization.
const pipe = [{$project: {a: {$divide: ["$_id", 0]}}}];
const divideByZeroErrorCode = 16608;
assertErrCodeAndErrMsgContains(coll, pipe, divideByZeroErrorCode, st.rs1.getPrimary().host);
st.stop();
}());
|