// 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);
}());