standardnotes-app-web/packages/snjs/lib/Domain/UseCase/GetRecoveryCodes/GetRecoveryCodes.spec.ts

import { AuthClientInterface, EncryptionService } from '@standardnotes/services'
import { SettingsClientInterface } from '@Lib/Services/Settings/SettingsClientInterface'

import { GetRecoveryCodes } from './GetRecoveryCodes'

describe('GetRecoveryCodes', () => {
  let authClient: AuthClientInterface
  let settingsClient: SettingsClientInterface
  let encryption: EncryptionService

  const createUseCase = () => new GetRecoveryCodes(authClient, settingsClient, encryption)

  beforeEach(() => {
    authClient = {} as jest.Mocked<AuthClientInterface>
    authClient.generateRecoveryCodes = jest.fn().mockResolvedValue('recovery-codes')

    settingsClient = {} as jest.Mocked<SettingsClientInterface>
    settingsClient.getSetting = jest.fn().mockResolvedValue('existing-recovery-codes')

    encryption = {} as jest.Mocked<EncryptionService>
    encryption.computeRootKey = jest.fn().mockResolvedValue({ serverPassword: 'test-server-password' })
    encryption.getRootKeyParams = jest.fn().mockReturnValue({ algorithm: 'test-algorithm' })
  })

  it('should return existing recovery code if they exist', async () => {
    const useCase = createUseCase()

    const result = await useCase.execute({ password: 'test-password' })

    expect(result.getValue()).toBe('existing-recovery-codes')
  })

  it('should generate recovery code if they do not exist', async () => {
    settingsClient.getSetting = jest.fn().mockResolvedValue(undefined)

    const useCase = createUseCase()

    const result = await useCase.execute({ password: 'test-password' })

    expect(result.getValue()).toBe('recovery-codes')
  })

  it('should return error if recovery code could not be generated', async () => {
    settingsClient.getSetting = jest.fn().mockResolvedValue(undefined)
    authClient.generateRecoveryCodes = jest.fn().mockResolvedValue(false)

    const useCase = createUseCase()

    const result = await useCase.execute({ password: 'test-password' })

    expect(result.isFailed()).toBe(true)
  })
})