jstests/concurrency/fsm_workloads/agg_sort.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
35
36
37
38
39

'use strict';

/**
 * agg_sort.js
 *
 * Runs an aggregation with a $match that returns half the documents followed
 * by a $sort on a field containing a random float.
 */
load('jstests/concurrency/fsm_libs/extend_workload.js');  // for extendWorkload
load('jstests/concurrency/fsm_workloads/agg_base.js');  // for $config
load('jstests/concurrency/fsm_workload_helpers/drop_utils.js');  // for dropCollections

var $config = extendWorkload(
    $config,
    function($config, $super) {

        $config.data.getOutputCollPrefix = function getOutputCollPrefix(collName) {
            return collName + '_out_agg_sort_';
        };

        $config.states.query = function query(db, collName) {
            var otherCollName = this.getOutputCollPrefix(collName) + this.tid;
            var cursor = db[collName].aggregate(
                [{$match: {flag: true}}, {$sort: {rand: 1}}, {$out: otherCollName}]);
            assertAlways.eq(0, cursor.itcount());
            assertWhenOwnColl.eq(db[collName].find().itcount() / 2,
                                 db[otherCollName].find().itcount());
        };

        $config.teardown = function teardown(db, collName, cluster) {
            $super.teardown.apply(this, arguments);

            // drop all collections with this workload's assumed-to-be-unique prefix
            // NOTE: assumes the prefix contains no special regex chars
            dropCollections(db, new RegExp('^' + this.getOutputCollPrefix(collName)));
        };

        return $config;
    });