overleaf/services/real-time/test/unit/js/DrainManagerTests.js

/* eslint-disable
    no-return-assign,
    no-unused-vars,
*/
// TODO: This file was created by bulk-decaffeinate.
// Fix any style issues and re-enable lint.
/*
 * decaffeinate suggestions:
 * DS102: Remove unnecessary code created because of implicit returns
 * Full docs: https://github.com/decaffeinate/decaffeinate/blob/master/docs/suggestions.md
 */
const should = require('chai').should()
const sinon = require('sinon')
const SandboxedModule = require('sandboxed-module')
const path = require('path')
const modulePath = path.join(__dirname, '../../../app/js/DrainManager')

describe('DrainManager', function () {
  beforeEach(function () {
    this.DrainManager = SandboxedModule.require(modulePath, {
      requires: {
        'logger-sharelatex': (this.logger = { log: sinon.stub() })
      }
    })
    return (this.io = {
      sockets: {
        clients: sinon.stub()
      }
    })
  })

  describe('startDrainTimeWindow', function () {
    beforeEach(function () {
      this.clients = []
      for (let i = 0; i <= 5399; i++) {
        this.clients[i] = {
          id: i,
          emit: sinon.stub()
        }
      }
      this.io.sockets.clients.returns(this.clients)
      return (this.DrainManager.startDrain = sinon.stub())
    })

    return it('should set a drain rate fast enough', function (done) {
      this.DrainManager.startDrainTimeWindow(this.io, 9)
      this.DrainManager.startDrain.calledWith(this.io, 10).should.equal(true)
      return done()
    })
  })

  return describe('reconnectNClients', function () {
    beforeEach(function () {
      this.clients = []
      for (let i = 0; i <= 9; i++) {
        this.clients[i] = {
          id: i,
          emit: sinon.stub()
        }
      }
      return this.io.sockets.clients.returns(this.clients)
    })

    return describe('after first pass', function () {
      beforeEach(function () {
        return this.DrainManager.reconnectNClients(this.io, 3)
      })

      it('should reconnect the first 3 clients', function () {
        return [0, 1, 2].map((i) =>
          this.clients[i].emit
            .calledWith('reconnectGracefully')
            .should.equal(true)
        )
      })

      it('should not reconnect any more clients', function () {
        return [3, 4, 5, 6, 7, 8, 9].map((i) =>
          this.clients[i].emit
            .calledWith('reconnectGracefully')
            .should.equal(false)
        )
      })

      return describe('after second pass', function () {
        beforeEach(function () {
          return this.DrainManager.reconnectNClients(this.io, 3)
        })

        it('should reconnect the next 3 clients', function () {
          return [3, 4, 5].map((i) =>
            this.clients[i].emit
              .calledWith('reconnectGracefully')
              .should.equal(true)
          )
        })

        it('should not reconnect any more clients', function () {
          return [6, 7, 8, 9].map((i) =>
            this.clients[i].emit
              .calledWith('reconnectGracefully')
              .should.equal(false)
          )
        })

        it('should not reconnect the first 3 clients again', function () {
          return [0, 1, 2].map((i) =>
            this.clients[i].emit.calledOnce.should.equal(true)
          )
        })

        return describe('after final pass', function () {
          beforeEach(function () {
            return this.DrainManager.reconnectNClients(this.io, 100)
          })

          it('should not reconnect the first 6 clients again', function () {
            return [0, 1, 2, 3, 4, 5].map((i) =>
              this.clients[i].emit.calledOnce.should.equal(true)
            )
          })

          return it('should log out that it reached the end', function () {
            return this.logger.log
              .calledWith('All clients have been told to reconnectGracefully')
              .should.equal(true)
          })
        })
      })
    })
  })
})