school/qiniu-js/lib/__tests__/pool.test.js

"use strict";

var _interopRequireDefault = require("@babel/runtime-corejs2/helpers/interopRequireDefault");

var _promise = _interopRequireDefault(require("@babel/runtime-corejs2/core-js/promise"));

var _pool = require("../pool");

var m = jest.fn();

var task = function task() {
  return new _promise["default"](function (resolve, _) {
    m();
    resolve();
  });
};

describe("test Pool for control concurrency", function () {
  var pool = new _pool.Pool(task, 2);
  test("pool.js", function () {
    var chunk = new Blob();
    var data = [{
      chunk: chunk,
      index: 0
    }, {
      chunk: chunk,
      index: 1
    }, {
      chunk: chunk,
      index: 2
    }, {
      chunk: chunk,
      index: 3
    }, {
      chunk: chunk,
      index: 4
    }, {
      chunk: chunk,
      index: 5
    }];
    return _promise["default"].all(data.map(function (value) {
      pool.enqueue(value);
      expect(pool.processing.length).toBeLessThanOrEqual(2);
    })).then(function () {
      expect(m.mock.calls.length).toBe(6);
    });
  });
});