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

/**
 * update_multifield_multiupdate.js
 *
 * Does updates that affect multiple fields on multiple documents.
 * The collection has an index for each field, and a multikey index for all fields.
 */
load('jstests/concurrency/fsm_libs/extend_workload.js');         // for extendWorkload
load('jstests/concurrency/fsm_workloads/update_multifield.js');  // for $config

// For isMongod and recordIdCanChangeOnUpdate.
load('jstests/concurrency/fsm_workload_helpers/server_types.js');

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

                       $config.data.multi = true;

                       $config.data.assertResult = function(res, db, collName, query) {
                           assertAlways.eq(0, res.nUpserted, tojson(res));

                           if (isMongod(db)) {
                               if (!recordIdCanChangeOnUpdate(db)) {
                                   // If a document's RecordId cannot change, then we should not
                                   // have updated any document more than once, since the update
                                   // stage internally de-duplicates based on RecordId.
                                   assertWhenOwnColl.lte(this.numDocs, res.nMatched, tojson(res));
                               } else {
                                   // If RecordIds can change, then there are no guarantees on how
                                   // many documents were updated.
                                   assertAlways.gte(res.nMatched, 0, tojson(res));
                               }
                           } else {  // mongos
                               assertAlways.gte(res.nMatched, 0, tojson(res));
                           }

                           if (db.getMongo().writeMode() === 'commands') {
                               assertWhenOwnColl.eq(res.nMatched, res.nModified, tojson(res));
                           }

                           var docs = db[collName].find().toArray();
                           docs.forEach(function(doc) {
                               assertWhenOwnColl.eq('number', typeof doc.z);
                               assertWhenOwnColl.gt(doc.z, 0);
                           });
                       };

                       return $config;
                   });