Windshaft-cartodb/test/integration/analysis-backend-limits.js

var testHelper = require('../support/test_helper');

var assert = require('assert');
var redis = require('redis');

var RedisPool = require('redis-mpool');
var cartodbRedis = require('cartodb-redis');

var AnalysisBackend = require('../../lib/cartodb/backends/analysis');

describe('analysis-backend limits', function() {

    var redisClient;
    var keysToDelete;
    var user = 'localhost';

    beforeEach(function() {
        redisClient = redis.createClient(global.environment.redis.port);
        keysToDelete = {};
        var redisPool = new RedisPool(global.environment.redis);
        this.metadataBackend = cartodbRedis({pool: redisPool});
    });

    afterEach(function(done) {
        redisClient.quit(function() {
            testHelper.deleteRedisKeys(keysToDelete, done);
        });
    });

    function withAnalysesLimits(limits, callback) {
        redisClient.SELECT(5, function(err) {
            if (err) {
                return callback(err);
            }
            var analysesLimitsKey = 'limits:analyses:' + user;
            redisClient.HMSET([analysesLimitsKey].concat(limits), function(err) {
                if (err) {
                    return callback(err);
                }
                keysToDelete[analysesLimitsKey] = 5;
                return callback();
            });
        });
    }

    it("should use limits from configuration", function(done) {
        var analysisBackend = new AnalysisBackend(this.metadataBackend, {
            limits: {
                moran: { timeout: 5000 },
                kmeans: { timeout: 5000 }
            }
        });
        analysisBackend.getAnalysesLimits(user, function(err, result) {
            assert.ok(!err, err);

            assert.ok(result.analyses.moran);
            assert.equal(result.analyses.moran.timeout, 5000);

            assert.ok(result.analyses.kmeans);
            assert.equal(result.analyses.kmeans.timeout, 5000);

            done();
        });
    });

    it("should use limits from redis", function(done) {
        var self = this;
        var limits = ['moran', 5000];

        withAnalysesLimits(limits, function(err) {
            if (err) {
                return done(err);
            }

            var analysisBackend = new AnalysisBackend(self.metadataBackend);
            analysisBackend.getAnalysesLimits(user, function(err, result) {
                assert.ok(!err, err);

                assert.ok(result.analyses.moran);
                assert.equal(result.analyses.moran.timeout, 5000);

                done();
            });
        });
    });

    it("should use limits from redis and configuration, redis takes priority", function(done) {
        var self = this;
        var limits = ['moran', 5000];

        withAnalysesLimits(limits, function(err) {
            if (err) {
                return done(err);
            }

            var analysisBackend = new AnalysisBackend(self.metadataBackend, {
                limits: {
                    moran: { timeout: 1000 }
                }
            });
            analysisBackend.getAnalysesLimits(user, function(err, result) {
                assert.ok(!err, err);

                assert.ok(result.analyses.moran);
                assert.equal(result.analyses.moran.timeout, 5000);

                done();
            });
        });
    });

    it("should use limits from redis and configuration, defaulting for values not present in redis", function(done) {
        var self = this;
        var limits = ['moran', 5000];

        withAnalysesLimits(limits, function(err) {
            if (err) {
                return done(err);
            }

            var analysisBackend = new AnalysisBackend(self.metadataBackend, {
                limits: {
                    moran: { timeout: 1000 },
                    kmeans: { timeout: 1000 }
                }
            });
            analysisBackend.getAnalysesLimits(user, function(err, result) {
                assert.ok(!err, err);

                assert.ok(result.analyses.moran);
                assert.equal(result.analyses.moran.timeout, 5000);

                assert.ok(result.analyses.kmeans);
                assert.equal(result.analyses.kmeans.timeout, 1000);

                done();
            });
        });
    });

    it("should allow to set other limits per analysis via configuration, and keep timeout from redis", function(done) {
        var self = this;
        var limits = ['aggregate-intersection', 5000];

        withAnalysesLimits(limits, function(err) {
            if (err) {
                return done(err);
            }

            var analysisBackend = new AnalysisBackend(self.metadataBackend, {
                limits: {
                    'aggregate-intersection': { timeout: 10000, maxNumberOfRows: 1e5 }
                }
            });
            analysisBackend.getAnalysesLimits(user, function(err, result) {
                assert.ok(!err, err);

                assert.ok(result.analyses['aggregate-intersection']);
                assert.equal(result.analyses['aggregate-intersection'].timeout, 5000);
                assert.equal(result.analyses['aggregate-intersection'].maxNumberOfRows, 1e5);

                done();
            });
        });
    });

});