Overleaf/services/web/test/unit/src/Referal/ReferalHandlerTests.js

const SandboxedModule = require('sandboxed-module')
const { expect } = require('chai')
const sinon = require('sinon')
const modulePath = require('path').join(
  __dirname,
  '../../../../app/src/Features/Referal/ReferalHandler.js'
)

describe('Referal handler', function () {
  beforeEach(function () {
    this.User = { findById: sinon.stub() }
    this.handler = SandboxedModule.require(modulePath, {
      requires: {
        '../../models/User': {
          User: this.User,
        },
      },
    })
    this.user_id = '12313'
  })

  describe('getting refered user_ids', function () {
    it('should get the user from mongo and return the refered users array', function (done) {
      const user = {
        refered_users: ['1234', '312312', '3213129'],
        refered_user_count: 3,
      }
      this.User.findById.callsArgWith(2, null, user)

      this.handler.getReferedUsers(
        this.user_id,
        (err, passedReferedUserIds, passedReferedUserCount) => {
          expect(err).not.to.exist
          passedReferedUserIds.should.deep.equal(user.refered_users)
          passedReferedUserCount.should.equal(3)
          done()
        }
      )
    })

    it('should return an empty array if it is not set', function (done) {
      const user = {}
      this.User.findById.callsArgWith(2, null, user)

      this.handler.getReferedUsers(
        this.user_id,
        (err, passedReferedUserIds, passedReferedUserCount) => {
          expect(err).not.to.exist
          passedReferedUserIds.length.should.equal(0)
          done()
        }
      )
    })

    it('should return a zero count if neither it or the array are set', function (done) {
      const user = {}
      this.User.findById.callsArgWith(2, null, user)

      this.handler.getReferedUsers(
        this.user_id,
        (err, passedReferedUserIds, passedReferedUserCount) => {
          expect(err).not.to.exist
          passedReferedUserCount.should.equal(0)
          done()
        }
      )
    })

    it('should return the array length if count is not set', function (done) {
      const user = { refered_users: ['1234', '312312', '3213129'] }
      this.User.findById.callsArgWith(2, null, user)

      this.handler.getReferedUsers(
        this.user_id,
        (err, passedReferedUserIds, passedReferedUserCount) => {
          expect(err).not.to.exist
          passedReferedUserCount.should.equal(3)
          done()
        }
      )
    })

    it('should error if finding the user fails', function (done) {
      this.User.findById.callsArgWith(2, new Error('user not found'))

      this.handler.getReferedUsers(this.user_id, err => {
        err.should.match(/user not found/)
        done()
      })
    })
  })
})