|
|
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]])
|
|
|
})
|
|
|
})
|
|
|
|
|
|
})
|