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
|
/**
* Tests that writes without shard key for only a single targetable shard succeed.
*
* @tags: [
* requires_sharding,
* requires_fcv_70,
* uses_transactions,
* uses_multi_shard_transaction,
* featureFlagUpdateOneWithoutShardKey,
* ]
*/
(function() {
"use strict";
load("jstests/libs/feature_flag_util.js");
load("jstests/sharding/updateOne_without_shard_key/libs/write_without_shard_key_test_util.js");
// Make sure we're testing with no implicit session.
TestData.disableImplicitSessions = true;
// 2 shards single node, 1 mongos, 1 config server 3-node.
const st = new ShardingTest({});
const dbName = "testDb";
const collName = "testColl";
const nss = dbName + "." + collName;
const splitPoint = 0;
const docsToInsert =
[{_id: 0, x: -2, y: 1}, {_id: 1, x: -1, y: 1}, {_id: 2, x: 1, y: 1}, {_id: 3, x: 2, y: 1}];
// Sets up a 2 shard cluster using 'x' as a shard key where one shard owns all chunks.
WriteWithoutShardKeyTestUtil.setupShardedCollection(st, nss, {x: 1}, [{x: splitPoint}], []);
let testCases = [
{
logMessage: "Running single shard test for updateOne.",
docsToInsert: docsToInsert,
cmdObj: {
update: collName,
updates: [{q: {y: 1}, u: {$set: {z: 3}}}],
},
expectedMods: [
{'z': 3},
],
expectedResponse: {n: 1, nModified: 1},
options: [{ordered: true}, {ordered: false}],
dbName: dbName,
collName: collName,
opType: WriteWithoutShardKeyTestUtil.OperationType.updateOne,
},
{
logMessage: "Running single shard test for findAndModify.",
docsToInsert: docsToInsert,
cmdObj: {
findAndModify: collName,
query: {y: 1},
update: {$set: {z: 4}},
},
expectedMods: [
{'z': 4},
],
expectedResponse: {lastErrorObject: {n: 1, updatedExisting: true}},
dbName: dbName,
collName: collName,
opType: WriteWithoutShardKeyTestUtil.OperationType.findAndModifyUpdate,
},
{
logMessage: "Running single shard test for deleteOne.",
docsToInsert: docsToInsert,
cmdObj: {
delete: collName,
deletes: [{q: {y: 1}, limit: 1}],
},
expectedResponse: {n: 1},
options: [{ordered: true}, {ordered: false}],
dbName: dbName,
collName: collName,
opType: WriteWithoutShardKeyTestUtil.OperationType.deleteOne,
}
];
let conn = WriteWithoutShardKeyTestUtil.getClusterConnection(
st, WriteWithoutShardKeyTestUtil.Configurations.sessionRetryableWrite);
testCases.forEach(testCase => {
WriteWithoutShardKeyTestUtil.runTestWithConfig(
conn,
testCase,
WriteWithoutShardKeyTestUtil.Configurations.sessionRetryableWrite,
testCase.opType);
});
const configurations = [
WriteWithoutShardKeyTestUtil.Configurations.noSession,
WriteWithoutShardKeyTestUtil.Configurations.sessionNotRetryableWrite,
WriteWithoutShardKeyTestUtil.Configurations.sessionRetryableWrite,
WriteWithoutShardKeyTestUtil.Configurations.transaction
];
configurations.forEach(config => {
let conn = WriteWithoutShardKeyTestUtil.getClusterConnection(st, config);
testCases.forEach(testCase => {
WriteWithoutShardKeyTestUtil.runTestWithConfig(conn, testCase, config, testCase.opType);
});
});
st.stop();
})();
|
