summaryrefslogtreecommitdiff
path: root/jstests/concurrency/fsm_workloads/update_where.js
blob: 614cbc860931e374d9f6589fa0e1cbdaaed19d25 (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
'use strict';

/**
 * update_where.js
 *
 * Bulk inserts documents in batches of 100, randomly selects ~1/10th of documents inserted by the
 * thread and updates them. Also queries by the thread that created the documents to verify counts.
 */

load('jstests/concurrency/fsm_libs/extend_workload.js');            // for extendWorkload
load('jstests/concurrency/fsm_workloads/indexed_insert_where.js');  // for $config

var $config = extendWorkload($config, function($config, $super) {
    $config.data.randomBound = 10;
    $config.data.generateDocumentToInsert = function generateDocumentToInsert() {
        return {tid: this.tid, x: Random.randInt(this.randomBound)};
    };

    $config.states.update = function update(db, collName) {
        var res = db[collName].update(
            // Server-side JS does not support Random.randInt, so use Math.floor/random instead
            {
              $where: 'this.x === Math.floor(Math.random() * ' + this.randomBound + ') ' +
                  '&& this.tid === ' + this.tid
            },
            {$set: {x: Random.randInt(this.randomBound)}},
            {multi: true});
        assertAlways.writeOK(res);

        if (db.getMongo().writeMode() === 'commands') {
            assertWhenOwnColl.gte(res.nModified, 0);
            assertWhenOwnColl.lte(res.nModified, this.insertedDocuments);
        }
    };

    $config.transitions = {
        insert: {insert: 0.2, update: 0.4, query: 0.4},
        update: {insert: 0.4, update: 0.2, query: 0.4},
        query: {insert: 0.4, update: 0.4, query: 0.2}
    };

    $config.setup = function setup(db, collName, cluster) {
        /* no-op to prevent index from being created */
    };

    return $config;
});