overleaf/services/web/test/unit/src/Subscription/PlansLocatorTests.js

const SandboxedModule = require('sandboxed-module')
const { expect } = require('chai')
const modulePath = '../../../../app/src/Features/Subscription/PlansLocator'

const plans = [
  {
    planCode: 'first',
    name: '1st',
    price_in_cents: 800,
    features: {},
    featureDescription: {},
  },
  {
    planCode: 'second',
    name: '2nd',
    price_in_cents: 1500,
    features: {},
    featureDescription: {},
  },
  {
    planCode: 'third',
    name: '3rd',
    price_in_cents: 3000,
    features: {},
    featureDescription: {},
  },
]

describe('PlansLocator', function () {
  beforeEach(function () {
    this.settings = { plans }

    this.PlansLocator = SandboxedModule.require(modulePath, {
      requires: {
        '@overleaf/settings': this.settings,
      },
    })
  })

  describe('findLocalPlanInSettings', function () {
    it('should return the found plan', function () {
      const plan = this.PlansLocator.findLocalPlanInSettings('second')
      expect(plan).to.have.property('name', '2nd')
      expect(plan).to.have.property('price_in_cents', 1500)
    })
    it('should return null if no matching plan is found', function () {
      const plan = this.PlansLocator.findLocalPlanInSettings('gibberish')
      expect(plan).to.be.a('null')
    })
  })

  describe('mapRecurlyPlanCodeToStripeLookupKey', function () {
    it('should map "collaborator" plan code to stripe lookup keys', function () {
      const planCode = 'collaborator'
      const lookupKey =
        this.PlansLocator.mapRecurlyPlanCodeToStripeLookupKey(planCode)
      expect(lookupKey).to.equal('standard_monthly')
    })

    it('should map "collaborator_free_trial_7_days" plan code to stripe lookup keys', function () {
      const planCode = 'collaborator_free_trial_7_days'
      const lookupKey =
        this.PlansLocator.mapRecurlyPlanCodeToStripeLookupKey(planCode)
      expect(lookupKey).to.equal('standard_monthly')
    })

    it('should map "collaborator-annual" plan code to stripe lookup keys', function () {
      const planCode = 'collaborator-annual'
      const lookupKey =
        this.PlansLocator.mapRecurlyPlanCodeToStripeLookupKey(planCode)
      expect(lookupKey).to.equal('standard_annual')
    })

    it('should map "professional" plan code to stripe lookup keys', function () {
      const planCode = 'professional'
      const lookupKey =
        this.PlansLocator.mapRecurlyPlanCodeToStripeLookupKey(planCode)
      expect(lookupKey).to.equal('professional_monthly')
    })

    it('should map "professional_free_trial_7_days" plan code to stripe lookup keys', function () {
      const planCode = 'professional_free_trial_7_days'
      const lookupKey =
        this.PlansLocator.mapRecurlyPlanCodeToStripeLookupKey(planCode)
      expect(lookupKey).to.equal('professional_monthly')
    })

    it('should map "professional-annual" plan code to stripe lookup keys', function () {
      const planCode = 'professional-annual'
      const lookupKey =
        this.PlansLocator.mapRecurlyPlanCodeToStripeLookupKey(planCode)
      expect(lookupKey).to.equal('professional_annual')
    })

    it('should map "student" plan code to stripe lookup keys', function () {
      const planCode = 'student'
      const lookupKey =
        this.PlansLocator.mapRecurlyPlanCodeToStripeLookupKey(planCode)
      expect(lookupKey).to.equal('student_monthly')
    })

    it('shoult map "student_free_trial_7_days" plan code to stripe lookup keys', function () {
      const planCode = 'student_free_trial_7_days'
      const lookupKey =
        this.PlansLocator.mapRecurlyPlanCodeToStripeLookupKey(planCode)
      expect(lookupKey).to.equal('student_monthly')
    })

    it('should map "student-annual" plan code to stripe lookup keys', function () {
      const planCode = 'student-annual'
      const lookupKey =
        this.PlansLocator.mapRecurlyPlanCodeToStripeLookupKey(planCode)
      expect(lookupKey).to.equal('student_annual')
    })
  })

  describe('mapStripeLookupKeyToRecurlyPlanCode', function () {
    it('should return the recurly plan code for existing plans', function () {
      const planCode = 'standard_monthly'
      const lookupKey =
        this.PlansLocator.mapStripeLookupKeyToRecurlyPlanCode(planCode)
      expect(lookupKey).to.equal('collaborator')
    })

    it('should return undefined for unknown plans', function () {
      const planCode = 'foobar'
      const lookupKey =
        this.PlansLocator.mapStripeLookupKeyToRecurlyPlanCode(planCode)
      expect(lookupKey).to.equal(undefined)
    })
  })

  describe('getPlanTypeAndPeriodFromRecurlyPlanCode', function () {
    it('should return the plan type and period for "collaborator"', function () {
      const { planType, period } =
        this.PlansLocator.getPlanTypeAndPeriodFromRecurlyPlanCode(
          'collaborator'
        )
      expect(planType).to.equal('standard')
      expect(period).to.equal('monthly')
    })

    it('should return the plan type and period for "collaborator_free_trial_7_days"', function () {
      const { planType, period } =
        this.PlansLocator.getPlanTypeAndPeriodFromRecurlyPlanCode(
          'collaborator_free_trial_7_days'
        )
      expect(planType).to.equal('standard')
      expect(period).to.equal('monthly')
    })

    it('should return the plan type and period for "collaborator-annual"', function () {
      const { planType, period } =
        this.PlansLocator.getPlanTypeAndPeriodFromRecurlyPlanCode(
          'collaborator-annual'
        )
      expect(planType).to.equal('standard')
      expect(period).to.equal('annual')
    })

    it('should return the plan type and period for "professional"', function () {
      const { planType, period } =
        this.PlansLocator.getPlanTypeAndPeriodFromRecurlyPlanCode(
          'professional'
        )
      expect(planType).to.equal('professional')
      expect(period).to.equal('monthly')
    })

    it('should return the plan type and period for "professional_free_trial_7_days"', function () {
      const { planType, period } =
        this.PlansLocator.getPlanTypeAndPeriodFromRecurlyPlanCode(
          'professional_free_trial_7_days'
        )
      expect(planType).to.equal('professional')
      expect(period).to.equal('monthly')
    })

    it('should return the plan type and period for "professional-annual"', function () {
      const { planType, period } =
        this.PlansLocator.getPlanTypeAndPeriodFromRecurlyPlanCode(
          'professional-annual'
        )
      expect(planType).to.equal('professional')
      expect(period).to.equal('annual')
    })

    it('should return the plan type and period for "student"', function () {
      const { planType, period } =
        this.PlansLocator.getPlanTypeAndPeriodFromRecurlyPlanCode('student')
      expect(planType).to.equal('student')
      expect(period).to.equal('monthly')
    })

    it('should return the plan type and period for "student_free_trial_7_days"', function () {
      const { planType, period } =
        this.PlansLocator.getPlanTypeAndPeriodFromRecurlyPlanCode(
          'student_free_trial_7_days'
        )
      expect(planType).to.equal('student')
      expect(period).to.equal('monthly')
    })

    it('should return the plan type and period for "student-annual"', function () {
      const { planType, period } =
        this.PlansLocator.getPlanTypeAndPeriodFromRecurlyPlanCode(
          'student-annual'
        )
      expect(planType).to.equal('student')
      expect(period).to.equal('annual')
    })
  })
})