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