telegram-api-rate-limit-handler/node_modules/bottleneck/test/priority.js

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

})