summaryrefslogtreecommitdiff
path: root/jstests/aggregation/bugs/server6179.js
blob: 4eba802e14eaf21d9e21bbca4727b73528995fea (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
// SERVER-6179: support for two $groups in sharded agg
(function() {

var s = new ShardingTest({ name: "aggregation_multiple_group", shards: 2, mongos: 1 });
s.stopBalancer();

s.adminCommand({ enablesharding:"test" });
s.ensurePrimaryShard('test', 'shard0001');
s.adminCommand({ shardcollection: "test.data", key:{ _id: 1 } });

var d = s.getDB( "test" );

// Insert _id values 0 - 99
var N = 100;

var bulkOp = d.data.initializeOrderedBulkOp();
for(var i = 0; i < N; ++i) {
    bulkOp.insert({ _id: i, i: i%10 });
}
bulkOp.execute();

// Split the data into 3 chunks
s.adminCommand( { split:"test.data", middle:{ _id:33 } } );
s.adminCommand( { split:"test.data", middle:{ _id:66 } } );

// Migrate the middle chunk to another shard
s.adminCommand({ movechunk: "test.data",
                 find: { _id: 50 },
                 to: s.getOther(s.getPrimaryShard("test")).name });

// Check that we get results rather than an error
var result = d.data.aggregate({$group: {_id: '$_id', i: {$first: '$i'}}},
                              {$group: {_id: '$i', avg_id: {$avg: '$_id'}}},
                              {$sort: {_id: 1}}).toArray();
expected = [
    {
        "_id" : 0,
        "avg_id" : 45
    },
    {
        "_id" : 1,
        "avg_id" : 46
    },
    {
        "_id" : 2,
        "avg_id" : 47
    },
    {
        "_id" : 3,
        "avg_id" : 48
    },
    {
        "_id" : 4,
        "avg_id" : 49
    },
    {
        "_id" : 5,
        "avg_id" : 50
    },
    {
        "_id" : 6,
        "avg_id" : 51
    },
    {
        "_id" : 7,
        "avg_id" : 52
    },
    {
        "_id" : 8,
        "avg_id" : 53
    },
    {
        "_id" : 9,
        "avg_id" : 54
    }
];

assert.eq(result, expected);

s.stop();

})();