CartoDB-SQL-API/test/unit/batch/job_publisher.js

var JobPublisher = require('../../../batch/job_publisher');
var assert = require('assert');

describe('batch API job publisher', function () {
    beforeEach(function () {
        var self = this;
        this.host = 'irrelevantHost';
        this.redis = {
            createClient: function () {
                return this;
            },
            publish: function () {
                var isValidFirstArg = arguments[0] === 'batch:hosts';
                var isValidSecondArg = arguments[1] === self.host;
                self.redis.publishIsCalledWithValidArgs = isValidFirstArg && isValidSecondArg;
            },
            on: function () {},
            ping: function (cb) {
                cb();
            }
        };
        this.pool = {
            acquire: function (db, cb) {
                cb(null, self.redis);
            }
        };

        this.jobPublisher = new JobPublisher(this.pool);
    });

    it('.publish() should publish new messages', function () {
        this.jobPublisher.publish(this.host);
        assert.ok(this.redis.publishIsCalledWithValidArgs);
    });

});