occaecatidicta/packages/omelox/test/manager/taskManager.ts

import * as should from 'should';
// import { describe, it } from "mocha-typescript"
let taskManager = require('../../lib/common/manager/taskManager');

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

let WAIT_TIME = 200;

describe('#taskManager', function () {
  it('should add task and execute it', function (done: MochaDone) {
    let key = 'key-1';
    let fn = function (task: any) {
      taskCount++;
      task.done();
    };
    let onTimeout = function () {
      should.fail('should not timeout.', null);
    };
    let 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: MochaDone) {
    let key = 'key-1';
    let fn = function (task: any) {
      taskCount++;
    };
    let onTimeout = function () {
      timeoutCount++;
    };
    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: MochaDone) {
    let key = 'key-1';
    let fn = function (task: any) {
      taskCount++;
    };
    let onTimeout = function () {
      timeoutCount++;
    };
    let taskCount = 0;
    let 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 () {
    let key = 'key-n';
    taskManager.closeQueue(key, true);
  });
});