CartoDB-SQL-API/test/acceptance/batch/queued-jobs-limit.test.js

require('../../helper');

var assert = require('../../support/assert');
var redisUtils = require('../../support/redis_utils');
var batchFactory = require('../../../batch/index');
var metadataBackend = require('cartodb-redis')({ pool: redisUtils.getPool() });
var TestClient = require('../../support/test-client');

describe('max queued jobs', function() {

    before(function(done) {
        this.batch_max_queued_jobs = global.settings.batch_max_queued_jobs;
        global.settings.batch_max_queued_jobs = 1;
        this.server = require('../../../app/server')();
        this.testClient = new TestClient();
        this.testClient.getResult(
            'drop table if exists max_queued_jobs_inserts; create table max_queued_jobs_inserts (status numeric)',
            done
        );
    });

    after(function (done) {
        var self = this;
        global.settings.batch_max_queued_jobs = this.batch_max_queued_jobs;
        var batch = batchFactory(metadataBackend, redisUtils.getPool());
        batch.start();
        batch.on('ready', function() {
            // this is not ideal as the first job might not be committed yet
            setTimeout(function() {
                batch.stop(function() {
                    self.testClient.getResult('select count(*) from max_queued_jobs_inserts', function(err, rows) {
                        assert.ok(!err);
                        assert.equal(rows[0].count, 1);

                        redisUtils.clean('batch:*', done);
                    });
                });
            }, 100);
        });
    });

    function createJob(server, status, callback) {
        assert.response(
            server,
            {
                url: '/api/v2/sql/job?api_key=1234',
                headers: {
                    host: 'vizzuality.cartodb.com',
                    'Content-Type': 'application/json'
                },
                method: 'POST',
                data: JSON.stringify({
                    query: "insert into max_queued_jobs_inserts values (1)"
                })
            },
            {
                status: status
            },
            function(err, res) {
                if (err) {
                    return callback(err);
                }

                return callback(null, JSON.parse(res.body));
            }
        );
    }

    it('POST /api/v2/sql/job should respond with 200 and the created job', function (done) {
        var self = this;
        createJob(this.server, 201, function(err) {
            assert.ok(!err);

            createJob(self.server, 400, function(err, res) {
                assert.ok(!err);
                assert.equal(res.error[0], "Failed to create job. Max number of jobs (" +
                    global.settings.batch_max_queued_jobs + ") queued reached");
                done();
            });
        });
    });

});