Ghost/core/test/unit/scheduling/SchedulingDefault_spec.js

/*globals describe, it, before, afterEach*/
var config = require(__dirname + '/../../../server/config'),
    moment = require('moment'),
    _ = require('lodash'),
    should = require('should'),
    express = require('express'),
    bodyParser = require('body-parser'),
    http = require('http'),
    sinon = require('sinon');

describe('Scheduling Default Adapter', function () {
    var scope = {};

    before(function () {
        scope.SchedulingDefault = require(config.paths.corePath + '/server/scheduling/SchedulingDefault');
        scope.adapter = new scope.SchedulingDefault();
    });

    afterEach(function () {
        scope.adapter.allJobs = {};
    });

    describe('success', function () {
        it('addJob (schedule)', function () {
            sinon.stub(scope.adapter, 'run');
            sinon.stub(scope.adapter, '_execute');

            var dates = [
                moment().add(1, 'day').subtract(30, 'seconds').toDate(),
                moment().add(7, 'minutes').toDate(),

                // over 10minutes offset
                moment().add(12, 'minutes').toDate(),
                moment().add(20, 'minutes').toDate(),
                moment().add(15, 'minutes').toDate(),
                moment().add(15, 'minutes').add(10, 'seconds').toDate(),
                moment().add(15, 'minutes').subtract(30, 'seconds').toDate(),
                moment().add(50, 'seconds').toDate()
            ];

            dates.forEach(function (time) {
                scope.adapter._addJob({
                    time: time,
                    url: 'something'
                });
            });

            // 2 jobs get immediately executed
            should.not.exist(scope.adapter.allJobs[moment(dates[1]).valueOf()]);
            should.not.exist(scope.adapter.allJobs[moment(dates[7]).valueOf()]);
            scope.adapter._execute.calledTwice.should.eql(true);

            Object.keys(scope.adapter.allJobs).length.should.eql(dates.length - 2);
            Object.keys(scope.adapter.allJobs).should.eql([
                moment(dates[2]).valueOf().toString(),
                moment(dates[6]).valueOf().toString(),
                moment(dates[4]).valueOf().toString(),
                moment(dates[5]).valueOf().toString(),
                moment(dates[3]).valueOf().toString(),
                moment(dates[0]).valueOf().toString()
            ]);

            scope.adapter.run.restore();
            scope.adapter._execute.restore();
        });

        it('run', function (done) {
            var timestamps = _.map(_.range(1000), function (i) {
                    return moment().add(i, 'seconds').valueOf();
                }),
                allJobs = {};

            sinon.stub(scope.adapter, '_execute', function (nextJobs) {
                Object.keys(nextJobs).length.should.eql(182);
                Object.keys(scope.adapter.allJobs).length.should.eql(1000 - 182);
                scope.adapter._execute.restore();
                done();
            });

            timestamps.forEach(function (timestamp) {
                allJobs[timestamp] = [{url: 'xxx'}];
            });

            scope.adapter.allJobs = allJobs;
            scope.adapter.runTimeoutInMs = 1000;
            scope.adapter.offsetInMinutes = 2;
            scope.adapter.run();
        });

        it('execute', function (done) {
            var pinged = 0,
                jobs = 3,
                timestamps = _.map(_.range(jobs), function (i) {
                    return moment().add(1, 'seconds').add(i * 100, 'milliseconds').valueOf();
                }),
                nextJobs = {};

            sinon.stub(scope.adapter, 'run');
            sinon.stub(scope.adapter, '_pingUrl', function () {
                pinged = pinged + 1;
            });

            timestamps.forEach(function (timestamp) {
                nextJobs[timestamp] = [{url: 'xxx'}];
            });

            scope.adapter._execute(nextJobs);

            (function retry() {
                if (pinged !== jobs) {
                    return setTimeout(retry, 100);
                }

                scope.adapter.run.restore();
                scope.adapter._pingUrl.restore();
                done();
            })();
        });

        it('delete job (unschedule)', function (done) {
            sinon.stub(scope.adapter, 'run');
            sinon.stub(scope.adapter, '_pingUrl');

            // add 3 jobs to delete
            var jobs = {};
            jobs[moment().add(500, 'milliseconds').valueOf()] = [{url: '/first', time: 1234}];
            jobs[moment().add(550, 'milliseconds').valueOf()] = [{url: '/first', time: 1235}];
            jobs[moment().add(600, 'milliseconds').valueOf()] = [{url: '/second', time: 1236}];

            _.map(jobs, function (value) {
                scope.adapter._deleteJob(value[0]);
            });

            // add another, which will be pinged
            jobs[moment().add(650, 'milliseconds').valueOf()] = [{url: '/third', time: 1237}];

            // simulate execute is called
            scope.adapter._execute(jobs);

            (function retry() {
                if (!scope.adapter._pingUrl.called) {
                    return setTimeout(retry, 10);
                }

                Object.keys(scope.adapter.deletedJobs).length.should.eql(0);
                scope.adapter._pingUrl.calledOnce.should.eql(true);

                scope.adapter.run.restore();
                scope.adapter._pingUrl.restore();
                done();
            })();
        });

        it('pingUrl (PUT)', function (done) {
            var app = express(),
                server = http.createServer(app),
                wasPinged = false,
                reqBody;

            app.use(bodyParser.json());

            app.put('/ping', function (req, res) {
                wasPinged = true;
                reqBody = req.body;
                res.sendStatus(200);
            });

            server.listen(1111);

            scope.adapter._pingUrl({
                url: 'http://localhost:1111/ping',
                time: moment().add(1, 'second').valueOf(),
                extra: {
                    httpMethod: 'PUT'
                }
            });

            (function retry() {
                if (wasPinged) {
                    should.not.exist(reqBody.force);
                    return server.close(done);
                }

                setTimeout(retry, 100);
            })();
        });

        it('pingUrl (GET)', function (done) {
            var app = express(),
                server = http.createServer(app),
                wasPinged = false,
                reqQuery;

            app.get('/ping', function (req, res) {
                wasPinged = true;
                reqQuery = req.query;
                res.sendStatus(200);
            });

            server.listen(1111);

            scope.adapter._pingUrl({
                url: 'http://localhost:1111/ping',
                time: moment().add(1, 'second').valueOf(),
                extra: {
                    httpMethod: 'GET'
                }
            });

            (function retry() {
                if (wasPinged) {
                    should.not.exist(reqQuery.force);
                    return server.close(done);
                }

                setTimeout(retry, 100);
            })();
        });

        it('pingUrl (PUT, and detect publish in the past)', function (done) {
            var app = express(),
                server = http.createServer(app),
                wasPinged = false,
                reqBody;

            app.use(bodyParser.json());

            app.put('/ping', function (req, res) {
                wasPinged = true;
                reqBody = req.body;
                res.sendStatus(200);
            });

            server.listen(1111);

            scope.adapter._pingUrl({
                url: 'http://localhost:1111/ping',
                time: moment().subtract(10, 'minutes').valueOf(),
                extra: {
                    httpMethod: 'PUT'
                }
            });

            (function retry() {
                if (wasPinged) {
                    should.exist(reqBody.force);
                    return server.close(done);
                }

                setTimeout(retry, 100);
            })();
        });

        it('pingUrl (GET, and detect publish in the past)', function (done) {
            var app = express(),
                server = http.createServer(app),
                wasPinged = false,
                reqQuery;

            app.get('/ping', function (req, res) {
                wasPinged = true;
                reqQuery = req.query;
                res.sendStatus(200);
            });

            server.listen(1111);

            scope.adapter._pingUrl({
                url: 'http://localhost:1111/ping',
                time: moment().subtract(10, 'minutes').valueOf(),
                extra: {
                    httpMethod: 'GET'
                }
            });

            (function retry() {
                if (wasPinged) {
                    should.exist(reqQuery.force);
                    return server.close(done);
                }

                setTimeout(retry, 100);
            })();
        });
    });
});