summaryrefslogtreecommitdiff
path: root/jstests/concurrency/fsm_workloads/explain_aggregate.js
blob: b59524c2abb0a57154499e9d37c44441755b6a8c (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
35
36
37
38
39
40
41
42
43
44
'use strict';

/**
 * explain_aggregate.js
 *
 * Runs explain() and aggregate() on a collection.
 *
 */
load('jstests/concurrency/fsm_libs/extend_workload.js');  // for extendWorkload
load('jstests/concurrency/fsm_workloads/explain.js');     // for $config

var $config = extendWorkload($config, function($config, $super) {
    function assertCursorStages(num, obj) {
        assertAlways(obj.stages, tojson(obj));
        assertAlways.eq(num, obj.stages.length, tojson(obj.stages));
        assertAlways(obj.stages[0].$cursor, tojson(obj.stages[0]));
        assertAlways(obj.stages[0].$cursor.hasOwnProperty('queryPlanner'),
                     tojson(obj.stages[0].$cursor));
    }

    $config.states = Object.extend({
        explainMatch: function explainMatch(db, collName) {
            var res = db[collName].explain().aggregate([{$match: {i: this.nInserted / 2}}]);
            assertAlways.commandWorked(res);

            // stages reported: $cursor
            assertCursorStages(1, res);
        },
        explainMatchProject: function explainMatchProject(db, collName) {
            var res = db[collName].explain().aggregate(
                [{$match: {i: this.nInserted / 3}}, {$project: {i: 1}}]);
            assertAlways.commandWorked(res);

            // stages reported: $cursor, $project
            assertCursorStages(2, res);
        }
    },
                                   $super.states);

    $config.transitions = Object.extend(
        {explain: $config.data.assignEqualProbsToTransitions($config.states)}, $super.transitions);

    return $config;
});