summaryrefslogtreecommitdiff
path: root/jstests/aggregation/sources/group/collation_group.js
blob: c497790007813e0072fcc42410842cf4eecace0c (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
// Cannot implicitly shard accessed collections because of collection existing when none expected.
// @tags: [assumes_no_implicit_collection_creation_after_drop]

// Test that the $group stage and all accumulators respect the collation.
(function() {
    "use strict";

    var coll = db.collation_group;
    coll.drop();

    var results;
    var caseInsensitive = {collation: {locale: "en_US", strength: 2}};
    var diacriticInsensitive = {collation: {locale: "en_US", strength: 1, caseLevel: true}};
    var numericOrdering = {collation: {locale: "en_US", numericOrdering: true}};
    var caseAndDiacriticInsensitive = {collation: {locale: "en_US", strength: 1}};

    assert.commandWorked(db.createCollection(coll.getName(), caseInsensitive));

    assert.writeOK(coll.insert({_id: 0, str: "A", str2: "á"}));
    assert.writeOK(coll.insert({_id: 1, str: "a", str2: "a"}));
    assert.writeOK(coll.insert({_id: 2, str: "B", str2: "é"}));
    assert.writeOK(coll.insert({_id: 3, str: "b", str2: "e"}));

    // Ensure that equality of groups respects the collation inherited from the collection default.
    assert.eq(2, coll.aggregate([{$group: {_id: "$str"}}]).itcount());

    // Ensure that equality of groups respects an explicit collation.
    assert.eq(2, coll.aggregate([{$group: {_id: "$str2"}}], diacriticInsensitive).itcount());

    // Ensure that equality of groups created by $sortByCount respects the inherited collation.
    assert.eq(2, coll.aggregate([{$sortByCount: "$str"}]).itcount());
    assert.eq(4, coll.aggregate([{$sortByCount: "$str2"}]).itcount());

    // Ensure that equality of groups created by $sortByCount respects an explicit collation.
    assert.eq(4, coll.aggregate([{$sortByCount: "$str"}], diacriticInsensitive).itcount());
    assert.eq(2, coll.aggregate([{$sortByCount: "$str2"}], diacriticInsensitive).itcount());

    // Ensure that equality of groups inside $facet stage respects the inherited collation.
    results =
        coll.aggregate([{
                $facet:
                    {facetStr: [{$group: {_id: "$str"}}], facetStr2: [{$group: {_id: "$str2"}}]}
            }])
            .toArray();
    assert.eq(1, results.length);
    assert.eq(2, results[0].facetStr.length);
    assert.eq(4, results[0].facetStr2.length);

    // Test that the $addToSet accumulator respects the inherited collation.
    results = coll.aggregate([{$group: {_id: null, set: {$addToSet: "$str"}}}]).toArray();
    assert.eq(1, results.length);
    assert.eq(2, results[0].set.length);

    // Test that the $addToSet accumulator respects an explicit collation.
    results =
        coll.aggregate([{$group: {_id: null, set: {$addToSet: "$str2"}}}], diacriticInsensitive)
            .toArray();
    assert.eq(1, results.length);
    assert.eq(2, results[0].set.length);

    // Ensure that a subexpression inside $push respects the collation.
    results = coll.aggregate(
                      [
                        {$match: {_id: 0}},
                        {$group: {_id: null, areEqual: {$push: {$eq: ["$str", "$str2"]}}}}
                      ],
                      caseAndDiacriticInsensitive)
                  .toArray();
    assert.eq(1, results.length);
    assert.eq(1, results[0].areEqual.length);
    assert.eq(true, results[0].areEqual[0]);

    // Test that the $min and $max accumulators respect the inherited collation.
    coll.drop();
    assert.commandWorked(db.createCollection(coll.getName(), numericOrdering));
    assert.writeOK(coll.insert({num: "100"}));
    assert.writeOK(coll.insert({num: "2"}));
    results = coll.aggregate([{$group: {_id: null, min: {$min: "$num"}}}]).toArray();
    assert.eq(1, results.length);
    assert.eq("2", results[0].min);
    results = coll.aggregate([{$group: {_id: null, max: {$max: "$num"}}}]).toArray();
    assert.eq(1, results.length);
    assert.eq("100", results[0].max);
})();