Windshaft-cartodb/test/acceptance/dataviews/formula-test.js

'use strict';

require('../../support/test-helper');
var assert = require('../../support/assert');
var TestClient = require('../../support/test-client');

function createMapConfig(layers, dataviews, analysis) {
    return {
        version: '1.5.0',
        layers: layers,
        dataviews: dataviews || {},
        analyses: analysis || []
    };
}

function getMapConfig(operation, lastNumber) {
    return createMapConfig([
        {
            "type": "cartodb",
            "options": {
                "source": {
                    "id": "a0"
                },
                "cartocss": "#points { marker-width: 10; marker-fill: red; }",
                "cartocss_version": "2.3.0"
            }
        }
    ],
    {
        val_formula: {
            source: {
                id: 'a0'
            },
            type: 'formula',
            options: {
                column: 'val',
                operation: operation
            }
        }
    },
    [
        {
            id: "a0",
            type: "source",
            params: {
                query: `
                    SELECT
                        null::geometry the_geom_webmercator,
                        CASE
                            WHEN x % 5 = 1 THEN 'infinity'::float
                            WHEN x % 5 = 2 THEN '-infinity'::float
                            WHEN x % 5 = 3 THEN 'NaN'::float
                            WHEN x % 5 = 4 THEN NULL
                            ELSE x
                        END AS val
                    FROM generate_series(1, ${lastNumber}) x
                `
            }
        }
    ]);
}

describe('formula-dataview: special float values', function() {
    describe('easy numbers', function() { // not needed, but I keep it here to help human calc
        afterEach(function(done) {
            if (this.testClient) {
                this.testClient.drain(done);
            } else {
                done();
            }
        });

        const lastNumber = 10;

        [
            {operation: 'count', result: 2},
            {operation: 'avg', result: 7.5},
            {operation: 'sum', result: 15},
            {operation: 'min', result: 5},
            {operation: 'max', result: 10}
        ]
        .forEach(operationData => {
            it(operationData.operation, function(done) {
                this.testClient = new TestClient(getMapConfig(operationData.operation, lastNumber), 1234);
                this.testClient.getDataview('val_formula', {}, function(err, dataview) {
                    assert.ok(!err, err);
                    assert.deepStrictEqual(dataview, {
                        type: 'formula',
                        operation: operationData.operation,
                        result: operationData.result,
                        nulls: lastNumber / 5,
                        nans: lastNumber / 5,
                        infinities: 2 * lastNumber / 5
                    });
                    done();
                });
            });
        });
    });

    describe('bigger numbers', function() {
        afterEach(function(done) {
            if (this.testClient) {
                this.testClient.drain(done);
            } else {
                done();
            }
        });

        const lastNumber = 1000;

        [
            {operation: 'count', result: 200},
            {operation: 'avg', result: 502.5},
            {operation: 'sum', result: 100500},
            {operation: 'min', result: 5},
            {operation: 'max', result: 1000}
        ]
        .forEach(operationData => {
            it(operationData.operation, function(done) {
                this.testClient = new TestClient(getMapConfig(operationData.operation, lastNumber), 1234);
                this.testClient.getDataview('val_formula', {}, function(err, dataview) {
                    assert.ok(!err, err);
                    assert.deepStrictEqual(dataview, {
                        type: 'formula',
                        operation: operationData.operation,
                        result: operationData.result,
                        nulls: lastNumber / 5,
                        nans: lastNumber / 5,
                        infinities: 2 * lastNumber / 5
                    });
                    done();
                });
            });
        });
    });
});