|
|
- var makeTest = require('./context')
- var Bottleneck = require('./bottleneck')
- var assert = require('assert')
-
- describe('Priority', function () {
- var c
-
- afterEach(function () {
- return c.limiter.disconnect(false)
- })
-
- it('Should do basic ordering', function () {
- c = makeTest({maxConcurrent: 1, minTime: 100, rejectOnDrop: false})
-
- c.limiter.submit(c.slowJob, 50, null, 1, c.noErrVal(1))
- c.limiter.submit(c.job, null, 2, c.noErrVal(2))
- c.limiter.submit(c.job, null, 3, c.noErrVal(3))
- c.limiter.submit(c.job, null, 4, c.noErrVal(4))
- c.limiter.submit({priority: 1}, c.job, null, 5, 6, c.noErrVal(5, 6))
-
- return c.last()
- .then(function (results) {
- c.checkResultsOrder([[1], [5,6], [2] ,[3], [4]])
- c.checkDuration(400)
- })
- })
-
- it('Should support LEAK', function () {
- c = makeTest({
- maxConcurrent: 1,
- minTime: 100,
- highWater: 3,
- strategy: Bottleneck.strategy.LEAK,
- rejectOnDrop: false
- })
-
- var called = false
- c.limiter.on('dropped', function (dropped) {
- c.mustExist(dropped.task)
- c.mustExist(dropped.args)
- c.mustExist(dropped.cb)
- called = true
- })
-
- c.limiter.submit(c.slowJob, 50, null, 1, c.noErrVal(1))
- c.limiter.submit(c.job, null, 2, c.noErrVal(2))
- c.limiter.submit(c.job, null, 3, c.noErrVal(3))
- c.limiter.submit(c.job, null, 4, c.noErrVal(4))
- c.limiter.submit({priority: 2}, c.job, null, 5, c.noErrVal(5))
- c.limiter.submit({priority: 1}, c.job, null, 6, c.noErrVal(6))
- c.limiter.submit({priority: 9}, c.job, null, 7, c.noErrVal(7))
-
- return c.last({ weight: 0 })
- .then(function (results) {
- c.checkDuration(200)
- c.checkResultsOrder([[1], [6], [5]])
- c.mustEqual(called, true)
- })
- })
-
- it('Should support OVERFLOW', function () {
- c = makeTest({
- maxConcurrent: 1,
- minTime: 100,
- highWater: 2,
- strategy: Bottleneck.strategy.OVERFLOW,
- rejectOnDrop: false
- })
- var called = false
- c.limiter.on('dropped', function (dropped) {
- c.mustExist(dropped.task)
- c.mustExist(dropped.args)
- c.mustExist(dropped.cb)
- called = true
- })
-
- c.limiter.submit(c.slowJob, 50, null, 1, c.noErrVal(1))
- c.limiter.submit(c.job, null, 2, c.noErrVal(2))
- c.limiter.submit(c.job, null, 3, c.noErrVal(3))
- c.limiter.submit(c.job, null, 4, c.noErrVal(4))
- c.limiter.submit({priority: 2}, c.job, null, 5, c.noErrVal(5))
- c.limiter.submit({priority: 1}, c.job, null, 6, c.noErrVal(6))
-
- return c.limiter.submit({priority: 9}, c.job, null, 7, c.noErrVal(7))
- .then(function () {
- return c.limiter.updateSettings({ highWater: null })
- })
- .then(c.last)
- .then(function (results) {
- c.checkDuration(200)
- c.checkResultsOrder([[1], [2], [3]])
- c.mustEqual(called, true)
- })
- })
-
- it('Should support OVERFLOW_PRIORITY', function () {
- c = makeTest({
- maxConcurrent: 1,
- minTime: 100,
- highWater: 2,
- strategy: Bottleneck.strategy.OVERFLOW_PRIORITY,
- rejectOnDrop: false
- })
- var called = false
- c.limiter.on('dropped', function (dropped) {
- c.mustExist(dropped.task)
- c.mustExist(dropped.args)
- c.mustExist(dropped.cb)
- called = true
- })
-
- c.limiter.submit(c.slowJob, 50, null, 1, c.noErrVal(1))
- c.limiter.submit(c.job, null, 2, c.noErrVal(2))
- c.limiter.submit(c.job, null, 3, c.noErrVal(3))
- c.limiter.submit(c.job, null, 4, c.noErrVal(4))
- c.limiter.submit({priority: 2}, c.job, null, 5, c.noErrVal(5))
- c.limiter.submit({priority: 2}, c.job, null, 6, c.noErrVal(6))
-
- return c.limiter.submit({priority: 2}, c.job, null, 7, c.noErrVal(7))
- .then(function () {
- return c.limiter.updateSettings({highWater: null})
- })
- .then(c.last)
- .then(function (results) {
- c.checkDuration(200)
- c.checkResultsOrder([[1], [5], [6]])
- c.mustEqual(called, true)
- })
- })
-
- it('Should support BLOCK', function (done) {
- c = makeTest({
- maxConcurrent: 1,
- minTime: 100,
- highWater: 2,
- trackDoneStatus: true,
- strategy: Bottleneck.strategy.BLOCK
- })
- var called = 0
-
- c.limiter.on('dropped', function (dropped) {
- c.mustExist(dropped.task)
- c.mustExist(dropped.args)
- c.mustExist(dropped.cb)
- called++
- if (called === 3) {
- c.limiter.updateSettings({ highWater: null })
- .then(function () {
- return c.limiter.schedule(c.job, null, 8)
- })
- .catch(function (err) {
- assert(err instanceof Bottleneck.BottleneckError)
- c.mustEqual(err.message, 'This job has been dropped by Bottleneck')
- c.limiter.removeAllListeners('error')
- done()
- })
- }
- })
-
- c.limiter.submit(c.slowJob, 20, null, 1, c.noErrVal(1))
- c.limiter.submit(c.slowJob, 20, null, 2, (err) => c.mustExist(err))
- c.limiter.submit(c.slowJob, 20, null, 3, (err) => c.mustExist(err))
- c.limiter.submit(c.slowJob, 20, null, 4, (err) => c.mustExist(err))
- })
-
- it('Should have the right priority', function () {
- c = makeTest({maxConcurrent: 1, minTime: 100})
-
- c.pNoErrVal(c.limiter.schedule({priority: 6}, c.slowPromise, 50, null, 1), 1)
- c.pNoErrVal(c.limiter.schedule({priority: 5}, c.promise, null, 2), 2)
- c.pNoErrVal(c.limiter.schedule({priority: 4}, c.promise, null, 3), 3)
- c.pNoErrVal(c.limiter.schedule({priority: 3}, c.promise, null, 4), 4)
-
- return c.last()
- .then(function (results) {
- c.checkDuration(300)
- c.checkResultsOrder([[1], [4], [3], [2]])
- })
- })
-
- })
|