simple-email/node_modules/bottleneck/test/promises.js

var makeTest = require('./context')
var Bottleneck = require('./bottleneck')
var assert = require('assert')

describe('Promises', function () {
  var c

  afterEach(function () {
    return c.limiter.disconnect(false)
  })

  it('Should support promises', function () {
    c = makeTest({maxConcurrent: 1, minTime: 100})

    c.limiter.submit(c.job, null, 1, 9, c.noErrVal(1, 9))
    c.limiter.submit(c.job, null, 2, c.noErrVal(2))
    c.limiter.submit(c.job, null, 3, c.noErrVal(3))
    c.pNoErrVal(c.limiter.schedule(c.promise, null, 4, 5), 4, 5)

    return c.last()
    .then(function (results) {
      c.checkResultsOrder([[1,9], [2], [3], [4,5]])
      c.checkDuration(300)
    })
  })

  it('Should pass error on failure', function () {
    var failureMessage = 'failed'
    c = makeTest({maxConcurrent: 1, minTime: 100})

    return c.limiter.schedule(c.promise, new Error(failureMessage))
    .catch(function (err) {
      c.mustEqual(err.message, failureMessage)
    })
  })

  it('Should allow non-Promise returns', function () {
    c = makeTest()
    var str = 'This is a string'

    return c.limiter.schedule(() => str)
    .then(function (x) {
      c.mustEqual(x, str)
    })
  })

  it('Should get rejected when rejectOnDrop is true', function () {
    c = makeTest({
      maxConcurrent: 1,
      minTime: 0,
      highWater: 1,
      strategy: Bottleneck.strategy.OVERFLOW,
      rejectOnDrop: true
    })
    var dropped = 0
    var caught = 0
    var p1
    var p2

    c.limiter.on('dropped', function () {
      dropped++
    })

    p1 = c.pNoErrVal(c.limiter.schedule({id: 1}, c.slowPromise, 50, null, 1), 1)
    p2 = c.pNoErrVal(c.limiter.schedule({id: 2}, c.slowPromise, 50, null, 2), 2)

    return c.limiter.schedule({id: 3}, c.slowPromise, 50, null, 3)
    .catch(function (err) {
      c.mustEqual(err.message, 'This job has been dropped by Bottleneck')
      assert(err instanceof Bottleneck.BottleneckError)
      caught++
      return Promise.all([p1, p2])
    })
    .then(c.last)
    .then(function (results) {
      c.checkResultsOrder([[1], [2]])
      c.checkDuration(100)
      c.mustEqual(dropped, 1)
      c.mustEqual(caught, 1)
    })
  })

  describe('Wrap', function () {
    it('Should wrap', function () {
      c = makeTest({maxConcurrent: 1, minTime: 100})

      c.limiter.submit(c.job, 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))

      var wrapped = c.limiter.wrap(c.promise)
      c.pNoErrVal(wrapped(null, 4), 4)

      return c.last()
      .then(function (results) {
        c.checkResultsOrder([[1], [2], [3], [4]])
        c.checkDuration(300)
      })
    })

    it('Should pass errors back', function () {
      var failureMessage = 'BLEW UP!!!'
      c = makeTest({maxConcurrent: 1, minTime: 100})

      var wrapped = c.limiter.wrap(c.promise)
      c.pNoErrVal(wrapped(null, 1), 1)
      c.pNoErrVal(wrapped(null, 2), 2)

      return wrapped(new Error(failureMessage), 3)
      .catch(function (err) {
        c.mustEqual(err.message, failureMessage)
        return c.last()
      })
      .then(function (results) {
        c.checkResultsOrder([[1], [2], [3]])
        c.checkDuration(200)
      })
    })

    it('Should allow passing options', function () {
      var failureMessage = 'BLEW UP!!!'
      c = makeTest({maxConcurrent: 1, minTime: 50})

      var wrapped = c.limiter.wrap(c.promise)
      c.pNoErrVal(wrapped(null, 1), 1)
      c.pNoErrVal(wrapped(null, 2), 2)
      c.pNoErrVal(wrapped(null, 3), 3)
      c.pNoErrVal(wrapped(null, 4), 4)
      c.pNoErrVal(wrapped.withOptions({ priority: 1 }, null, 5), 5)

      return wrapped.withOptions({ priority: 1 }, new Error(failureMessage), 6)
      .catch(function (err) {
        c.mustEqual(err.message, failureMessage)
        return c.last()
      })
      .then(function (results) {
        c.checkResultsOrder([[1], [2], [5], [6], [3], [4]])
        c.checkDuration(250)
      })
    })
  })
})