78 lines
2.6 KiB
JavaScript
78 lines
2.6 KiB
JavaScript
require('../../helper');
|
|
|
|
var assert = require('../../support/assert');
|
|
var BatchTestClient = require('../../support/batch-test-client');
|
|
|
|
describe('batch work in progress endpoint happy cases', function() {
|
|
|
|
before(function() {
|
|
this.batchTestClient = new BatchTestClient();
|
|
});
|
|
|
|
after(function(done) {
|
|
this.batchTestClient.drain(done);
|
|
});
|
|
|
|
function jobPayload(query) {
|
|
return {
|
|
query: query
|
|
};
|
|
}
|
|
|
|
it('should get a list of work in progress jobs group by user', function (done) {
|
|
var self = this;
|
|
var user = 'vizzuality';
|
|
var queries = ['select pg_sleep(0.5)'];
|
|
var payload = jobPayload(queries);
|
|
|
|
self.batchTestClient.createJob(payload, function(err, jobResult) {
|
|
if (err) {
|
|
return done(err);
|
|
}
|
|
|
|
setTimeout(function () {
|
|
self.batchTestClient.getWorkInProgressJobs(function (err, workInProgressJobs) {
|
|
if (err) {
|
|
return done(err);
|
|
}
|
|
assert.ok(Array.isArray(workInProgressJobs[user]));
|
|
assert.ok(workInProgressJobs[user].length >= 1);
|
|
for (var i = 0; i < workInProgressJobs[user].length; i++) {
|
|
if (workInProgressJobs[user][i] === jobResult.job.job_id) {
|
|
return done();
|
|
}
|
|
}
|
|
});
|
|
}, 100);
|
|
});
|
|
});
|
|
|
|
it('should get a list of work in progress jobs w/o the finished ones', function (done) {
|
|
var self = this;
|
|
var user = 'vizzuality';
|
|
var queries = ['select pg_sleep(0.1)'];
|
|
var payload = jobPayload(queries);
|
|
|
|
self.batchTestClient.createJob(payload, function(err, jobResult) {
|
|
if (err) {
|
|
return done(err);
|
|
}
|
|
setTimeout(function () {
|
|
self.batchTestClient.getWorkInProgressJobs(function (err, workInProgressJobs) {
|
|
if (err) {
|
|
return done(err);
|
|
}
|
|
assert.ok(Array.isArray(workInProgressJobs[user]));
|
|
assert.ok(workInProgressJobs[user].length >= 1);
|
|
for (var i = 0; i < workInProgressJobs[user].length; i++) {
|
|
if (workInProgressJobs[user][i] === jobResult.job.job_id) {
|
|
return done(new Error('Job should not be in work-in-progress list'));
|
|
}
|
|
}
|
|
return done();
|
|
});
|
|
}, 200);
|
|
});
|
|
});
|
|
});
|