summaryrefslogtreecommitdiff
path: root/jstests/core/batch_write_collation_estsize.js
blob: bfaf1e54bd37d5bee56d9ab7ae1521fd0c8a0f33 (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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
// Cannot implicitly shard accessed collections because of following errmsg: A single
// update/delete on a sharded collection must contain an exact match on _id or contain the shard
// key.
// @tags: [assumes_unsharded_collection, requires_multi_updates, requires_non_retryable_writes]

// Tests that the update and delete batch write operations account for the size of the collation
// specification in the write operation document.

(function() {
"use strict";

// Setup the test collection.
db.batch_write_collation_estsize.drop();
assert.commandWorked(db.batch_write_collation_estsize.insert({str: "FOO"}));

// Test updateOne bulk write operation with collation specification.
let res = db.batch_write_collation_estsize.bulkWrite([{
    updateOne: {
        filter: {str: "FOO"},
        update: {$set: {str: "BAR"}},
        collation: {
            locale: "en_US",
            caseLevel: false,
            caseFirst: "off",
            strength: 3,
            numericOrdering: false,
            alternate: "non-ignorable",
            maxVariable: "punct",
            normalization: false,
            backwards: false
        }
    }
}]);
assert.eq(1, res.matchedCount);

// Test updateMany bulk write operation with collation specification.
res = db.batch_write_collation_estsize.bulkWrite([{
    updateMany: {
        filter: {str: "BAR"},
        update: {$set: {str: "FOO"}},
        collation: {
            locale: "en_US",
            caseLevel: false,
            caseFirst: "off",
            strength: 3,
            numericOrdering: false,
            alternate: "non-ignorable",
            maxVariable: "punct",
            normalization: false,
            backwards: false
        }
    }
}]);
assert.eq(1, res.matchedCount);

// Test replaceOne bulk write operation with collation specification.
res = db.batch_write_collation_estsize.bulkWrite([{
    replaceOne: {
        filter: {str: "FOO"},
        replacement: {str: "BAR"},
        collation: {
            locale: "en_US",
            caseLevel: false,
            caseFirst: "off",
            strength: 3,
            numericOrdering: false,
            alternate: "non-ignorable",
            maxVariable: "punct",
            normalization: false,
            backwards: false
        }
    }
}]);
assert.eq(1, res.matchedCount);

// Test deleteMany bulk write operation with collation specification.
res = db.batch_write_collation_estsize.bulkWrite([{
    deleteOne: {
        filter: {str: "BAR"},
        collation: {
            locale: "en_US",
            caseLevel: false,
            caseFirst: "off",
            strength: 3,
            numericOrdering: false,
            alternate: "non-ignorable",
            maxVariable: "punct",
            normalization: false,
            backwards: false
        }
    }
}]);
assert.eq(1, res.deletedCount);

// Reinsert a document to test deleteMany bulk write operation.
assert.commandWorked(db.batch_write_collation_estsize.insert({str: "FOO"}));

// Test deleteMany bulk write operation with collation specification.
res = db.batch_write_collation_estsize.bulkWrite([{
    deleteMany: {
        filter: {str: "FOO"},
        collation: {
            locale: "en_US",
            caseLevel: false,
            caseFirst: "off",
            strength: 3,
            numericOrdering: false,
            alternate: "non-ignorable",
            maxVariable: "punct",
            normalization: false,
            backwards: false
        }
    }
}]);
assert.eq(1, res.deletedCount);
})();