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() }) }) }) })