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

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

if (process.env.DATASTORE === 'redis') {
  describe('node_redis-only', function () {
    var c

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

    it('Should accept node_redis lib override', function () {
      c = makeTest({
        maxConcurrent: 2,
        Redis,
        clientOptions: {}
      })

      c.mustEqual(c.limiter.datastore, 'redis')
    })

    it('Should accept existing connections', function () {
      var connection = new Bottleneck.RedisConnection()
      connection.id = 'super-connection'
      c = makeTest({
        minTime: 50,
        connection
      })

      c.pNoErrVal(c.limiter.schedule(c.promise, null, 1), 1)
      c.pNoErrVal(c.limiter.schedule(c.promise, null, 2), 2)

      return c.last()
        .then(function (results) {
          c.checkResultsOrder([[1], [2]])
          c.checkDuration(50)
          c.mustEqual(c.limiter.connection.id, 'super-connection')
          c.mustEqual(c.limiter.datastore, 'redis')

          return c.limiter.disconnect()
        })
        .then(function () {
        // Shared connections should not be disconnected by the limiter
          c.mustEqual(c.limiter.clients().client.ready, true)
          return connection.disconnect()
        })
    })

    it('Should accept existing redis clients', function () {
      var client = Redis.createClient()
      client.id = 'super-client'

      var connection = new Bottleneck.RedisConnection({ client })
      connection.id = 'super-connection'
      c = makeTest({
        minTime: 50,
        connection
      })

      c.pNoErrVal(c.limiter.schedule(c.promise, null, 1), 1)
      c.pNoErrVal(c.limiter.schedule(c.promise, null, 2), 2)

      return c.last()
        .then(function (results) {
          c.checkResultsOrder([[1], [2]])
          c.checkDuration(50)
          c.mustEqual(c.limiter.clients().client.id, 'super-client')
          c.mustEqual(c.limiter.connection.id, 'super-connection')
          c.mustEqual(c.limiter.datastore, 'redis')

          return c.limiter.disconnect()
        })
        .then(function () {
        // Shared connections should not be disconnected by the limiter
          c.mustEqual(c.limiter.clients().client.ready, true)
          return connection.disconnect()
        })
    })

    it('Should trigger error events on the shared connection', function (done) {
      var connection = new Bottleneck.RedisConnection({
        clientOptions: {
          port: 1
        }
      })
      connection.on('error', function (err) {
        c.mustEqual(c.limiter.datastore, 'redis')
        connection.disconnect()
        done()
      })

      c = makeTest({ connection })
      c.limiter.on('error', function (err) {
        done(err)
      })
    })
  })
}