standardnotes-app-web/packages/services/src/Domain/Subscription/SubscriptionManager.spec.ts

import { ApplicationStage } from './../Application/ApplicationStage'
import { SessionEvent } from './../Session/SessionEvent'
import { ApplicationEvent } from './../Event/ApplicationEvent'
import { StorageServiceInterface } from './../Storage/StorageServiceInterface'
import { SessionsClientInterface } from './../Session/SessionsClientInterface'
import { SubscriptionApiServiceInterface } from '@standardnotes/api'
import { Invitation } from '@standardnotes/models'
import { InternalEventBusInterface } from '..'
import { SubscriptionManager } from './SubscriptionManager'

describe('SubscriptionManager', () => {
  let subscriptionApiService: SubscriptionApiServiceInterface
  let internalEventBus: InternalEventBusInterface
  let sessions: SessionsClientInterface
  let storage: StorageServiceInterface

  const createManager = () => new SubscriptionManager(subscriptionApiService, sessions, storage, internalEventBus)

  beforeEach(() => {
    subscriptionApiService = {} as jest.Mocked<SubscriptionApiServiceInterface>
    subscriptionApiService.cancelInvite = jest.fn()
    subscriptionApiService.acceptInvite = jest.fn()
    subscriptionApiService.invite = jest.fn()
    subscriptionApiService.listInvites = jest.fn()

    sessions = {} as jest.Mocked<SessionsClientInterface>
    sessions.isSignedIn = jest.fn().mockReturnValue(true)

    storage = {} as jest.Mocked<StorageServiceInterface>

    internalEventBus = {} as jest.Mocked<InternalEventBusInterface>
    internalEventBus.addEventHandler = jest.fn()
    internalEventBus.publish = jest.fn()
  })

  describe('event handling', () => {
    it('should fetch subscriptions when the application has launched', async () => {
      const manager = createManager()
      jest.spyOn(manager, 'fetchOnlineSubscription')
      jest.spyOn(manager, 'fetchAvailableSubscriptions')

      await manager.handleEvent({ type: ApplicationEvent.Launched, payload: {} })

      expect(manager.fetchOnlineSubscription).toHaveBeenCalledTimes(1)
      expect(manager.fetchAvailableSubscriptions).toHaveBeenCalledTimes(1)
    })

    it('should fetch online subscription when user roles have changed', async () => {
      const manager = createManager()
      jest.spyOn(manager, 'fetchOnlineSubscription')

      await manager.handleEvent({ type: ApplicationEvent.UserRolesChanged, payload: {} })

      expect(manager.fetchOnlineSubscription).toHaveBeenCalledTimes(1)
    })

    it('should fetch online subscription when session is restored', async () => {
      const manager = createManager()
      jest.spyOn(manager, 'fetchOnlineSubscription')

      await manager.handleEvent({ type: SessionEvent.Restored, payload: {} })

      expect(manager.fetchOnlineSubscription).toHaveBeenCalledTimes(1)
    })

    it('should fetch online subscription when user has signed in', async () => {
      const manager = createManager()
      jest.spyOn(manager, 'fetchOnlineSubscription')

      await manager.handleEvent({ type: ApplicationEvent.SignedIn, payload: {} })

      expect(manager.fetchOnlineSubscription).toHaveBeenCalledTimes(1)
    })

    it('should handle stage change and notify event', async () => {
      const manager = createManager()
      jest.spyOn(manager, 'loadSubscriptionFromStorage')
      storage.getValue = jest.fn().mockReturnValue({})

      await manager.handleEvent({
        type: ApplicationEvent.ApplicationStageChanged,
        payload: { stage: ApplicationStage.StorageDecrypted_09 },
      })

      expect(manager.loadSubscriptionFromStorage).toHaveBeenCalled()
    })
  })

  it('should invite user by email to a shared subscription', async () => {
    subscriptionApiService.invite = jest.fn().mockReturnValue({ data: { success: true } })

    expect(await createManager().inviteToSubscription('test@test.te')).toBeTruthy()
  })

  it('should not invite user by email if the api fails to do so', async () => {
    subscriptionApiService.invite = jest.fn().mockReturnValue({ data: { error: 'foobar' } })

    expect(await createManager().inviteToSubscription('test@test.te')).toBeFalsy()
  })

  it('should not invite user by email if the api throws an error', async () => {
    subscriptionApiService.invite = jest.fn().mockImplementation(() => {
      throw new Error('Oops')
    })

    expect(await createManager().inviteToSubscription('test@test.te')).toBeFalsy()
  })

  it('should cancel invite to a shared subscription', async () => {
    subscriptionApiService.cancelInvite = jest.fn().mockReturnValue({ data: { success: true } })

    expect(await createManager().cancelInvitation('1-2-3')).toBeTruthy()
  })

  it('should not cancel invite if the api fails to do so', async () => {
    subscriptionApiService.cancelInvite = jest.fn().mockReturnValue({ data: { error: 'foobar' } })

    expect(await createManager().cancelInvitation('1-2-3')).toBeFalsy()
  })

  it('should not cancel invite if the api throws an error', async () => {
    subscriptionApiService.cancelInvite = jest.fn().mockImplementation(() => {
      throw new Error('Oops')
    })

    expect(await createManager().cancelInvitation('1-2-3')).toBeFalsy()
  })

  it('should list invites to a shared subscription', async () => {
    const invitation = {
      uuid: '1-2-3',
    } as jest.Mocked<Invitation>
    subscriptionApiService.listInvites = jest.fn().mockReturnValue({ data: { invitations: [invitation] } })

    expect(await createManager().listSubscriptionInvitations()).toEqual([invitation])
  })

  it('should return an empty list of invites if the api fails to fetch them', async () => {
    subscriptionApiService.listInvites = jest.fn().mockReturnValue({ data: { error: 'foobar' } })

    expect(await createManager().listSubscriptionInvitations()).toEqual([])
  })

  it('should return an empty list of invites if the api throws an error', async () => {
    subscriptionApiService.listInvites = jest.fn().mockImplementation(() => {
      throw new Error('Oops')
    })

    expect(await createManager().listSubscriptionInvitations()).toEqual([])
  })

  it('should accept invite to a shared subscription', async () => {
    subscriptionApiService.acceptInvite = jest.fn().mockReturnValue({ data: { success: true } })

    expect(await createManager().acceptInvitation('1-2-3')).toEqual({ success: true })
  })

  it('should not accept invite if the api fails to do so', async () => {
    subscriptionApiService.acceptInvite = jest.fn().mockReturnValue({ data: { error: { message: 'foobar' } } })

    expect(await createManager().acceptInvitation('1-2-3')).toEqual({ success: false, message: 'foobar' })
  })

  it('should not accept invite if the api throws an error', async () => {
    subscriptionApiService.acceptInvite = jest.fn().mockImplementation(() => {
      throw new Error('Oops')
    })

    expect(await createManager().acceptInvitation('1-2-3')).toEqual({
      success: false,
      message: 'Could not accept invitation.',
    })
  })
})