codecombat/test/server/functional/subscription.spec.coffee

async = require 'async'
config = require '../../../server_config'
require '../common'
utils = require '../../../app/core/utils' # Must come after require /common
mongoose = require 'mongoose'

# sample data that comes in through the webhook when you subscribe
invoiceChargeSampleEvent = {
  id: 'evt_155TBeKaReE7xLUdrKM72O5R',
  created: 1417574898,
  livemode: false,
  type: 'invoice.payment_succeeded',
  data: {
    object: {
      date: 1417574897,
      id: 'in_155TBdKaReE7xLUdv8z8ipWl',
      period_start: 1417574897,
      period_end: 1417574897,
      lines: {},
      subtotal: 999,
      total: 999,
      customer: 'cus_5Fz9MVWP2bDPGV',
      object: 'invoice',
      attempted: true,
      closed: true,
      forgiven: false,
      paid: true,
      livemode: false,
      attempt_count: 1,
      amount_due: 999,
      currency: 'usd',
      starting_balance: 0,
      ending_balance: 0,
      next_payment_attempt: null,
      webhooks_delivered_at: null,
      charge: 'ch_155TBdKaReE7xLUdRU0WcMzR',
      discount: null,
      application_fee: null,
      subscription: 'sub_5Fz99gXrBtreNe',
      metadata: {},
      statement_description: null,
      description: null,
      receipt_number: null
    }
  },
  object: 'event',
  pending_webhooks: 1,
  request: 'iar_5Fz9c4BZJyNNsM',
  api_version: '2015-02-18'
}

customerSubscriptionDeletedSampleEvent = {
  id: 'evt_155Tj4KaReE7xLUdpoMx0UaA',
  created: 1417576970,
  livemode: false,
  type: 'customer.subscription.deleted',
  data: {
    object: {
      id: 'sub_5FziOkege03vT7',
      plan: [Object],
      object: 'subscription',
      start: 1417576967,
      status: 'canceled',
      customer: 'cus_5Fzi54gMvGG9Px',
      cancel_at_period_end: true,
      current_period_start: 1417576967,
      current_period_end: 1420255367,
      ended_at: 1417576970,
      trial_start: null,
      trial_end: null,
      canceled_at: 1417576970,
      quantity: 1,
      application_fee_percent: null,
      discount: null,
      metadata: {}
    }
  },
  object: 'event',
  pending_webhooks: 1,
  request: 'iar_5FziYQJ4oQdL6w',
  api_version: '2015-02-18'
}


describe '/db/user, editing stripe property', ->

  stripe = require('stripe')(config.stripe.secretKey)
  userURL = getURL('/db/user')
  webhookURL = getURL('/stripe/webhook')
  headers = {'X-Change-Plan': 'true'}

  it 'clears the db first', (done) ->
    clearModels [User, Payment], (err) ->
      throw err if err
      done()

  it 'denies anonymous users trying to subscribe', (done) ->
    request.get getURL('/auth/whoami'), (err, res, body) ->
      body = JSON.parse(body)
      body.stripe = { planID: 'basic', token: '12345' }
      request.put {uri: userURL, json: body, headers: headers}, (err, res, body) ->
        expect(res.statusCode).toBe 403
        done()

  #- shared data between tests
  joeData = null
  firstSubscriptionID = null

  it 'returns client error when a token fails to charge', (done) ->
    stripe.tokens.create {
      card: { number: '4000000000000002', exp_month: 12, exp_year: 2020, cvc: '123' }
    }, (err, token) ->
      stripeTokenID = token.id
      loginJoe (joe) ->
        joeData = joe.toObject()
        joeData.stripe = {
          token: stripeTokenID
          planID: 'basic'
        }
        request.put {uri: userURL, json: joeData, headers: headers }, (err, res, body) ->
          expect(res.statusCode).toBe(402)
          done()

  it 'creates a subscription when you put a token and plan', (done) ->
    stripe.tokens.create {
      card: { number: '4242424242424242', exp_month: 12, exp_year: 2020, cvc: '123' }
    }, (err, token) ->
      stripeTokenID = token.id
      loginJoe (joe) ->
        joeData = joe.toObject()
        joeData.stripe = {
          token: stripeTokenID
          planID: 'basic'
        }
        request.put {uri: userURL, json: joeData, headers: headers }, (err, res, body) ->
          joeData = body
          expect(res.statusCode).toBe(200)
          expect(joeData.purchased.gems).toBe(3500)
          expect(joeData.stripe.customerID).toBeDefined()
          expect(firstSubscriptionID = joeData.stripe.subscriptionID).toBeDefined()
          expect(joeData.stripe.planID).toBe('basic')
          expect(joeData.stripe.token).toBeUndefined()
          done()

  it 'records a payment through the webhook', (done) ->
    # Don't even want to think about hooking in tests to webhooks, so... put in some data manually
    stripe.invoices.list {customer: joeData.stripe.customerID}, (err, invoices) ->
      expect(invoices.data.length).toBe(1)
      event = _.cloneDeep(invoiceChargeSampleEvent)
      event.data.object = invoices.data[0]

      request.post {uri: webhookURL, json: event}, (err, res, body) ->
        expect(res.statusCode).toBe(201)
        Payment.find {}, (err, payments) ->
          expect(payments.length).toBe(1)
          User.findById joeData._id, (err, user) ->
            expect(user.get('purchased').gems).toBe(3500)
            done()

  it 'schedules the stripe subscription to be cancelled when stripe.planID is removed from the user', (done) ->
    delete joeData.stripe.planID
    request.put {uri: userURL, json: joeData, headers: headers }, (err, res, body) ->
      joeData = body
      expect(res.statusCode).toBe(200)
      expect(joeData.stripe.subscriptionID).toBeDefined()
      expect(joeData.stripe.planID).toBeUndefined()
      expect(joeData.stripe.customerID).toBeDefined()
      stripe.customers.retrieve joeData.stripe.customerID, (err, customer) ->
        expect(customer.subscriptions.data.length).toBe(1)
        expect(customer.subscriptions.data[0].cancel_at_period_end).toBe(true)
        done()

  it 'allows you to sign up again using the same customer ID as before, no token necessary', (done) ->
    joeData.stripe.planID = 'basic'
    request.put {uri: userURL, json: joeData, headers: headers }, (err, res, body) ->
      joeData = body

      expect(res.statusCode).toBe(200)
      expect(joeData.stripe.customerID).toBeDefined()
      expect(joeData.stripe.subscriptionID).toBeDefined()
      expect(joeData.stripe.subscriptionID).not.toBe(firstSubscriptionID)
      expect(joeData.stripe.planID).toBe('basic')
      done()

  it 'will not have immediately created new payments when signing back up from a cancelled subscription', (done) ->
    stripe.invoices.list {customer: joeData.stripe.customerID}, (err, invoices) ->
      expect(invoices.data.length).toBe(2)
      expect(invoices.data[0].total).toBe(0)
      event = _.cloneDeep(invoiceChargeSampleEvent)
      event.data.object = invoices.data[0]

      request.post {uri: webhookURL, json: event}, (err, res, body) ->
        expect(res.statusCode).toBe(200)
        Payment.find {}, (err, payments) ->
          expect(payments.length).toBe(1)
          User.findById joeData._id, (err, user) ->
            expect(user.get('purchased').gems).toBe(3500)
            done()

  it 'deletes the subscription from the user object when an event about it comes through the webhook', (done) ->
    stripe.customers.retrieveSubscription joeData.stripe.customerID, joeData.stripe.subscriptionID, (err, subscription) ->
      event = _.cloneDeep(customerSubscriptionDeletedSampleEvent)
      event.data.object = subscription
      request.post {uri: webhookURL, json: event}, (err, res, body) ->
        User.findById joeData._id, (err, user) ->
          expect(user.get('purchased').gems).toBe(3500)
          expect(user.get('stripe').subscriptionID).toBeUndefined()
          expect(user.get('stripe').planID).toBeUndefined()
          done()

  it "updates the customer's email when you change the user's email", (done) ->
    joeData.email = 'newEmail@gmail.com'
    request.put {uri: userURL, json: joeData, headers: headers }, (err, res, body) ->
      f = -> stripe.customers.retrieve joeData.stripe.customerID, (err, customer) ->
        expect(customer.email).toBe('newEmail@gmail.com')
        done()
      setTimeout(f, 500) # bit of a race condition here, response returns before stripe has been updated


describe 'Sponsored subscriptions', ->
  # TODO: Test recurring billing via webhooks
  # TODO: Test error rollbacks, Stripe is authority

  stripe = require('stripe')(config.stripe.secretKey)
  userURL = getURL('/db/user')
  webhookURL = getURL('/stripe/webhook')
  headers = {'X-Change-Plan': 'true'}
  subPrice = 999
  subGems = 3500
  invoicesWebHooked = {}

  # Start helpers

  getSubscribedQuantity = (numSponsored) ->
    return 0 if numSponsored < 1
    if numSponsored <= 10
      Math.round(numSponsored  * subPrice * 0.8)
    else
      Math.round(10 * subPrice * 0.8 + (numSponsored - 10) * subPrice * 0.6)

  getUnsubscribedQuantity = (numSponsored) ->
    return 0 if numSponsored < 1
    if numSponsored <= 1
      subPrice
    else if numSponsored <= 11
      Math.round(subPrice + (numSponsored - 1) * subPrice * 0.8)
    else
      Math.round(subPrice + 10 * subPrice * 0.8 + (numSponsored - 11) * subPrice * 0.6)

  verifyNotRecipient = (userID, done) ->
    User.findById userID, (err, user) ->
      expect(err).toBeNull()
      if stripeInfo = user.get('stripe')
        expect(stripeInfo.sponsorID).toBeUndefined()
      done()

  verifyNotSponsoring = (sponsorID, recipientID, done) ->
    # console.log 'verifyNotSponsoring', sponsorID, recipientID
    User.findById sponsorID, (err, sponsor) ->
      expect(err).toBeNull()
      stripeInfo = sponsor.get('stripe')
      return done() unless stripeInfo?.customerID?
      checkSubscriptions = (starting_after, done) ->
        options = {}
        options.starting_after = starting_after if starting_after
        stripe.customers.listSubscriptions stripeInfo.customerID, options, (err, subscriptions) ->
          expect(err).toBeNull()
          for subscription in subscriptions.data
            if subscription.plan.id is 'basic'
              expect(subscription.metadata.id).not.toEqual(recipientID)
            if subscription.plan.id is 'incremental'
              expect(subscription.metadata.id).toEqual(sponsorID)
          if subscriptions.has_more
            checkSubscriptions subscriptions.data[subscriptions.data.length - 1].id, done
          else
            done()
      checkSubscriptions null, done

  verifySponsorship = (sponsorUserID, sponsoredUserID, done) ->
    # console.log 'verifySponsorship', sponsorUserID, sponsoredUserID
    User.findById sponsorUserID, (err, user) ->
      expect(err).toBeNull()
      expect(user).not.toBeNull()
      sponsorStripe = user.get('stripe')
      sponsorCustomerID = sponsorStripe.customerID
      numSponsored = sponsorStripe.recipients.length
      expect(sponsorCustomerID).toBeDefined()
      expect(sponsorStripe.sponsorSubscriptionID).toBeDefined()
      expect(sponsorStripe.token).toBeUndefined()
      expect(numSponsored).toBeGreaterThan(0)

      # Verify Stripe sponsor subscription data
      stripe.customers.retrieveSubscription sponsorCustomerID, sponsorStripe.sponsorSubscriptionID, (err, subscription) ->
        expect(err).toBeNull()
        expect(subscription.plan.amount).toEqual(1)
        expect(subscription.customer).toEqual(sponsorCustomerID)
        expect(subscription.quantity).toEqual(utils.getSponsoredSubsAmount(subPrice, numSponsored, sponsorStripe.subscriptionID?))

        # Verify sponsor payment
        # May be greater than expected amount due to multiple subscribes and unsubscribes
        paymentQuery =
          purchaser: mongoose.Types.ObjectId(sponsorUserID)
          recipient: mongoose.Types.ObjectId(sponsorUserID)
          "stripe.customerID": sponsorCustomerID
          "stripe.subscriptionID": sponsorStripe.sponsorSubscriptionID
        expectedAmount = utils.getSponsoredSubsAmount(subPrice, numSponsored, sponsorStripe.subscriptionID?)
        Payment.find paymentQuery, (err, payments) ->
          expect(err).toBeNull()
          expect(payments).not.toBeNull()
          amount = 0
          for payment in payments
            amount += payment.get('amount')
            expect(payment.get('gems')).toBeUndefined()

          # NOTE: this amount may be greater than the expected amount due to proration accumlation
          # NOTE: during localy execution, this is usually only 1-2 cents
          expect(amount).toBeGreaterThan(expectedAmount - 50)

          # Find recipient info from sponsor stripe data
          for r in sponsorStripe.recipients
            if r.userID is sponsoredUserID
              recipientInfo = r
              break
          expect(recipientInfo).toBeDefined()
          expect(recipientInfo.subscriptionID).toBeDefined()
          expect(recipientInfo.subscriptionID).toNotEqual(sponsorStripe.sponsorSubscriptionID)
          expect(recipientInfo.couponID).toEqual('free')

          # Verify Stripe recipient subscription data
          stripe.customers.retrieveSubscription sponsorCustomerID, recipientInfo.subscriptionID, (err, subscription) ->
            expect(err).toBeNull()
            expect(subscription.plan.amount).toEqual(subPrice)
            expect(subscription.customer).toEqual(sponsorCustomerID)
            expect(subscription.quantity).toEqual(1)
            expect(subscription.metadata.id).toEqual(sponsoredUserID)
            expect(subscription.discount.coupon.id).toEqual(recipientInfo.couponID)

            # Verify recipient internal data
            User.findById sponsoredUserID, (err, recipient) ->
              expect(err).toBeNull()
              stripeInfo = recipient.get('stripe')
              expect(stripeInfo.sponsorID).toEqual(sponsorUserID)
              unless stripeInfo.sponsorSubscriptionID?
                expect(stripeInfo.customerID).toBeUndefined()
              expect(stripeInfo.token).toBeUndefined()
              expect(recipient.get('purchased').gems).toBeGreaterThan(subGems - 1)
              expect(recipient.isPremium()).toEqual(true)

              # Verify recipient payment
              # TODO: Not accurate enough when resubscribing a user
              paymentQuery =
                purchaser: mongoose.Types.ObjectId(sponsorUserID)
                recipient: mongoose.Types.ObjectId(sponsoredUserID)
                "stripe.customerID": sponsorCustomerID
              Payment.findOne paymentQuery, (err, payment) ->
                expect(err).toBeNull()
                expect(payment).not.toBeNull()
                expect(payment.get('amount')).toEqual(0)
                expect(payment.get('gems')).toBeGreaterThan(subGems - 1)
                done()

  subscribeUser = (user, token, done) ->
    requestBody = user.toObject()
    requestBody.stripe =
      planID: 'basic'
    requestBody.stripe.token = token.id if token?
    request.put {uri: userURL, json: requestBody, headers: headers }, (err, res, body) ->
      expect(err).toBeNull()
      expect(res.statusCode).toBe(200)
      expect(body.stripe.customerID).toBeDefined()
      expect(body.stripe.planID).toBe('basic')
      expect(body.stripe.token).toBeUndefined()
      expect(body.purchased.gems).toBeGreaterThan(subGems - 1)
      done()

  unsubscribeUser = (user, done) ->
    requestBody = user.toObject()
    delete requestBody.stripe.planID
    request.put {uri: userURL, json: requestBody, headers: headers }, (err, res, body) ->
      expect(err).toBeNull()
      expect(res.statusCode).toBe(200)
      User.findById user.id, (err, user) ->
        expect(user.get('stripe').customerID).toBeDefined()
        expect(user.get('stripe').planID).toBeUndefined()
        expect(user.get('stripe').token).toBeUndefined()
        done()

  subscribeRecipients = (sponsor, recipients, token, done) ->
    # console.log 'subscribeRecipients', sponsor.id, (recipient.id for recipient in recipients), token?
    requestBody = sponsor.toObject()
    requestBody.stripe =
      subscribeEmails: (recipient.get('email') for recipient in recipients)
    requestBody.stripe.token = token.id if token?
    request.put {uri: userURL, json: requestBody, headers: headers }, (err, res, body) ->
      expect(err).toBeNull()
      expect(res.statusCode).toBe(200)
      expect(body.stripe.customerID).toBeDefined()
      updatedUser = body

      # Call webhooks for invoices
      options = customer: body.stripe.customerID, limit: 100
      stripe.invoices.list options, (err, invoices) ->
        expect(err).toBeNull()
        expect(invoices.has_more).toEqual(false)
        makeWebhookCall = (invoice) ->
          (callback) ->
            event = _.cloneDeep(invoiceChargeSampleEvent)
            event.data.object = invoice
            # console.log 'Calling webhook', event.type, invoice.id
            request.post {uri: webhookURL, json: event}, (err, res, body) ->
              callback err
        webhookTasks = []
        for invoice in invoices.data
          unless invoice.id of invoicesWebHooked
            invoicesWebHooked[invoice.id] = true
            webhookTasks.push makeWebhookCall(invoice)
        async.parallel webhookTasks, (err, results) ->
          expect(err?).toEqual(false)
          done(updatedUser)

  unsubscribeRecipient = (sponsor, recipient, immediately, done) ->
    # console.log 'unsubscribeRecipient', sponsor.id, recipient.id
    stripeInfo = sponsor.get('stripe')
    customerID = stripeInfo.customerID
    for r in stripeInfo.recipients
      if r.userID is recipient.id
        subscriptionID = r.subscriptionID
        break
    expect(customerID).toBeDefined()
    expect(subscriptionID).toBeDefined()

    # Find Stripe subscription
    stripe.customers.retrieveSubscription customerID, subscriptionID, (err, subscription) ->
      expect(err).toBeNull()
      expect(subscription).not.toBeNull()

      # Call unsubscribe API
      requestBody = sponsor.toObject()
      requestBody.stripe = unsubscribeEmail: recipient.get('email')
      request.put {uri: userURL, json: requestBody, headers: headers }, (err, res, body) ->
        expect(err).toBeNull()
        expect(res.statusCode).toBe(200)

        # Simulate subscription ending after cancellation
        return done() unless immediately

        # Simulate subscription cancelling a trial end
        stripe.customers.cancelSubscription customerID, subscriptionID, (err) ->
          expect(err).toBeNull()

          # Simulate customer.subscription.deleted webhook event
          event = _.cloneDeep(customerSubscriptionDeletedSampleEvent)
          event.data.object = subscription
          request.post {uri: webhookURL, json: event}, (err, res, body) ->
            expect(err).toBeNull()
            done()

  # Subscribe a bunch of recipients at once, used for bulk discount testing
  class SubbedRecipients
    constructor: (@count, @toVerify) ->
      @index = 0
      @recipients = []

    length: ->
      @recipients.length

    get: (i) ->
      @recipients[i]

    createRecipients: (done) ->
      return done() if @recipients.length is @count
      createNewUser (user) =>
        @recipients.push user
        @createRecipients done

    subRecipients: (user1, token=null, done) ->
      # console.log 'subRecipients', user1.id, @recipients.length
      User.findById user1.id, (err, user1) =>
        subscribeRecipients user1, @recipients, token, (updatedUser) =>
          verifyIndex = 0
          verify = =>
            return done(updatedUser) if verifyIndex >= @toVerify.length
            verifySponsorship user1.id, @recipients[verifyIndex].id, =>
              verifyIndex++
              verify()
          verify()

  # End helpers


  # TODO: Use beforeAll()
  it 'Clear database users and payments', (done) ->
    clearModels [User, Payment], (err) ->
      throw err if err
      done()

  describe 'Basic', ->
    it 'Unsubscribed user1 subscribes user2', (done) ->
      stripe.tokens.create {
        card: { number: '4242424242424242', exp_month: 12, exp_year: 2020, cvc: '123' }
      }, (err, token) ->
        createNewUser (user2) ->
          loginNewUser (user1) ->
            subscribeRecipients user1, [user2], token, (updatedUser) ->
              verifySponsorship user1.id, user2.id, done

    it 'Unsubscribed user1 unsubscribes user2 and their sub ends', (done) ->
      stripe.tokens.create {
        card: { number: '4242424242424242', exp_month: 12, exp_year: 2020, cvc: '123' }
      }, (err, token) ->
        createNewUser (user2) ->
          loginNewUser (user1) ->
            subscribeRecipients user1, [user2], token, (updatedUser) ->
              User.findById user1.id, (err, user1) ->
                unsubscribeRecipient user1, user2, true, ->
                  verifyNotSponsoring user1.id, user2.id, ->
                    verifyNotRecipient user2.id, done

    it 'Unsubscribed user1 immediately resubscribes user2, one token', (done) ->
      stripe.tokens.create {
        card: { number: '4242424242424242', exp_month: 12, exp_year: 2020, cvc: '123' }
      }, (err, token) ->
        createNewUser (user2) ->
          loginNewUser (user1) ->
            subscribeRecipients user1, [user2], token, (updatedUser) ->
              User.findById user1.id, (err, user1) ->
                unsubscribeRecipient user1, user2, false, ->
                  subscribeRecipients user1, [user2], null, (updatedUser) ->
                    verifySponsorship user1.id, user2.id, done

    it 'Sponsored user2 subscribes their sponsor user1', (done) ->
      stripe.tokens.create {
        card: { number: '4242424242424242', exp_month: 12, exp_year: 2020, cvc: '123' }
      }, (err, token) ->
        createNewUser (user2) ->
          loginNewUser (user1) ->
            subscribeRecipients user1, [user2], token, (updatedUser) ->
             loginUser user2, (user2) ->
                stripe.tokens.create {
                  card: { number: '4242424242424242', exp_month: 12, exp_year: 2020, cvc: '123' }
                }, (err, token) ->
                  subscribeRecipients user2, [user1], token, (updatedUser) ->
                    verifySponsorship user1.id, user2.id, ->
                      verifySponsorship user2.id, user1.id, done

    it 'Unsubscribed user1 subscribes user1', (done) ->
      stripe.tokens.create {
        card: { number: '4242424242424242', exp_month: 12, exp_year: 2020, cvc: '123' }
      }, (err, token) ->
        loginNewUser (user1) ->

          requestBody = user1.toObject()
          requestBody.stripe =
            subscribeEmails: [user1.get('email')]
            token: token.id
          request.put {uri: userURL, json: requestBody, headers: headers }, (err, res, body) ->
            expect(err).toBeNull()
            expect(res.statusCode).toBe(200)

            User.findById user1.id, (err, user) ->
              expect(err).toBeNull()
              stripeInfo = user.get('stripe')
              expect(stripeInfo.customerID).toBeDefined()
              expect(stripeInfo.planID).toBeUndefined()
              expect(stripeInfo.subscriptionID).toBeUndefined()
              expect(stripeInfo.recipients.length).toEqual(0)
              done()

    it 'Subscribed user1 subscribes user2, one token', (done) ->
      stripe.tokens.create {
        card: { number: '4242424242424242', exp_month: 12, exp_year: 2020, cvc: '123' }
      }, (err, token) ->
        createNewUser (user2) ->
          loginNewUser (user1) ->
            subscribeUser user1, token, (updatedUser) ->
              User.findById user1.id, (err, user1) ->
                expect(err).toBeNull()
                subscribeRecipients user1, [user2], null, (updatedUser) ->
                  User.findById user1.id, (err, user1) ->
                    expect(err).toBeNull()
                    expect(user1.get('stripe').subscriptionID).toBeDefined()
                    expect(user1.isPremium()).toEqual(true)
                    verifySponsorship user1.id, user2.id, done

    it 'Subscribed user1 unsubscribes user2', (done) ->
      stripe.tokens.create {
        card: { number: '4242424242424242', exp_month: 12, exp_year: 2020, cvc: '123' }
      }, (err, token) ->
        createNewUser (user2) ->
          loginNewUser (user1) ->
            subscribeUser user1, token, (updatedUser) ->
              User.findById user1.id, (err, user1) ->
                expect(err).toBeNull()
                subscribeRecipients user1, [user2], null, (updatedUser) ->
                  User.findById user1.id, (err, user1) ->
                    unsubscribeRecipient user1, user2, true, ->
                      User.findById user1.id, (err, user1) ->
                        expect(err).toBeNull()
                        expect(user1.get('stripe').subscriptionID).toBeDefined()
                        expect(user1.isPremium()).toEqual(true)
                        User.findById user2.id, (err, user2) ->
                          verifyNotSponsoring user1.id, user2.id, ->
                            verifyNotRecipient user2.id, done

    it 'Subscribed user1 subscribes user2, unsubscribes themselves', (done) ->
      stripe.tokens.create {
        card: { number: '4242424242424242', exp_month: 12, exp_year: 2020, cvc: '123' }
      }, (err, token) ->
        createNewUser (user2) ->
          loginNewUser (user1) ->
            subscribeUser user1, token, (updatedUser) ->
              User.findById user1.id, (err, user1) ->
                expect(err).toBeNull()
                subscribeRecipients user1, [user2], null, (updatedUser) ->
                  User.findById user1.id, (err, user1) ->
                    unsubscribeUser user1, (updatedUser) ->
                      verifySponsorship user1.id, user2.id, done

    it 'Sponsored user2 tries to subscribe', (done) ->
      stripe.tokens.create {
        card: { number: '4242424242424242', exp_month: 12, exp_year: 2020, cvc: '123' }
      }, (err, token) ->
        createNewUser (user2) ->
          loginNewUser (user1) ->
            subscribeRecipients user1, [user2], token, (updatedUser) ->
               loginUser user2, (user2) ->
                stripe.tokens.create {
                  card: { number: '4242424242424242', exp_month: 12, exp_year: 2020, cvc: '123' }
                }, (err, token) ->
                  requestBody = user2.toObject()
                  requestBody.stripe =
                    token: token.id
                    planID: 'basic'
                  request.put {uri: userURL, json: requestBody, headers: headers }, (err, res, body) ->
                    expect(err).toBeNull()
                    expect(res.statusCode).toBe(403)
                    done()

    it 'Sponsored user2 tries to unsubscribe', (done) ->
      stripe.tokens.create {
        card: { number: '4242424242424242', exp_month: 12, exp_year: 2020, cvc: '123' }
      }, (err, token) ->
        createNewUser (user2) ->
          loginNewUser (user1) ->
            subscribeRecipients user1, [user2], token, (updatedUser) ->
              loginUser user2, (user2) ->
                requestBody = user2.toObject()
                requestBody.stripe =
                  recipient: user2.id
                request.put {uri: userURL, json: requestBody, headers: headers }, (err, res, body) ->
                  expect(err).toBeNull()
                  expect(res.statusCode).toBe(200)
                  verifySponsorship user1.id, user2.id, done

    it 'Cancel sponsor subscription with 2 recipient subscriptions, then subscribe 1 old and 1 new', (done) ->
      stripe.tokens.create {
        card: { number: '4242424242424242', exp_month: 12, exp_year: 2020, cvc: '123' }
      }, (err, token) ->
        createNewUser (user3) ->
          createNewUser (user2) ->
            loginNewUser (user1) ->
              subscribeRecipients user1, [user2, user3], token, (updatedUser) ->
                customerID = updatedUser.stripe.customerID
                subscriptionID = updatedUser.stripe.sponsorSubscriptionID

                # Find Stripe sponsor subscription
                stripe.customers.retrieveSubscription customerID, subscriptionID, (err, subscription) ->
                  expect(err).toBeNull()
                  expect(subscription).not.toBeNull()

                  # Cancel Stripe sponsor subscription
                  stripe.customers.cancelSubscription customerID, subscriptionID, (err) ->
                    expect(err).toBeNull()

                    # Simulate customer.subscription.deleted webhook event for sponsor subscription
                    event = _.cloneDeep(customerSubscriptionDeletedSampleEvent)
                    event.data.object = subscription
                    request.post {uri: webhookURL, json: event}, (err, res, body) ->
                      expect(err).toBeNull()

                      # Should have 2 cancelled recipient subs with cancel_at_period_end = true
                      User.findById user1.id, (err, user1) ->
                        expect(err).toBeNull()
                        stripeInfo = user1.get('stripe')
                        expect(stripeInfo.sponsorSubscriptionID).toBeUndefined()
                        expect(stripeInfo.recipients).toBeUndefined()
                        stripe.customers.listSubscriptions stripeInfo.customerID, (err, subscriptions) ->
                          expect(err).toBeNull()
                          expect(subscriptions.data.length).toEqual(2)
                          for sub in subscriptions.data
                            expect(sub.plan.id).toEqual('basic')
                            expect(sub.cancel_at_period_end).toEqual(true)

                          # Subscribe user3 back
                          User.findById user1.id, (err, user1) ->
                            subscribeRecipients user1, [user3], null, (updatedUser) ->
                              verifySponsorship user1.id, user3.id, ->

                                # Subscribe new user4
                                createNewUser (user4) ->
                                  loginUser user1, (user1) ->
                                    User.findById user1.id, (err, user1) ->
                                      subscribeRecipients user1, [user4], null, (updatedUser) ->
                                        verifySponsorship user1.id, user4.id, done

    it 'Subscribing two users separately yields proration payment', (done) ->
      # TODO: Use test plan with low duration + setTimeout to test delay between 2 subscribes
      stripe.tokens.create {
        card: { number: '4242424242424242', exp_month: 12, exp_year: 2020, cvc: '123' }
      }, (err, token) ->
        createNewUser (user3) ->
          createNewUser (user2) ->
            loginNewUser (user1) ->
              subscribeRecipients user1, [user2], token, (updatedUser) ->
                User.findById user1.id, (err, user1) ->
                  subscribeRecipients user1, [user3], null, (updatedUser) ->
                    # TODO: What do we expect invoices to show here?
                    stripe.invoices.list {customer: updatedUser.stripe.customerID}, (err, invoices) ->
                      expect(err).toBeNull()

                      # Verify for proration invoice
                      foundProratedInvoice = false
                      for invoice in invoices.data
                        line = invoice.lines.data[0]
                        if line.type is 'invoiceitem' and line.proration
                          totalAmount = utils.getSponsoredSubsAmount(subPrice, 2, false)
                          expect(invoice.total).toBeLessThan(totalAmount)
                          expect(invoice.total).toEqual(totalAmount - subPrice)
                          Payment.findOne "stripe.invoiceID": invoice.id, (err, payment) ->
                            expect(err).toBeNull()
                            expect(payment.get('amount')).toEqual(invoice.total)
                            done()
                          foundProratedInvoice = true
                          break
                      unless foundProratedInvoice
                        expect(foundProratedInvoice).toEqual(true)
                        done()

    it 'Invalid subscribeEmails', (done) ->
      stripe.tokens.create {
        card: { number: '4242424242424242', exp_month: 12, exp_year: 2020, cvc: '123' }
      }, (err, token) ->
        loginNewUser (user1) ->
          requestBody = user1.toObject()
          requestBody.stripe =
            subscribeEmails: ['invalid@user.com', 'notemailformat', '', null, undefined]
          requestBody.stripe.token = token.id
          request.put {uri: userURL, json: requestBody, headers: headers }, (err, res, body) ->
            expect(err).toBeNull()
            expect(res.statusCode).toBe(200)
            expect(body.stripe).toBeDefined()
            User.findById user1.id, (err, sponsor) ->
              expect(err).toBeNull()
              expect(sponsor.get('stripe')).toBeDefined()
              expect(sponsor.get('stripe').customerID).toBeDefined()
              expect(sponsor.get('stripe').sponsorSubscriptionID).toBeDefined()
              expect(sponsor.get('stripe').recipients?.length).toEqual(0)
              done()

    it 'User1 subscribes user2 then themselves', (done) ->
      stripe.tokens.create {
        card: { number: '4242424242424242', exp_month: 12, exp_year: 2020, cvc: '123' }
      }, (err, token) ->
        createNewUser (user2) ->
          loginNewUser (user1) ->
            subscribeRecipients user1, [user2], token, (updatedUser) ->
              User.findById user1.id, (err, user1) ->
                expect(err).toBeNull()
                verifySponsorship user1.id, user2.id, ->

                  stripe.tokens.create {
                    card: { number: '4242424242424242', exp_month: 12, exp_year: 2020, cvc: '123' }
                  }, (err, token) ->
                    subscribeUser user1, token, (updatedUser) ->
                      User.findById user1.id, (err, user1) ->
                        expect(err).toBeNull()
                        expect(user1.get('stripe').subscriptionID).toBeDefined()
                        expect(user1.isPremium()).toEqual(true)

                        stripe.customers.listSubscriptions user1.get('stripe').customerID, (err, subscriptions) ->
                          expect(err).toBeNull()
                          expect(subscriptions.data.length).toEqual(3)
                          for sub in subscriptions.data
                            if sub.plan.id is 'basic'
                              if sub.discount?.coupon?.id is 'free'
                                expect(sub.metadata?.id).toEqual(user2.id)
                              else
                                expect(sub.metadata?.id).toEqual(user1.id)
                            else
                              expect(sub.plan.id).toEqual('incremental')
                              expect(sub.metadata?.id).toEqual(user1.id)
                          done()

  describe 'Bulk discounts', ->
    # Bulk discount algorithm (includes personal sub):
    # 1 100%
    # 2-11 80%
    # 12+ 60%

    it 'Unsubscribed user1 subscribes two users', (done) ->
      stripe.tokens.create {
        card: { number: '4242424242424242', exp_month: 12, exp_year: 2020, cvc: '123' }
      }, (err, token) ->
        createNewUser (user3) ->
          createNewUser (user2) ->
            loginNewUser (user1) ->
              subscribeRecipients user1, [user2, user3], token, (updatedUser) ->
                verifySponsorship user1.id, user2.id, ->
                  verifySponsorship user1.id, user3.id, done

    it 'Subscribed user1 subscribes 2 users, unsubscribes 2', (done) ->
      recipientCount = 2
      recipientsToVerify = [0, 1]
      recipients = new SubbedRecipients recipientCount, recipientsToVerify

      # Create recipients
      recipients.createRecipients ->
        expect(recipients.length()).toEqual(recipientCount)

        stripe.tokens.create {
          card: { number: '4242424242424242', exp_month: 12, exp_year: 2020, cvc: '123' }
        }, (err, token) ->

          # Create sponsor user
          loginNewUser (user1) ->
            subscribeUser user1, token, (updatedUser) ->
              User.findById user1.id, (err, user1) ->
                expect(err).toBeNull()

                # Subscribe recipients
                recipients.subRecipients user1, null, ->
                  User.findById user1.id, (err, user1) ->

                    # Unsubscribe recipient0
                    unsubscribeRecipient user1, recipients.get(0), true, ->
                      User.findById user1.id, (err, user1) ->
                        stripeInfo = user1.get('stripe')
                        expect(stripeInfo.recipients.length).toEqual(1)
                        verifyNotSponsoring user1.id, recipients.get(0).id, ->
                          verifyNotRecipient recipients.get(0).id, ->
                            stripe.customers.retrieveSubscription stripeInfo.customerID, stripeInfo.sponsorSubscriptionID, (err, subscription) ->
                              expect(err).toBeNull()
                              expect(subscription).not.toBeNull()
                              expect(subscription.quantity).toEqual(getSubscribedQuantity(1))

                              # Unsubscribe recipient1
                              unsubscribeRecipient user1, recipients.get(1), true, ->
                                User.findById user1.id, (err, user1) ->
                                  stripeInfo = user1.get('stripe')
                                  expect(stripeInfo.recipients.length).toEqual(0)
                                  verifyNotSponsoring user1.id, recipients.get(1).id, ->
                                    verifyNotRecipient recipients.get(1).id, ->
                                      stripe.customers.retrieveSubscription stripeInfo.customerID, stripeInfo.sponsorSubscriptionID, (err, subscription) ->
                                        expect(err).toBeNull()
                                        expect(subscription).not.toBeNull()
                                        expect(subscription.quantity).toEqual(0)
                                        done()

    it 'Subscribed user1 subscribes 3 users, unsubscribes 2, themselves, then 1', (done) ->
      recipientCount = 3
      recipientsToVerify = [0, 1, 2]
      recipients = new SubbedRecipients recipientCount, recipientsToVerify

      # Create recipients
      recipients.createRecipients ->
        expect(recipients.length()).toEqual(recipientCount)
        stripe.tokens.create {
          card: { number: '4242424242424242', exp_month: 12, exp_year: 2020, cvc: '123' }
        }, (err, token) ->

          # Create sponsor user
          loginNewUser (user1) ->
            subscribeUser user1, token, (updatedUser) ->
              User.findById user1.id, (err, user1) ->
                expect(err).toBeNull()

                # Subscribe recipients
                recipients.subRecipients user1, null, ->
                  User.findById user1.id, (err, user1) ->

                    # Unsubscribe first recipient
                    unsubscribeRecipient user1, recipients.get(0), true, ->
                      User.findById user1.id, (err, user1) ->
                        stripeInfo = user1.get('stripe')
                        expect(stripeInfo.recipients.length).toEqual(recipientCount - 1)
                        verifyNotSponsoring user1.id, recipients.get(0).id, ->
                          verifyNotRecipient recipients.get(0).id, ->
                            stripe.customers.retrieveSubscription stripeInfo.customerID, stripeInfo.sponsorSubscriptionID, (err, subscription) ->
                              expect(err).toBeNull()
                              expect(subscription).not.toBeNull()
                              expect(subscription.quantity).toEqual(getSubscribedQuantity(recipientCount - 1))

                              # Unsubscribe second recipient
                              unsubscribeRecipient user1, recipients.get(1), true, ->
                                User.findById user1.id, (err, user1) ->
                                  stripeInfo = user1.get('stripe')
                                  expect(stripeInfo.recipients.length).toEqual(recipientCount - 2)
                                  verifyNotSponsoring user1.id, recipients.get(1).id, ->
                                    verifyNotRecipient recipients.get(1).id, ->
                                      stripe.customers.retrieveSubscription stripeInfo.customerID, stripeInfo.sponsorSubscriptionID, (err, subscription) ->
                                        expect(err).toBeNull()
                                        expect(subscription).not.toBeNull()
                                        expect(subscription.quantity).toEqual(getSubscribedQuantity(recipientCount - 2))

                                        # Unsubscribe self
                                        User.findById user1.id, (err, user1) ->
                                          unsubscribeUser user1, ->
                                            User.findById user1.id, (err, user1) ->
                                              stripeInfo = user1.get('stripe')
                                              expect(stripeInfo.planID).toBeUndefined()

                                              # Unsubscribe third recipient
                                              verifySponsorship user1.id, recipients.get(2).id, ->
                                                unsubscribeRecipient user1, recipients.get(2), true, ->
                                                  User.findById user1.id, (err, user1) ->
                                                    stripeInfo = user1.get('stripe')
                                                    expect(stripeInfo.recipients.length).toEqual(recipientCount - 3)
                                                    verifyNotSponsoring user1.id, recipients.get(2).id, ->
                                                      verifyNotRecipient recipients.get(2).id, ->
                                                        stripe.customers.retrieveSubscription stripeInfo.customerID, stripeInfo.sponsorSubscriptionID, (err, subscription) ->
                                                          expect(err).toBeNull()
                                                          expect(subscription).not.toBeNull()
                                                          expect(subscription.quantity).toEqual(getSubscribedQuantity(recipientCount - 3))
                                                          done()

    it 'Unsubscribed user1 subscribes 13 users, unsubcribes 2', (done) ->
      # TODO: verify interim invoices?
      recipientCount = 13
      recipientsToVerify = [0, 1, 10, 11, 12]
      recipients = new SubbedRecipients recipientCount, recipientsToVerify

      # Create recipients
      recipients.createRecipients ->
        expect(recipients.length()).toEqual(recipientCount)

        stripe.tokens.create {
          card: { number: '4242424242424242', exp_month: 12, exp_year: 2020, cvc: '123' }
        }, (err, token) ->

          # Create sponsor user
          loginNewUser (user1) ->

            # Subscribe recipients
            recipients.subRecipients user1, token, ->
              User.findById user1.id, (err, user1) ->

                # Unsubscribe first recipient
                unsubscribeRecipient user1, recipients.get(0), true, ->
                  User.findById user1.id, (err, user1) ->
                    stripeInfo = user1.get('stripe')
                    expect(stripeInfo.recipients.length).toEqual(recipientCount - 1)
                    verifyNotSponsoring user1.id, recipients.get(0).id, ->
                      verifyNotRecipient recipients.get(0).id, ->
                        stripe.customers.retrieveSubscription stripeInfo.customerID, stripeInfo.sponsorSubscriptionID, (err, subscription) ->
                          expect(err).toBeNull()
                          expect(subscription).not.toBeNull()
                          expect(subscription.quantity).toEqual(getUnsubscribedQuantity(recipientCount - 1))

                          # Unsubscribe last recipient
                          unsubscribeRecipient user1, recipients.get(recipientCount - 1), true, ->
                            User.findById user1.id, (err, user1) ->
                              stripeInfo = user1.get('stripe')
                              expect(stripeInfo.recipients.length).toEqual(recipientCount - 2)
                              verifyNotSponsoring user1.id, recipients.get(recipientCount - 1).id, ->
                                verifyNotRecipient recipients.get(recipientCount - 1).id, ->
                                  stripe.customers.retrieveSubscription stripeInfo.customerID, stripeInfo.sponsorSubscriptionID, (err, subscription) ->
                                    expect(err).toBeNull()
                                    expect(subscription).not.toBeNull()
                                    numSponsored = recipientCount - 2
                                    if numSponsored <= 1
                                      expect(subscription.quantity).toEqual(subPrice)
                                    else if numSponsored <= 11
                                      expect(subscription.quantity).toEqual(subPrice + (numSponsored - 1) * subPrice * 0.8)
                                    else
                                      expect(subscription.quantity).toEqual(subPrice + 10 * subPrice * 0.8 + (numSponsored - 11) * subPrice * 0.6)
                                    done()