summaryrefslogtreecommitdiff
path: root/jstests/concurrency/fsm_workloads/count_limit_skip.js
blob: 999fc941f8b487c0938f3716d23acaa0334704e0 (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
45
46
47
48
49
50
51
52
53
54
55
'use strict';

/**
 * count_limit_skip.js
 *
 * Runs count with skip, limit, and a query (without using hint)
 * resulting in a collection scan and then verifies the result.
 * Each thread picks a random 'modulus' in range [5, 10]
 * and a random 'countPerNum' in range [50, 100]
 * and then inserts 'modulus * countPerNum' documents. [250, 1000]
 * Each thread inserts docs into a unique collection.
 */
load('jstests/concurrency/fsm_libs/extend_workload.js');  // for extendWorkload
load('jstests/concurrency/fsm_workloads/count.js');  // for $config
load('jstests/concurrency/fsm_workload_helpers/drop_utils.js');  // for dropCollections

var $config = extendWorkload(
    $config,
    function($config, $super) {
        $config.data.prefix = 'count_fsm_q_l_s';

        $config.data.getCount = function getCount(db, predicate) {
            var query = Object.extend({tid: this.tid}, predicate);
            return db[this.threadCollName]
                .find(query)
                .skip(this.countPerNum - 1)
                .limit(10)
                .count(true);
        };

        $config.states.init = function init(db, collName) {
            this.threadCollName = this.prefix + '_' + this.tid;

            $super.states.init.apply(this, arguments);
        };

        $config.states.count = function count(db, collName) {
            assertWhenOwnColl.eq(this.getCount(db),
                                 // having done 'skip(this.countPerNum - 1).limit(10)'
                                 10);

            var num = Random.randInt(this.modulus);
            assertWhenOwnColl.eq(this.getCount(db, {i: num}),
                                 // having done 'skip(this.countPerNum - 1).limit(10)'
                                 1);
        };

        $config.teardown = function teardown(db, collName) {
            var pattern = new RegExp('^' + this.prefix + '_\\d+$');
            dropCollections(db, pattern);
            $super.teardown.apply(this, arguments);
        };

        return $config;
    });