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
|
// Test simple updates issued through mongos. Updates have different constraints through mongos,
// since shard key is immutable.
(function() {
var s = new ShardingTest({ name: "auto1", shards: 2, mongos: 1 });
s.adminCommand( { enablesharding : "test" } );
s.ensurePrimaryShard('test', 'shard0001');
// repeat same tests with hashed shard key, to ensure identical behavior
s.adminCommand( { shardcollection : "test.update0" , key : { key : 1 } } );
s.adminCommand( { shardcollection : "test.update1" , key : { key : "hashed" } } );
db = s.getDB( "test" );
for(i=0; i < 2; i++){
coll = db.getCollection("update" + i);
coll.insert({_id:1, key:1});
// these are both upserts
coll.save({_id:2, key:2});
coll.update({_id:3, key:3}, {$set: {foo: 'bar'}}, {upsert: true});
assert.eq(coll.count(), 3, "count A");
assert.eq(coll.findOne({_id:3}).key, 3 , "findOne 3 key A");
assert.eq(coll.findOne({_id:3}).foo, 'bar' , "findOne 3 foo A");
// update existing using save()
coll.save({_id:1, key:1, other:1});
// update existing using update()
coll.update({_id:2}, {key:2, other:2});
coll.update({_id:3}, {key:3, other:3});
// do a replacement-style update which queries the shard key and keeps it constant
coll.save( {_id:4, key:4} );
coll.update({key:4}, {key:4, other:4});
assert.eq( coll.find({key:4, other:4}).count() , 1 , 'replacement update error');
coll.remove( {_id:4} );
assert.eq(coll.count(), 3, "count B");
coll.find().forEach(function(x){
assert.eq(x._id, x.key, "_id == key");
assert.eq(x._id, x.other, "_id == other");
});
assert.writeError(coll.update({ _id: 1, key: 1 }, { $set: { key: 2 }}));
assert.eq(coll.findOne({_id:1}).key, 1, 'key unchanged');
assert.writeOK(coll.update({ _id: 1, key: 1 }, { $set: { foo: 2 }}));
coll.update( { key : 17 } , { $inc : { x : 5 } } , true );
assert.eq( 5 , coll.findOne( { key : 17 } ).x , "up1" );
coll.update( { key : 18 } , { $inc : { x : 5 } } , true , true );
assert.eq( 5 , coll.findOne( { key : 18 } ).x , "up2" );
// Make sure we can extract exact _id from certain queries
assert.writeOK(coll.update({_id : ObjectId()}, {$set : {x : 1}}, {multi : false}));
assert.writeOK(coll.update({_id : {$eq : ObjectId()}}, {$set : {x : 1}}, {multi : false}));
assert.writeOK(coll.update({_id : {$all : [ObjectId()]}}, {$set : {x : 1}}, {multi : false}));
assert.writeOK(coll.update({$or : [{_id : ObjectId()}]}, {$set : {x : 1}}, {multi : false}));
assert.writeOK(coll.update({$and : [{_id : ObjectId()}]}, {$set : {x : 1}}, {multi : false}));
// Invalid extraction of exact _id from query
assert.writeError(coll.update({}, {$set : {x : 1}}, {multi : false}));
assert.writeError(coll.update({_id : {$gt : ObjectId()}}, {$set : {x : 1}}, {multi : false}));
assert.writeError(coll.update({_id : {$in : [ObjectId()]}}, {$set : {x : 1}}, {multi : false}));
assert.writeError(coll.update({$or : [{_id : ObjectId()}, {_id : ObjectId()}]},
{$set : {x : 1}},
{multi : false}));
assert.writeError(coll.update({$and : [{_id : ObjectId()}, {_id : ObjectId()}]},
{$set : {x : 1}},
{multi : false}));
assert.writeError(coll.update({'_id.x' : ObjectId()}, {$set : {x : 1}}, {multi : false}));
// Make sure we can extract exact shard key from certain queries
assert.writeOK(coll.update({key : ObjectId()}, {$set : {x : 1}}, {multi : false}));
assert.writeOK(coll.update({key : {$eq : ObjectId()}}, {$set : {x : 1}}, {multi : false}));
assert.writeOK(coll.update({key : {$all : [ObjectId()]}}, {$set : {x : 1}}, {multi : false}));
assert.writeOK(coll.update({$or : [{key : ObjectId()}]}, {$set : {x : 1}}, {multi : false}));
assert.writeOK(coll.update({$and : [{key : ObjectId()}]}, {$set : {x : 1}}, {multi : false}));
// Invalid extraction of exact key from query
assert.writeError(coll.update({}, {$set : {x : 1}}, {multi : false}));
assert.writeError(coll.update({key : {$gt : ObjectId()}}, {$set : {x : 1}}, {multi : false}));
assert.writeError(coll.update({key : {$in : [ObjectId()]}}, {$set : {x : 1}}, {multi : false}));
assert.writeError(coll.update({$or : [{key : ObjectId()}, {key : ObjectId()}]},
{$set : {x : 1}},
{multi : false}));
assert.writeError(coll.update({$and : [{key : ObjectId()}, {key : ObjectId()}]},
{$set : {x : 1}},
{multi : false}));
assert.writeError(coll.update({'key.x' : ObjectId()}, {$set : {x : 1}}, {multi : false}));
// Make sure failed shard key or _id extraction doesn't affect the other
assert.writeOK(coll.update({'_id.x' : ObjectId(), key : 1}, {$set : {x : 1}}, {multi : false}));
assert.writeOK(coll.update({_id : ObjectId(), 'key.x' : 1}, {$set : {x : 1}}, {multi : false}));
}
s.stop();
})();
|