jstests/sharding/agg_error_reports_shard_host_and_port.js


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 assertErrMsgContains.

    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;

    assertErrMsgContains(coll, pipe, divideByZeroErrorCode, st.rs1.getPrimary().host);

    st.stop();
}());