kafka-node/test/test.consumerGroupRecovery.js

Client for Apache Kafka v0.8+

'use strict';

var should = require('should');
var _ = require('lodash');
var sinon = require('sinon');
var ConsumerGroupRecovery = require('../lib/consumerGroupRecovery');
var GroupCoordinatorNotAvailable = require('../lib/errors/GroupCoordinatorNotAvailableError');
var GroupLoadInProgress = require('../lib/errors/GroupLoadInProgressError');
var EventEmitter = require('events').EventEmitter;

describe('ConsumerGroupRecovery', function () {
  var consumerGroupRecovery, fakeClient;

  beforeEach(function () {
    fakeClient = new EventEmitter();
    Object.assign(fakeClient, {
      stopHeartbeats: sinon.stub(),
      options: {
        retries: 10,
        retryFactor: 1.8
      }
    });
    consumerGroupRecovery = new ConsumerGroupRecovery(fakeClient);
  });

  describe('#tryToRecoverFrom', function () {
    it('should emit error on the client when calling trying to recover from a unknown error', function (done) {
      var testError = new Error('My test error');

      fakeClient.once('error', function (error) {
        error.should.be.eql(testError);
        done();
      });

      consumerGroupRecovery.tryToRecoverFrom(testError, 'test');
      sinon.assert.calledOnce(fakeClient.stopHeartbeats);
      fakeClient.ready.should.be.false;
      consumerGroupRecovery.lastError.should.be.eql(testError);
    });
  });

  describe('#getRetryTimeout', function () {
    it('should reset backoff timeouts when calling with different error', function () {
      var error = new GroupCoordinatorNotAvailable();
      error.errorCode = 15;

      var differentError = new GroupLoadInProgress();
      differentError.errorCode = 14;

      consumerGroupRecovery.options.retries = 5;

      var results = _.times(3, function () {
        var ret = consumerGroupRecovery.getRetryTimeout(error);
        consumerGroupRecovery.lastError = error;
        return ret;
      });

      consumerGroupRecovery._timeouts.should.have.length(5);

      results.should.be.length(3);
      results.forEach(function (result, index) {
        result.should.eql(consumerGroupRecovery._timeouts[index]);
      });

      results = _.times(5, function () {
        var ret = consumerGroupRecovery.getRetryTimeout(differentError);
        consumerGroupRecovery.lastError = differentError;
        return ret;
      });

      consumerGroupRecovery._timeouts.should.have.length(5);

      results.should.be.length(5);
      results.forEach(function (result, index) {
        result.should.eql(consumerGroupRecovery._timeouts[index]);
      });
    });

    it('should return backoff timeouts when calling same error', function () {
      var error = new GroupCoordinatorNotAvailable();
      error.errorCode = 15;

      consumerGroupRecovery.options.retries = 5;

      var results = _.times(6, function () {
        var ret = consumerGroupRecovery.getRetryTimeout(error);
        consumerGroupRecovery.lastError = error;
        return ret;
      });

      consumerGroupRecovery._timeouts.should.have.length(5);
      results.pop().should.be.false;
      results.forEach(function (result, index) {
        result.should.eql(consumerGroupRecovery._timeouts[index]);
      });
    });

    it('should initalize timeout array when invoked', function () {
      var error = new GroupCoordinatorNotAvailable();
      error.errorCode = 15;

      should(consumerGroupRecovery._timeouts).be.undefined;

      var retryTime = consumerGroupRecovery.getRetryTimeout(error);

      consumerGroupRecovery._timeouts.should.have.length(10);
      retryTime.should.be.eql(consumerGroupRecovery._timeouts[0]);
    });

    it('should throw if empty arguments', function () {
      should.throws(function () {
        consumerGroupRecovery.getRetryTimeout();
      });

      should.throws(function () {
        consumerGroupRecovery.getRetryTimeout(null);
      });
    });
  });
});