summaryrefslogtreecommitdiff
path: root/jstests/aggregation/bugs/server8568.js
blob: 7490b286e38ecde03321ea65cc67e8bfd819d9ab (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
// SERVER-8568: Adding $sqrt expression

// For assertErrorCode.
load('jstests/aggregation/extras/utils.js');

(function() {
    'use strict';
    var coll = db.sqrt;
    coll.drop();
    assert.writeOK(coll.insert({_id: 0}));

    // Helper for testing that op returns expResult.
    function testOp(op, expResult) {
        var pipeline = [{$project: {_id: 0, result: op}}];
        assert.eq(coll.aggregate(pipeline).toArray(), [{result: expResult}]);
    }

    // Helper for testing that op results in error with code errorCode.
    function testError(op, errorCode) {
      var pipeline = [{$project: {_id: 0, result: op}}];
      assertErrorCode(coll, pipeline, errorCode);
    }

    // Valid input: Numeric arg >= 0, null, or NaN.

    testOp({$sqrt: [100]}, 10);
    testOp({$sqrt: [0]}, 0);
    // All types converted to doubles.
    testOp({$sqrt: [NumberLong("100")]}, 10);
    // LLONG_MAX is converted to a double.
    testOp({$sqrt: [NumberLong("9223372036854775807")]}, 3037000499.97605);
    // Null inputs result in null.
    testOp({$sqrt: [null]}, null);
    // NaN inputs result in NaN.
    testOp({$sqrt: [NaN]}, NaN);

    // Invalid input: non-numeric/non-null, arg is negative.

    // Arg must be numeric or null.
    testError({$sqrt: ["string"]}, 28765);
    // Args cannot be negative.
    testError({$sqrt: [-1]}, 28714);
}());