Overleaf/services/web/test/UnitTests/coffee/Subscription/LimitationsManagerTests.coffee

SandboxedModule = require('sandboxed-module')
sinon = require('sinon')
require('chai').should()
modulePath = require('path').join __dirname, '../../../../app/js/Features/Subscription/LimitationsManager'
Settings = require("settings-sharelatex")

describe "LimitationsManager", ->
	beforeEach ->
		@project = { _id: "project-id" }
		@user = { _id: "user-id", features:{} }
		@Project =
			findById: (project_id, fields, callback) =>
				if project_id == @project_id
					callback null, @project
				else
					callback null, null
		@User =
			findById: (user_id, callback) =>
				if user_id == @user.id
					callback null, @user
				else
					callback null, null

		@SubscriptionLocator =
			getUsersSubscription: sinon.stub()

		@LimitationsManager = SandboxedModule.require modulePath, requires:
			'../../models/Project' : Project: @Project
			'../../models/User' : User: @User
			'./SubscriptionLocator':@SubscriptionLocator
			'settings-sharelatex' : @Settings = {}
			'logger-sharelatex':log:->

	describe "allowedNumberOfCollaboratorsInProject", ->
		describe "when the project is owned by a user without a subscription", ->
			beforeEach ->
				@Settings.defaultPlanCode = collaborators: 23
				delete @user.features
				@callback = sinon.stub()
				@LimitationsManager.allowedNumberOfCollaboratorsInProject(@project_id, @callback)

			it "should return the default number", ->
				@callback.calledWith(null, @Settings.defaultPlanCode.collaborators).should.equal true

		describe "when the project is owned by a user with a subscription", ->
			beforeEach ->
				@user.features =
					collaborators: 21
				@callback = sinon.stub()
				@LimitationsManager.allowedNumberOfCollaboratorsInProject(@project_id, @callback)

			it "should return the number of collaborators the user is allowed", ->
				@callback.calledWith(null, @user.features.collaborators).should.equal true
	
	describe "currentNumberOfCollaboratorsInProject", ->
		beforeEach ->
			@project.collaberator_refs = ["one", "two"]
			@project.readOnly_refs = ["three"]
			@callback = sinon.stub()
			@LimitationsManager.currentNumberOfCollaboratorsInProject(@project_id, @callback)

		it "should return the total number of collaborators", ->
			@callback.calledWith(null, 3).should.equal true

	describe "isCollaboratorLimitReached", ->
		beforeEach ->
			sinon.stub @LimitationsManager,
					   "currentNumberOfCollaboratorsInProject",
					   (project_id, callback) => callback(null, @current_number)
			sinon.stub @LimitationsManager,
					   "allowedNumberOfCollaboratorsInProject",
					   (project_id, callback) => callback(null, @allowed_number)
			@callback = sinon.stub()

		describe "when the project has fewer collaborators than allowed", ->
			beforeEach ->
				@current_number = 1
				@allowed_number = 2
				@LimitationsManager.isCollaboratorLimitReached(@project_id, @callback)

			it "should return false", ->
				@callback.calledWith(null, false).should.equal true

		describe "when the project has more collaborators than allowed", ->
			beforeEach ->
				@current_number = 3
				@allowed_number = 2
				@LimitationsManager.isCollaboratorLimitReached(@project_id, @callback)

			it "should return true", ->
				@callback.calledWith(null, true).should.equal true

		describe "when the project has infinite collaborators", ->
			beforeEach ->
				@current_number = 100
				@allowed_number = -1
				@LimitationsManager.isCollaboratorLimitReached(@project_id, @callback)

			it "should return false", ->
				@callback.calledWith(null, false).should.equal true


	describe "userHasSubscription", ->
		beforeEach ->
			@SubscriptionLocator.getUsersSubscription = sinon.stub()

		it "should return true if the recurly token is set", (done)->
			@SubscriptionLocator.getUsersSubscription.callsArgWith(1, null, recurlySubscription_id : "1234")
			@LimitationsManager.userHasSubscription @user, (err, hasSubscription)->
				hasSubscription.should.equal true
				done()

		it "should return false if the recurly token is not set", (done)->
			@SubscriptionLocator.getUsersSubscription.callsArgWith(1, null, {})
			@subscription = {}
			@LimitationsManager.userHasSubscription @user, (err, hasSubscription)->
				hasSubscription.should.equal false
				done()

		it "should return false if the subscription is undefined", (done)->
			@SubscriptionLocator.getUsersSubscription.callsArgWith(1)
			@LimitationsManager.userHasSubscription @user, (err, hasSubscription)->
				hasSubscription.should.equal false
				done()

		it "should return the subscription", (done)->
			stubbedSubscription = {freeTrial:{}, token:""}
			@SubscriptionLocator.getUsersSubscription.callsArgWith(1, null, stubbedSubscription)
			@LimitationsManager.userHasSubscription @user, (err, hasSubOrFreeTrial, subscription)->
				subscription.should.deep.equal stubbedSubscription
				done()

	describe "userHasFreeTrial", ->
		beforeEach ->
			@SubscriptionLocator.getUsersSubscription = sinon.stub()

		it "should return true if the free trial is set", (done)->
			@SubscriptionLocator.getUsersSubscription.callsArgWith(1, null, freeTrial:{expiresAt : "1234"})
			@LimitationsManager.userHasFreeTrial @user, (err, hasFreeTrial)->
				hasFreeTrial.should.equal true
				done()

		it "should return false if the free trial is not set", (done)->
			@SubscriptionLocator.getUsersSubscription.callsArgWith(1, null, {freeTrial:{}})
			@LimitationsManager.userHasFreeTrial @user, (err, hasFreeTrial)->
				hasFreeTrial.should.equal false
				done()

		it "should return false if the free trial is not there", (done)->
			@SubscriptionLocator.getUsersSubscription.callsArgWith(1, null, {})
			@LimitationsManager.userHasFreeTrial @user, (err, hasFreeTrial)->
				hasFreeTrial.should.equal false
				done()

		it "should return false if the free trial is undefined", (done)->
			@SubscriptionLocator.getUsersSubscription.callsArgWith(1)
			@LimitationsManager.userHasFreeTrial @user, (err, hasFreeTrial)->
				hasFreeTrial.should.equal false
				done()

		it "should return the subscription", (done)->
			stubbedSubscription = {freeTrial:{}, token:""}
			@SubscriptionLocator.getUsersSubscription.callsArgWith(1, null, stubbedSubscription)
			@LimitationsManager.userHasFreeTrial @user, (err, hasSubOrFreeTrial, subscription)->
				subscription.should.deep.equal stubbedSubscription
				done()

	describe "userHasSubscriptionOrFreeTrial", ->
		beforeEach ->
			@hasFreeTrial = true
			@hasSubscription = true
			@LimitationsManager.userHasFreeTrial = sinon.stub().callsArgWith(1, null, @hasFreeTrial)
			@LimitationsManager.userHasSubscription = sinon.stub().callsArgWith(1, null, @hasSubscription)

		it "should return true if both are true", (done)->
			@hasFreeTrial = true
			@hasSubscription = true
			@LimitationsManager.userHasSubscriptionOrFreeTrial @user, (err, hasSubOrFreeTrial)->
				hasSubOrFreeTrial.should.equal true
				done()

		it "should return true if one is true", (done)->
			@hasFreeTrial = false
			@hasSubscription = true
			@LimitationsManager.userHasSubscriptionOrFreeTrial @user, (err, hasSubOrFreeTrial)->
				hasSubOrFreeTrial.should.equal true
				done()

		it "should return true if other is true", (done)->
			@hasFreeTrial = true
			@hasSubscription = false
			@LimitationsManager.userHasSubscriptionOrFreeTrial @user, (err, hasSubOrFreeTrial)->
				hasSubOrFreeTrial.should.equal true
				done()

		it "should return false if both are false", (done)->
			@hasSubscription = false
			@hasFreeTrial = false
			@LimitationsManager.userHasSubscriptionOrFreeTrial @user, (err, hasSubOrFreeTrial)->
				hasSubOrFreeTrial.should.equal true
				done()

		it "should return the subscription", (done)->
			stubbedSubscription = {freeTrial:{}, token:""}
			@LimitationsManager.userHasSubscription = sinon.stub().callsArgWith(1, null, null, stubbedSubscription)
			@LimitationsManager.userHasSubscriptionOrFreeTrial @user, (err, hasSubOrFreeTrial, subscription)->
				subscription.should.deep.equal stubbedSubscription
				done()

	describe "hasGroupMembersLimitReached", ->

		beforeEach ->
			@user_id = "12312"
			@subscription =
				membersLimit: 2
				member_ids: ["", ""]

		it "should return true if the limit is hit", (done)->
			@SubscriptionLocator.getUsersSubscription.callsArgWith(1, null, @subscription)
			@LimitationsManager.hasGroupMembersLimitReached @user_id, (err, limitReached)->
				limitReached.should.equal true
				done()

		it "should return false if the limit is not hit", (done)->
			@subscription.membersLimit = 3
			@SubscriptionLocator.getUsersSubscription.callsArgWith(1, null, @subscription)
			@LimitationsManager.hasGroupMembersLimitReached @user_id, (err, limitReached)->
				limitReached.should.equal false
				done()
				
		it "should return true if the limit has been exceded", (done)->
			@subscription.membersLimit = 0
			@SubscriptionLocator.getUsersSubscription.callsArgWith(1, null, @subscription)
			@LimitationsManager.hasGroupMembersLimitReached @user_id, (err, limitReached)->
				limitReached.should.equal true
				done()