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

import esmock from 'esmock'
import { expect } from 'chai'
import sinon from 'sinon'
const modulePath = new URL(
  '../../../../app/src/Features/Referal/ReferalHandler.mjs',
  import.meta.url
).pathname

describe('Referal handler', function () {
  beforeEach(async function () {
    this.User = {
      findById: sinon.stub().returns({
        exec: sinon.stub(),
      }),
    }
    this.handler = await esmock.strict(modulePath, {
      '../../../../app/src/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', async function () {
      const user = {
        refered_users: ['1234', '312312', '3213129'],
        refered_user_count: 3,
      }
      this.User.findById.returns({
        exec: sinon.stub().resolves(user),
      })

      const {
        referedUsers: passedReferedUserIds,
        referedUserCount: passedReferedUserCount,
      } = await this.handler.promises.getReferedUsers(this.user_id)

      passedReferedUserIds.should.deep.equal(user.refered_users)
      passedReferedUserCount.should.equal(3)
    })

    it('should return an empty array if it is not set', async function () {
      const user = {}
      this.User.findById.returns({
        exec: sinon.stub().resolves(user),
      })

      const { referedUsers: passedReferedUserIds } =
        await this.handler.promises.getReferedUsers(this.user_id)

      passedReferedUserIds.length.should.equal(0)
    })

    it('should return a zero count if neither it or the array are set', async function () {
      const user = {}
      this.User.findById.returns({
        exec: sinon.stub().resolves(user),
      })

      const { referedUserCount: passedReferedUserCount } =
        await this.handler.promises.getReferedUsers(this.user_id)

      passedReferedUserCount.should.equal(0)
    })

    it('should return the array length if count is not set', async function () {
      const user = { refered_users: ['1234', '312312', '3213129'] }
      this.User.findById.returns({
        exec: sinon.stub().resolves(user),
      })

      const { referedUserCount: passedReferedUserCount } =
        await this.handler.promises.getReferedUsers(this.user_id)

      passedReferedUserCount.should.equal(3)
    })

    it('should error if finding the user fails', async function () {
      this.User.findById.returns({
        exec: sinon.stub().rejects(new Error('user not found')),
      })

      expect(
        this.handler.promises.getReferedUsers(this.user_id)
      ).to.be.rejectedWith('user not found')
    })
  })
})