84 lines
2.8 KiB
JavaScript
84 lines
2.8 KiB
JavaScript
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();
|
|
});
|
|
});
|
|
});
|
|
|
|
});
|