setup integration test

node_modules/bottleneck/test/priority.js

@@ -0,0 +1,184 @@
+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})
+
+    return Promise.all([
+      c.pNoErrVal(c.limiter.schedule(c.slowPromise, 50, null, 1), 1),
+      c.pNoErrVal(c.limiter.schedule(c.promise, null, 2), 2),
+      c.pNoErrVal(c.limiter.schedule({priority: 1}, c.promise, null, 5, 6), 5, 6),
+      c.pNoErrVal(c.limiter.schedule(c.promise, null, 3), 3),
+      c.pNoErrVal(c.limiter.schedule(c.promise, null, 4), 4)
+    ])
+    .then(function () {
+      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.promise)
+      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.promise)
+      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.promise)
+      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.promise)
+      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]])
+    })
+  })
+
+})