This commit is contained in:
Daniel García Aubert 2017-04-03 12:59:23 +02:00
parent 24ff1cf808
commit e4e6207565

View File

@ -0,0 +1,72 @@
require('../../helper');
var assert = require('../../support/assert');
var redisUtils = require('../../support/redis_utils');
var batchFactory = require('../../../batch/index');
var JobPublisher = require('../../../batch/pubsub/job-publisher');
var JobQueue = require('../../../batch/job_queue');
var JobBackend = require('../../../batch/job_backend');
var JobService = require('../../../batch/job_service');
var UserDatabaseMetadataService = require('../../../batch/user_database_metadata_service');
var JobCanceller = require('../../../batch/job_canceller');
var metadataBackend = require('cartodb-redis')({ pool: redisUtils.getPool() });
var queueDiscover = require('../../../batch/pubsub/queue-discover');
describe('batch startup', function() {
var dbInstance = 'localhost';
var username = 'vizzuality';
var pool = redisUtils.getPool();
var jobPublisher = new JobPublisher(pool);
var jobQueue = new JobQueue(metadataBackend, jobPublisher);
var jobBackend = new JobBackend(metadataBackend, jobQueue);
var userDatabaseMetadataService = new UserDatabaseMetadataService(metadataBackend);
var jobCanceller = new JobCanceller(userDatabaseMetadataService);
var jobService = new JobService(jobBackend, jobCanceller);
var batch = batchFactory(metadataBackend, pool);
after(function (done) {
batch.stop();
redisUtils.clean('batch:*', done);
});
function createJob(sql, done) {
var data = {
user: username,
query: sql,
host: dbInstance
};
jobService.create(data, function (err, job) {
if (err) {
return done(err);
}
done(null, job.serialize());
});
}
it('should feed queue index at startup', function (done) {
createJob('select pg_sleep(3)', function (err) {
if (err) {
return done(err);
}
batch.start();
batch.on('ready', function () {
var onDiscoveredQueue = function () {};
queueDiscover(pool, onDiscoveredQueue, function (err, client, queues) {
if (err) {
done(err);
}
assert.equal(queues.length, 1);
assert.equal(queues[0], 'vizzuality');
done();
});
});
});
});
});