pomelo-upgrade/Test/manager/taskManager.js

网易Pomelo升级版本

'use strict';
const should = require('should');
const taskManager = require('../../lib/common/manager/taskManager');

// set timeout for test
taskManager.timeout = 100;

const WAIT_TIME = 200;

describe('#taskManager',function() {
    it('should add task and execute it',function(done) {
        const key = 'key-1';
        const fn = function(task) {
            taskCount++;
            task.done();
        };
        const onTimeout = function() {
            should.fail('should not timeout.');
        };
        var taskCount = 0;

        taskManager.addTask(key, fn, onTimeout);

        setTimeout(function() {
            taskCount.should.equal(1);
            done();
        }, WAIT_TIME);
    });

    it('should fire timeout callback if task timeout',function(done) {
        const key = 'key-1';
        const fn = function(task) {
            taskCount++;
        };
        const onTimeout = function() {
            timeoutCount++;
            // taskCount++;
        };
        let taskCount = 0;
        let timeoutCount = 0;

        taskManager.addTask(key, fn, onTimeout);

        setTimeout(function() {
            taskCount.should.equal(1);
            timeoutCount.should.equal(1);
            done();
        }, WAIT_TIME);
    });

    it('should not fire timeout after close the task',function(done) {
        const key = 'key-1';
        const fn = function(task) {
            taskCount++;
        };
        const onTimeout = function() {
            timeoutCount++;
        };
        var taskCount = 0;
        var timeoutCount = 0;

        taskManager.addTask(key, fn, onTimeout);

        process.nextTick(function() {
            taskManager.closeQueue(key, true);

            setTimeout(function() {
                taskCount.should.equal(1);
                timeoutCount.should.equal(0);
                done();
            }, WAIT_TIME);
        });
    });

    it('should be ok to remove a queue not exist',function() {
        const key = 'key-n';
        taskManager.closeQueue(key, true);
    });
});