60 lines
1.9 KiB
JavaScript
60 lines
1.9 KiB
JavaScript
var JobSubscriber = require('../../../batch/job_subscriber');
|
|
var assert = require('assert');
|
|
|
|
describe('batch API job subscriber', function () {
|
|
beforeEach(function () {
|
|
var self = this;
|
|
|
|
this.onMessageListener = function () {};
|
|
this.redis = {
|
|
createClient: function () {
|
|
return this;
|
|
},
|
|
subscribe: function () {
|
|
var isValidFirstArg = arguments[0] === 'batch:hosts';
|
|
self.redis.subscribeIsCalledWithValidArgs = isValidFirstArg;
|
|
},
|
|
on: function () {
|
|
if (arguments[0] === 'message') {
|
|
self.redis.onIsCalledWithValidArgs = true;
|
|
}
|
|
},
|
|
unsubscribe: function () {
|
|
var isValidFirstArg = arguments[0] === 'batch:hosts';
|
|
self.redis.unsubscribeIsCalledWithValidArgs = isValidFirstArg;
|
|
},
|
|
removeAllListeners: function () {
|
|
return this;
|
|
},
|
|
connected: true
|
|
};
|
|
this.pool = {
|
|
acquire: function (db, cb) {
|
|
cb(null, self.redis);
|
|
}
|
|
};
|
|
this.queueSeeker = {
|
|
seek: function () {
|
|
var callback = arguments[1];
|
|
|
|
callback(null, []);
|
|
}
|
|
};
|
|
|
|
this.jobSubscriber = new JobSubscriber(this.pool, this.queueSeeker);
|
|
});
|
|
|
|
it('.subscribe() should listen for incoming messages', function () {
|
|
this.jobSubscriber.subscribe(this.onMessageListener);
|
|
assert.ok(this.redis.onIsCalledWithValidArgs);
|
|
assert.ok(this.redis.subscribeIsCalledWithValidArgs);
|
|
});
|
|
|
|
it('.unsubscribe() should stop listening for incoming messages', function () {
|
|
this.jobSubscriber.subscribe(this.onMessageListener);
|
|
this.jobSubscriber.unsubscribe();
|
|
assert.ok(this.redis.unsubscribeIsCalledWithValidArgs);
|
|
});
|
|
|
|
});
|