mirror of
https://github.com/overleaf/overleaf.git
synced 2024-11-21 20:47:08 -05:00
a18176b2fa
Change database operator to add user to team GitOrigin-RevId: 0bf23c1f8958f74fa00a97559ebbf03e4edadfa6
248 lines
9.7 KiB
CoffeeScript
248 lines
9.7 KiB
CoffeeScript
SandboxedModule = require('sandboxed-module')
|
|
should = require('chai').should()
|
|
expect = require('chai').expect
|
|
sinon = require 'sinon'
|
|
modulePath = "../../../../app/js/Features/Subscription/SubscriptionUpdater"
|
|
assert = require("chai").assert
|
|
ObjectId = require('mongoose').Types.ObjectId
|
|
|
|
describe "SubscriptionUpdater", ->
|
|
|
|
beforeEach ->
|
|
@recurlySubscription =
|
|
uuid: "1238uoijdasjhd"
|
|
plan:
|
|
plan_code: "kjhsakjds"
|
|
@adminUser =
|
|
_id: @adminuser_id = "5208dd34438843e2db000007"
|
|
@otherUserId = "5208dd34438842e2db000005"
|
|
@allUserIds = ["13213", "dsadas", "djsaiud89"]
|
|
@userStub = _id: 'mock-user-stub-id', email: 'mock-stub-email@baz.com'
|
|
@subscription = subscription =
|
|
_id: "111111111111111111111111"
|
|
admin_id: @adminUser._id
|
|
manager_ids: [@adminUser._id]
|
|
member_ids: @allUserIds
|
|
save: sinon.stub().callsArgWith(0)
|
|
freeTrial:{}
|
|
planCode:"student_or_something"
|
|
@user_id = @adminuser_id
|
|
|
|
@groupSubscription =
|
|
_id: "222222222222222222222222"
|
|
admin_id: @adminUser._id
|
|
manager_ids: [@adminUser._id]
|
|
member_ids: @allUserIds
|
|
save: sinon.stub().callsArgWith(0)
|
|
freeTrial:{}
|
|
planCode:"group_subscription"
|
|
|
|
|
|
@updateStub = sinon.stub().callsArgWith(2, null)
|
|
@findAndModifyStub = sinon.stub().callsArgWith(2, null, @subscription)
|
|
@SubscriptionModel = class
|
|
constructor: (opts)->
|
|
subscription.admin_id = opts.admin_id
|
|
subscription.manager_ids = [opts.admin_id]
|
|
return subscription
|
|
@remove: sinon.stub().yields()
|
|
@SubscriptionModel.update = @updateStub
|
|
@SubscriptionModel.findAndModify = @findAndModifyStub
|
|
|
|
@SubscriptionLocator =
|
|
getUsersSubscription: sinon.stub()
|
|
getGroupSubscriptionMemberOf:sinon.stub()
|
|
|
|
@Settings =
|
|
freeTrialPlanCode: "collaborator"
|
|
defaultPlanCode: "personal"
|
|
defaultFeatures: { "default": "features" }
|
|
|
|
@UserFeaturesUpdater =
|
|
updateFeatures : sinon.stub().yields()
|
|
|
|
@PlansLocator =
|
|
findLocalPlanInSettings: sinon.stub().returns({})
|
|
|
|
@UserGetter =
|
|
getUsers: (memberIds, projection, callback) ->
|
|
users = memberIds.map (id) -> { _id: id }
|
|
callback(null, users)
|
|
getUserOrUserStubById: sinon.stub()
|
|
|
|
@ReferalFeatures = getBonusFeatures: sinon.stub().callsArgWith(1)
|
|
@Modules = {hooks: {fire: sinon.stub().callsArgWith(2, null, null)}}
|
|
@SubscriptionUpdater = SandboxedModule.require modulePath, requires:
|
|
'../../models/Subscription': Subscription:@SubscriptionModel
|
|
'./UserFeaturesUpdater': @UserFeaturesUpdater
|
|
'./SubscriptionLocator': @SubscriptionLocator
|
|
'../User/UserGetter': @UserGetter
|
|
'./PlansLocator': @PlansLocator
|
|
"logger-sharelatex": log:->
|
|
'settings-sharelatex': @Settings
|
|
"./FeaturesUpdater": @FeaturesUpdater = {}
|
|
|
|
|
|
describe "syncSubscription", ->
|
|
|
|
beforeEach ->
|
|
@SubscriptionLocator.getUsersSubscription.callsArgWith(1, null, @subscription)
|
|
@SubscriptionUpdater._updateSubscriptionFromRecurly = sinon.stub().callsArgWith(2)
|
|
|
|
it "should update the subscription if the user already is admin of one", (done)->
|
|
@SubscriptionUpdater._createNewSubscription = sinon.stub()
|
|
|
|
@SubscriptionUpdater.syncSubscription @recurlySubscription, @adminUser._id, (err)=>
|
|
@SubscriptionLocator.getUsersSubscription.calledWith(@adminUser._id).should.equal true
|
|
@SubscriptionUpdater._updateSubscriptionFromRecurly.called.should.equal true
|
|
@SubscriptionUpdater._updateSubscriptionFromRecurly.calledWith(@recurlySubscription, @subscription).should.equal true
|
|
done()
|
|
|
|
it "should not call updateFeatures with group subscription if recurly subscription is not expired", (done)->
|
|
@SubscriptionUpdater.syncSubscription @recurlySubscription, @adminUser._id, (err)=>
|
|
@SubscriptionLocator.getUsersSubscription.calledWith(@adminUser._id).should.equal true
|
|
@SubscriptionUpdater._updateSubscriptionFromRecurly.called.should.equal true
|
|
@SubscriptionUpdater._updateSubscriptionFromRecurly.calledWith(@recurlySubscription, @subscription).should.equal true
|
|
@UserFeaturesUpdater.updateFeatures.called.should.equal false
|
|
done()
|
|
|
|
|
|
describe "_updateSubscriptionFromRecurly", ->
|
|
beforeEach ->
|
|
@FeaturesUpdater.refreshFeatures = sinon.stub().callsArgWith(1)
|
|
@SubscriptionUpdater.deleteSubscription = sinon.stub().yields()
|
|
|
|
it "should update the subscription with token etc when not expired", (done)->
|
|
@SubscriptionUpdater._updateSubscriptionFromRecurly @recurlySubscription, @subscription, (err)=>
|
|
@subscription.recurlySubscription_id.should.equal @recurlySubscription.uuid
|
|
@subscription.planCode.should.equal @recurlySubscription.plan.plan_code
|
|
|
|
@subscription.freeTrial.allowed.should.equal true
|
|
assert.equal(@subscription.freeTrial.expiresAt, undefined)
|
|
assert.equal(@subscription.freeTrial.planCode, undefined)
|
|
@subscription.save.called.should.equal true
|
|
@FeaturesUpdater.refreshFeatures.calledWith(@adminUser._id).should.equal true
|
|
done()
|
|
|
|
it "should remove the subscription when expired", (done)->
|
|
@recurlySubscription.state = "expired"
|
|
@SubscriptionUpdater._updateSubscriptionFromRecurly @recurlySubscription, @subscription, (err)=>
|
|
@SubscriptionUpdater.deleteSubscription.calledWith(@subscription._id).should.equal true
|
|
done()
|
|
|
|
it "should update all the users features", (done)->
|
|
@SubscriptionUpdater._updateSubscriptionFromRecurly @recurlySubscription, @subscription, (err)=>
|
|
@FeaturesUpdater.refreshFeatures.calledWith(@adminUser._id).should.equal true
|
|
@FeaturesUpdater.refreshFeatures.calledWith(@allUserIds[0]).should.equal true
|
|
@FeaturesUpdater.refreshFeatures.calledWith(@allUserIds[1]).should.equal true
|
|
@FeaturesUpdater.refreshFeatures.calledWith(@allUserIds[2]).should.equal true
|
|
done()
|
|
|
|
it "should set group to true and save how many members can be added to group", (done)->
|
|
@PlansLocator.findLocalPlanInSettings.withArgs(@recurlySubscription.plan.plan_code).returns({groupPlan:true, membersLimit:5})
|
|
@SubscriptionUpdater._updateSubscriptionFromRecurly @recurlySubscription, @subscription, (err)=>
|
|
@subscription.membersLimit.should.equal 5
|
|
@subscription.groupPlan.should.equal true
|
|
done()
|
|
|
|
it "should not set group to true or set groupPlan", (done)->
|
|
@SubscriptionUpdater._updateSubscriptionFromRecurly @recurlySubscription, @subscription, (err)=>
|
|
assert.notEqual @subscription.membersLimit, 5
|
|
assert.notEqual @subscription.groupPlan, true
|
|
done()
|
|
|
|
|
|
describe "_createNewSubscription", ->
|
|
it "should create a new subscription then update the subscription", (done)->
|
|
@SubscriptionUpdater._createNewSubscription @adminUser._id, =>
|
|
@subscription.admin_id.should.equal @adminUser._id
|
|
@subscription.manager_ids.should.deep.equal [@adminUser._id]
|
|
@subscription.freeTrial.allowed.should.equal false
|
|
@subscription.save.called.should.equal true
|
|
done()
|
|
|
|
describe "addUserToGroup", ->
|
|
beforeEach ->
|
|
@SubscriptionUpdater.addUsersToGroup = sinon.stub().yields(null)
|
|
|
|
it "delegates to addUsersToGroup", (done)->
|
|
@SubscriptionUpdater.addUserToGroup @subscription._id, @otherUserId, =>
|
|
@SubscriptionUpdater.addUsersToGroup
|
|
.calledWith(@subscription._id, [@otherUserId]).should.equal true
|
|
done()
|
|
|
|
describe "addUsersToGroup", ->
|
|
beforeEach ->
|
|
@FeaturesUpdater.refreshFeatures = sinon.stub().callsArgWith(1)
|
|
|
|
it "should add the user ids to the group as a set", (done)->
|
|
@SubscriptionUpdater.addUsersToGroup @subscription._id, [@otherUserId], =>
|
|
searchOps =
|
|
_id: @subscription._id
|
|
insertOperation =
|
|
{ $addToSet: { member_ids: { $each: [@otherUserId] } } }
|
|
@findAndModifyStub.calledWith(searchOps, insertOperation).should.equal true
|
|
done()
|
|
|
|
it "should update the users features", (done)->
|
|
@SubscriptionUpdater.addUserToGroup @subscription._id, @otherUserId, =>
|
|
@FeaturesUpdater.refreshFeatures.calledWith(@otherUserId).should.equal true
|
|
done()
|
|
|
|
describe "removeUserFromGroup", ->
|
|
beforeEach ->
|
|
@FeaturesUpdater.refreshFeatures = sinon.stub().callsArgWith(1)
|
|
@UserGetter.getUserOrUserStubById.yields(null, {}, false)
|
|
|
|
it "should pull the users id from the group", (done)->
|
|
@SubscriptionUpdater.removeUserFromGroup @subscription._id, @otherUserId, =>
|
|
searchOps =
|
|
_id: @subscription._id
|
|
removeOperation =
|
|
"$pull": {member_ids:@otherUserId}
|
|
@updateStub.calledWith(searchOps, removeOperation).should.equal true
|
|
done()
|
|
|
|
it "should update the users features", (done)->
|
|
@SubscriptionUpdater.removeUserFromGroup @subscription._id, @otherUserId, =>
|
|
@FeaturesUpdater.refreshFeatures.calledWith(@otherUserId).should.equal true
|
|
done()
|
|
|
|
it "should not update features for user stubs", (done)->
|
|
@UserGetter.getUserOrUserStubById.yields(null, {}, true)
|
|
@SubscriptionUpdater.removeUserFromGroup @subscription._id, @userStub._id, =>
|
|
@FeaturesUpdater.refreshFeatures.called.should.equal false
|
|
done()
|
|
|
|
describe "deleteSubscription", ->
|
|
beforeEach (done) ->
|
|
@subscription_id = ObjectId().toString()
|
|
@subscription = {
|
|
"mock": "subscription",
|
|
admin_id: ObjectId(),
|
|
member_ids: [ ObjectId(), ObjectId(), ObjectId() ]
|
|
}
|
|
@SubscriptionLocator.getSubscription = sinon.stub().yields(null, @subscription)
|
|
@FeaturesUpdater.refreshFeatures = sinon.stub().yields()
|
|
@SubscriptionUpdater.deleteSubscription @subscription_id, done
|
|
|
|
it "should look up the subscription", ->
|
|
@SubscriptionLocator.getSubscription
|
|
.calledWith(@subscription_id)
|
|
.should.equal true
|
|
|
|
it "should remove the subscription", ->
|
|
@SubscriptionModel.remove
|
|
.calledWith({_id: ObjectId(@subscription_id)})
|
|
.should.equal true
|
|
|
|
it "should downgrade the admin_id", ->
|
|
@FeaturesUpdater.refreshFeatures
|
|
.calledWith(@subscription.admin_id)
|
|
.should.equal true
|
|
|
|
it "should downgrade all of the members", ->
|
|
for user_id in @subscription.member_ids
|
|
@FeaturesUpdater.refreshFeatures
|
|
.calledWith(user_id)
|
|
.should.equal true
|