standardnotes-app-web/packages/snjs/lib/Services/Features/FeaturesService.spec.ts

import { ItemInterface, SNComponent, SNFeatureRepo } from '@standardnotes/models'
import { SNSyncService } from '../Sync/SyncService'
import { SettingName } from '@standardnotes/settings'
import { SNFeaturesService } from '@Lib/Services/Features'
import { ContentType } from '@standardnotes/common'
import { RoleName } from '@standardnotes/domain-core'
import { FeatureDescription, FeatureIdentifier, GetFeatures } from '@standardnotes/features'
import { SNWebSocketsService } from '../Api/WebsocketsService'
import { SNSettingsService } from '../Settings'
import { PureCryptoInterface } from '@standardnotes/sncrypto-common'
import { convertTimestampToMilliseconds } from '@standardnotes/utils'
import {
  AlertService,
  FeaturesEvent,
  FeatureStatus,
  InternalEventBusInterface,
  StorageKey,
  UserService,
} from '@standardnotes/services'
import { SNApiService, SNSessionManager } from '../Api'
import { ItemManager } from '../Items'
import { DiskStorageService } from '../Storage/DiskStorageService'

describe('featuresService', () => {
  let storageService: DiskStorageService
  let apiService: SNApiService
  let itemManager: ItemManager
  let webSocketsService: SNWebSocketsService
  let settingsService: SNSettingsService
  let userService: UserService
  let syncService: SNSyncService
  let alertService: AlertService
  let sessionManager: SNSessionManager
  let crypto: PureCryptoInterface
  let roles: string[]
  let features: FeatureDescription[]
  let items: ItemInterface[]
  let now: Date
  let tomorrow_server: number
  let tomorrow_client: number
  let internalEventBus: InternalEventBusInterface
  const expiredDate = new Date(new Date().getTime() - 1000).getTime()

  const createService = () => {
    return new SNFeaturesService(
      storageService,
      apiService,
      itemManager,
      webSocketsService,
      settingsService,
      userService,
      syncService,
      alertService,
      sessionManager,
      crypto,
      internalEventBus,
    )
  }

  beforeEach(() => {
    roles = [RoleName.NAMES.CoreUser, RoleName.NAMES.PlusUser]

    now = new Date()
    tomorrow_client = now.setDate(now.getDate() + 1)
    tomorrow_server = convertTimestampToMilliseconds(tomorrow_client * 1_000)

    features = [
      {
        ...GetFeatures().find((f) => f.identifier === FeatureIdentifier.MidnightTheme),
        expires_at: tomorrow_server,
      },
      {
        ...GetFeatures().find((f) => f.identifier === FeatureIdentifier.PlusEditor),
        expires_at: tomorrow_server,
      },
    ] as jest.Mocked<FeatureDescription[]>

    items = [] as jest.Mocked<ItemInterface[]>

    storageService = {} as jest.Mocked<DiskStorageService>
    storageService.setValue = jest.fn()
    storageService.getValue = jest.fn()

    apiService = {} as jest.Mocked<SNApiService>
    apiService.addEventObserver = jest.fn()
    apiService.getUserFeatures = jest.fn().mockReturnValue({
      data: {
        features,
      },
    })
    apiService.downloadOfflineFeaturesFromRepo = jest.fn().mockReturnValue({
      features,
    })
    apiService.isThirdPartyHostUsed = jest.fn().mockReturnValue(false)

    itemManager = {} as jest.Mocked<ItemManager>
    itemManager.getItems = jest.fn().mockReturnValue(items)
    itemManager.createItem = jest.fn()
    itemManager.createTemplateItem = jest.fn().mockReturnValue({})
    itemManager.changeComponent = jest.fn().mockReturnValue({} as jest.Mocked<ItemInterface>)
    itemManager.setItemsToBeDeleted = jest.fn()
    itemManager.addObserver = jest.fn()
    itemManager.changeItem = jest.fn()
    itemManager.changeFeatureRepo = jest.fn()

    webSocketsService = {} as jest.Mocked<SNWebSocketsService>
    webSocketsService.addEventObserver = jest.fn()

    settingsService = {} as jest.Mocked<SNSettingsService>
    settingsService.updateSetting = jest.fn()

    userService = {} as jest.Mocked<UserService>
    userService.addEventObserver = jest.fn()

    syncService = {} as jest.Mocked<SNSyncService>
    syncService.sync = jest.fn()

    alertService = {} as jest.Mocked<AlertService>
    alertService.confirm = jest.fn().mockReturnValue(true)
    alertService.alert = jest.fn()

    sessionManager = {} as jest.Mocked<SNSessionManager>
    sessionManager.isSignedIntoFirstPartyServer = jest.fn()
    sessionManager.getUser = jest.fn()

    crypto = {} as jest.Mocked<PureCryptoInterface>
    crypto.base64Decode = jest.fn()

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

  describe('experimental features', () => {
    it('enables/disables an experimental feature', async () => {
      storageService.getValue = jest.fn().mockReturnValue(GetFeatures())

      const featuresService = createService()
      featuresService.getExperimentalFeatures = jest.fn().mockReturnValue([FeatureIdentifier.PlusEditor])
      featuresService.initializeFromDisk()

      featuresService.enableExperimentalFeature(FeatureIdentifier.PlusEditor)

      expect(featuresService.isExperimentalFeatureEnabled(FeatureIdentifier.PlusEditor)).toEqual(true)

      featuresService.disableExperimentalFeature(FeatureIdentifier.PlusEditor)

      expect(featuresService.isExperimentalFeatureEnabled(FeatureIdentifier.PlusEditor)).toEqual(false)
    })

    it('does not create a component for not enabled experimental feature', async () => {
      const features = [
        {
          identifier: FeatureIdentifier.PlusEditor,
          expires_at: tomorrow_server,
          content_type: ContentType.Component,
        },
      ]

      apiService.getUserFeatures = jest.fn().mockReturnValue({
        data: {
          features,
        },
      })

      const newRoles = [...roles, RoleName.NAMES.PlusUser]

      storageService.getValue = jest.fn().mockReturnValue(roles)

      const featuresService = createService()
      featuresService.getExperimentalFeatures = jest.fn().mockReturnValue([FeatureIdentifier.PlusEditor])

      featuresService.initializeFromDisk()
      await featuresService.updateOnlineRolesAndFetchFeatures('123', newRoles)
      expect(itemManager.createItem).not.toHaveBeenCalled()
    })

    it('does create a component for enabled experimental feature', async () => {
      apiService.getUserFeatures = jest.fn().mockReturnValue({
        data: {
          features: GetFeatures(),
        },
      })

      const newRoles = [...roles, RoleName.NAMES.PlusUser]

      storageService.getValue = jest.fn().mockReturnValue(roles)

      const featuresService = createService()
      featuresService.getExperimentalFeatures = jest.fn().mockReturnValue([FeatureIdentifier.PlusEditor])

      featuresService.getEnabledExperimentalFeatures = jest.fn().mockReturnValue([FeatureIdentifier.PlusEditor])

      featuresService.initializeFromDisk()
      await featuresService.updateOnlineRolesAndFetchFeatures('123', newRoles)
      expect(itemManager.createItem).toHaveBeenCalled()
    })
  })

  describe('loadUserRoles()', () => {
    it('retrieves user roles and features from storage', async () => {
      createService().initializeFromDisk()
      expect(storageService.getValue).toHaveBeenCalledWith(StorageKey.UserRoles, undefined, [])
      expect(storageService.getValue).toHaveBeenCalledWith(StorageKey.UserFeatures, undefined, [])
    })
  })

  describe('updateRoles()', () => {
    it('setRoles should notify event if roles changed', async () => {
      storageService.getValue = jest.fn().mockReturnValue(roles)
      const featuresService = createService()
      featuresService.initializeFromDisk()

      const mock = (featuresService['notifyEvent'] = jest.fn())

      const newRoles = [...roles, RoleName.NAMES.PlusUser]
      await featuresService.setOnlineRoles(newRoles)

      expect(mock.mock.calls[0][0]).toEqual(FeaturesEvent.UserRolesChanged)
    })

    it('should notify of subscription purchase', async () => {
      storageService.getValue = jest.fn().mockReturnValue(roles)
      const featuresService = createService()
      featuresService.initializeFromDisk()

      const spy = jest.spyOn(featuresService, 'notifyEvent' as never)

      const newRoles = [...roles, RoleName.NAMES.ProUser]
      await featuresService.updateOnlineRolesAndFetchFeatures('123', newRoles)

      expect(spy.mock.calls[2][0]).toEqual(FeaturesEvent.DidPurchaseSubscription)
    })

    it('should not notify of subscription purchase on initial roles load after sign in', async () => {
      storageService.getValue = jest.fn().mockReturnValue(roles)
      const featuresService = createService()
      featuresService.initializeFromDisk()
      featuresService['onlineRoles'] = []

      const spy = jest.spyOn(featuresService, 'notifyEvent' as never)

      const newRoles = [...roles, RoleName.NAMES.ProUser]
      await featuresService.updateOnlineRolesAndFetchFeatures('123', newRoles)

      const triggeredEvents = spy.mock.calls.map((call) => call[0])
      expect(triggeredEvents).not.toContain(FeaturesEvent.DidPurchaseSubscription)
    })

    it('saves new roles to storage and fetches features if a role has been added', async () => {
      const newRoles = [...roles, RoleName.NAMES.PlusUser]

      storageService.getValue = jest.fn().mockReturnValue(roles)
      const featuresService = createService()
      featuresService.initializeFromDisk()
      await featuresService.updateOnlineRolesAndFetchFeatures('123', newRoles)
      expect(storageService.setValue).toHaveBeenCalledWith(StorageKey.UserRoles, newRoles)
      expect(apiService.getUserFeatures).toHaveBeenCalledWith('123')
    })

    it('saves new roles to storage and fetches features if a role has been removed', async () => {
      const newRoles = [RoleName.NAMES.CoreUser]

      storageService.getValue = jest.fn().mockReturnValue(roles)
      const featuresService = createService()
      featuresService.initializeFromDisk()
      await featuresService.updateOnlineRolesAndFetchFeatures('123', newRoles)
      expect(storageService.setValue).toHaveBeenCalledWith(StorageKey.UserRoles, newRoles)
      expect(apiService.getUserFeatures).toHaveBeenCalledWith('123')
    })

    it('saves features to storage when roles change', async () => {
      const newRoles = [...roles, RoleName.NAMES.PlusUser]

      storageService.getValue = jest.fn().mockReturnValue(roles)
      const featuresService = createService()
      featuresService.initializeFromDisk()
      await featuresService.updateOnlineRolesAndFetchFeatures('123', newRoles)
      expect(storageService.setValue).toHaveBeenCalledWith(StorageKey.UserFeatures, features)
    })

    it('creates items for non-expired features with content type if they do not exist', async () => {
      const newRoles = [...roles, RoleName.NAMES.PlusUser]

      storageService.getValue = jest.fn().mockReturnValue(roles)
      const featuresService = createService()
      featuresService.initializeFromDisk()
      await featuresService.updateOnlineRolesAndFetchFeatures('123', newRoles)
      expect(itemManager.createItem).toHaveBeenCalledTimes(2)
      expect(itemManager.createItem).toHaveBeenCalledWith(
        ContentType.Theme,
        expect.objectContaining({
          package_info: expect.objectContaining({
            content_type: ContentType.Theme,
            expires_at: tomorrow_client,
            identifier: FeatureIdentifier.MidnightTheme,
          }),
        }),
        true,
      )
      expect(itemManager.createItem).toHaveBeenCalledWith(
        ContentType.Component,
        expect.objectContaining({
          package_info: expect.objectContaining({
            content_type: ContentType.Component,
            expires_at: tomorrow_client,
            identifier: FeatureIdentifier.PlusEditor,
          }),
        }),
        true,
      )
    })

    it('if item for a feature exists updates its content', async () => {
      const existingItem = new SNComponent({
        uuid: '789',
        content_type: ContentType.Component,
        content: {
          package_info: {
            identifier: FeatureIdentifier.PlusEditor,
            valid_until: new Date(),
          },
        },
      } as never)

      const newRoles = [...roles, RoleName.NAMES.PlusUser]

      storageService.getValue = jest.fn().mockReturnValue(roles)
      itemManager.getItems = jest.fn().mockReturnValue([existingItem])
      const featuresService = createService()
      featuresService.initializeFromDisk()
      await featuresService.updateOnlineRolesAndFetchFeatures('123', newRoles)

      expect(itemManager.changeComponent).toHaveBeenCalledWith(existingItem, expect.any(Function))
    })

    it('creates items for expired components if they do not exist', async () => {
      const newRoles = [...roles, RoleName.NAMES.PlusUser]

      const now = new Date()
      const yesterday_client = now.setDate(now.getDate() - 1)
      const yesterday_server = yesterday_client * 1_000

      storageService.getValue = jest.fn().mockReturnValue(roles)
      apiService.getUserFeatures = jest.fn().mockReturnValue({
        data: {
          features: [
            {
              ...features[1],
              expires_at: yesterday_server,
            },
          ],
        },
      })

      const featuresService = createService()
      featuresService.initializeFromDisk()
      await featuresService.updateOnlineRolesAndFetchFeatures('123', newRoles)
      expect(itemManager.createItem).toHaveBeenCalledWith(
        ContentType.Component,
        expect.objectContaining({
          package_info: expect.objectContaining({
            content_type: ContentType.Component,
            expires_at: yesterday_client,
            identifier: FeatureIdentifier.PlusEditor,
          }),
        }),
        true,
      )
    })

    it('deletes items for expired themes', async () => {
      const existingItem = new SNComponent({
        uuid: '456',
        content_type: ContentType.Theme,
        content: {
          package_info: {
            identifier: FeatureIdentifier.MidnightTheme,
            valid_until: new Date(),
          },
        },
      } as never)

      const newRoles = [...roles, RoleName.NAMES.PlusUser]

      const now = new Date()
      const yesterday = now.setDate(now.getDate() - 1)

      itemManager.changeComponent = jest.fn().mockReturnValue(existingItem)
      storageService.getValue = jest.fn().mockReturnValue(roles)
      itemManager.getItems = jest.fn().mockReturnValue([existingItem])
      apiService.getUserFeatures = jest.fn().mockReturnValue({
        data: {
          features: [
            {
              ...features[0],
              expires_at: yesterday,
            },
          ],
        },
      })

      const featuresService = createService()
      featuresService.initializeFromDisk()
      await featuresService.updateOnlineRolesAndFetchFeatures('123', newRoles)
      expect(itemManager.setItemsToBeDeleted).toHaveBeenCalledWith([existingItem])
    })

    it('does not create an item for a feature without content type', async () => {
      const features = [
        {
          identifier: FeatureIdentifier.TagNesting,
          expires_at: tomorrow_server,
        },
      ]

      apiService.getUserFeatures = jest.fn().mockReturnValue({
        data: {
          features,
        },
      })

      const newRoles = [...roles, RoleName.NAMES.PlusUser]

      storageService.getValue = jest.fn().mockReturnValue(roles)
      const featuresService = createService()
      featuresService.initializeFromDisk()
      await featuresService.updateOnlineRolesAndFetchFeatures('123', newRoles)
      expect(itemManager.createItem).not.toHaveBeenCalled()
    })

    it('does not create an item for deprecated features', async () => {
      const features = [
        {
          identifier: FeatureIdentifier.DeprecatedBoldEditor,
          expires_at: tomorrow_server,
        },
      ]

      apiService.getUserFeatures = jest.fn().mockReturnValue({
        data: {
          features,
        },
      })

      const newRoles = [...roles, RoleName.NAMES.PlusUser]

      storageService.getValue = jest.fn().mockReturnValue(roles)
      const featuresService = createService()
      featuresService.initializeFromDisk()
      await featuresService.updateOnlineRolesAndFetchFeatures('123', newRoles)
      expect(itemManager.createItem).not.toHaveBeenCalled()
    })

    it('does nothing after initial update if roles have not changed', async () => {
      storageService.getValue = jest.fn().mockReturnValue(roles)
      const featuresService = createService()
      featuresService.initializeFromDisk()
      await featuresService.updateOnlineRolesAndFetchFeatures('123', roles)
      await featuresService.updateOnlineRolesAndFetchFeatures('123', roles)
      await featuresService.updateOnlineRolesAndFetchFeatures('123', roles)
      await featuresService.updateOnlineRolesAndFetchFeatures('123', roles)
      expect(storageService.setValue).toHaveBeenCalledTimes(2)
    })

    it('remote native features should be swapped with compiled version', async () => {
      const remoteFeature = {
        identifier: FeatureIdentifier.PlusEditor,
        content_type: ContentType.Component,
        expires_at: tomorrow_server,
      } as FeatureDescription

      const newRoles = [...roles, RoleName.NAMES.PlusUser]

      storageService.getValue = jest.fn().mockReturnValue(roles)
      apiService.getUserFeatures = jest.fn().mockReturnValue({
        data: {
          features: [remoteFeature],
        },
      })

      const featuresService = createService()
      const nativeFeature = featuresService['mapRemoteNativeFeatureToStaticFeature'](remoteFeature)
      featuresService['mapRemoteNativeFeatureToItem'] = jest.fn()
      featuresService.initializeFromDisk()
      await featuresService.updateOnlineRolesAndFetchFeatures('123', newRoles)
      expect(featuresService['mapRemoteNativeFeatureToItem']).toHaveBeenCalledWith(
        nativeFeature,
        expect.anything(),
        expect.anything(),
      )
    })

    it('mapRemoteNativeFeatureToItem should throw if called with client controlled feature', async () => {
      const clientFeature = {
        identifier: FeatureIdentifier.DarkTheme,
        content_type: ContentType.Theme,
        clientControlled: true,
      } as FeatureDescription

      storageService.getValue = jest.fn().mockReturnValue(roles)
      apiService.getUserFeatures = jest.fn().mockReturnValue({
        data: {
          features: [clientFeature],
        },
      })

      const featuresService = createService()
      featuresService.initializeFromDisk()
      await expect(() => featuresService['mapRemoteNativeFeatureToItem'](clientFeature, [], [])).rejects.toThrow()
    })

    it('role-based feature status', async () => {
      const featuresService = createService()

      features = [] as jest.Mocked<FeatureDescription[]>

      apiService.getUserFeatures = jest.fn().mockReturnValue({
        data: {
          features,
        },
      })

      sessionManager.isSignedIntoFirstPartyServer = jest.fn().mockReturnValue(true)

      await featuresService.updateOnlineRolesAndFetchFeatures('123', [RoleName.NAMES.CoreUser, RoleName.NAMES.PlusUser])

      expect(featuresService.getFeatureStatus(FeatureIdentifier.MidnightTheme)).toBe(FeatureStatus.Entitled)
      expect(featuresService.getFeatureStatus(FeatureIdentifier.SuperEditor)).toBe(FeatureStatus.Entitled)
    })

    it('feature status with no paid role but features listings', async () => {
      const featuresService = createService()

      features = [
        {
          identifier: FeatureIdentifier.MidnightTheme,
          content_type: ContentType.Theme,
          expires_at: tomorrow_server,
          role_name: RoleName.NAMES.PlusUser,
        },
        {
          identifier: FeatureIdentifier.PlusEditor,
          content_type: ContentType.Component,
          expires_at: expiredDate,
          role_name: RoleName.NAMES.ProUser,
        },
      ] as jest.Mocked<FeatureDescription[]>

      apiService.getUserFeatures = jest.fn().mockReturnValue({
        data: {
          features,
        },
      })

      sessionManager.isSignedIntoFirstPartyServer = jest.fn().mockReturnValue(true)

      await featuresService.updateOnlineRolesAndFetchFeatures('123', [RoleName.NAMES.CoreUser])

      expect(featuresService.getFeatureStatus(FeatureIdentifier.MidnightTheme)).toBe(FeatureStatus.NoUserSubscription)
      expect(featuresService.getFeatureStatus(FeatureIdentifier.PlusEditor)).toBe(FeatureStatus.NoUserSubscription)
      expect(featuresService.getFeatureStatus(FeatureIdentifier.SheetsEditor)).toBe(FeatureStatus.NoUserSubscription)
    })

    it('role-based features while not signed into first party server', async () => {
      const featuresService = createService()

      sessionManager.isSignedIntoFirstPartyServer = jest.fn().mockReturnValue(false)

      await featuresService.updateOnlineRolesAndFetchFeatures('123', [RoleName.NAMES.ProUser])

      expect(featuresService.getFeatureStatus(FeatureIdentifier.SuperEditor)).toBe(FeatureStatus.NotInCurrentPlan)
    })

    it('third party feature status', async () => {
      const featuresService = createService()

      const themeFeature = {
        identifier: 'third-party-theme' as FeatureIdentifier,
        content_type: ContentType.Theme,
        expires_at: tomorrow_server,
        role_name: RoleName.NAMES.CoreUser,
      }

      const editorFeature = {
        identifier: 'third-party-editor' as FeatureIdentifier,
        content_type: ContentType.Component,
        expires_at: expiredDate,
        role_name: RoleName.NAMES.PlusUser,
      }

      features = [themeFeature, editorFeature] as jest.Mocked<FeatureDescription[]>

      featuresService['features'] = features

      itemManager.getDisplayableComponents = jest.fn().mockReturnValue([
        new SNComponent({
          uuid: '123',
          content_type: ContentType.Theme,
          content: {
            valid_until: themeFeature.expires_at,
            package_info: {
              ...themeFeature,
            },
          },
        } as never),
        new SNComponent({
          uuid: '456',
          content_type: ContentType.Component,
          content: {
            valid_until: new Date(editorFeature.expires_at),
            package_info: {
              ...editorFeature,
            },
          },
        } as never),
      ])

      await featuresService.updateOnlineRolesAndFetchFeatures('123', [RoleName.NAMES.CoreUser])

      expect(featuresService.getFeatureStatus(themeFeature.identifier)).toBe(FeatureStatus.Entitled)
      expect(featuresService.getFeatureStatus(editorFeature.identifier)).toBe(FeatureStatus.InCurrentPlanButExpired)
      expect(featuresService.getFeatureStatus('missing-feature-identifier' as FeatureIdentifier)).toBe(
        FeatureStatus.NoUserSubscription,
      )
    })

    it('feature status should be not entitled if no account or offline repo', async () => {
      const featuresService = createService()

      await featuresService.updateOnlineRolesAndFetchFeatures('123', [RoleName.NAMES.CoreUser])

      sessionManager.isSignedIntoFirstPartyServer = jest.fn().mockReturnValue(false)

      featuresService['completedSuccessfulFeaturesRetrieval'] = false

      expect(featuresService.getFeatureStatus(FeatureIdentifier.MidnightTheme)).toBe(FeatureStatus.NoUserSubscription)
      expect(featuresService.getFeatureStatus(FeatureIdentifier.TokenVaultEditor)).toBe(
        FeatureStatus.NoUserSubscription,
      )
    })

    it('didDownloadFeatures should filter out client controlled features', async () => {
      const featuresService = createService()

      featuresService['mapRemoteNativeFeaturesToItems'] = jest.fn()

      await featuresService.didDownloadFeatures(GetFeatures().filter((f) => f.clientControlled))

      expect(featuresService['mapRemoteNativeFeaturesToItems']).toHaveBeenCalledWith([])
    })

    it('feature status for offline subscription', async () => {
      const featuresService = createService()

      await featuresService.updateOnlineRolesAndFetchFeatures('123', [RoleName.NAMES.CoreUser, RoleName.NAMES.PlusUser])

      sessionManager.isSignedIntoFirstPartyServer = jest.fn().mockReturnValue(false)
      featuresService.onlineRolesIncludePaidSubscription = jest.fn().mockReturnValue(false)
      featuresService['completedSuccessfulFeaturesRetrieval'] = true

      expect(featuresService.getFeatureStatus(FeatureIdentifier.MidnightTheme)).toBe(FeatureStatus.NoUserSubscription)
      expect(featuresService.getFeatureStatus(FeatureIdentifier.TokenVaultEditor)).toBe(
        FeatureStatus.NoUserSubscription,
      )

      featuresService.hasOfflineRepo = jest.fn().mockReturnValue(true)
      featuresService.hasFirstPartySubscription = jest.fn().mockReturnValue(true)
      await featuresService.setOfflineRoles([RoleName.NAMES.CoreUser, RoleName.NAMES.PlusUser])

      expect(featuresService.getFeatureStatus(FeatureIdentifier.MidnightTheme)).toBe(FeatureStatus.Entitled)
      expect(featuresService.getFeatureStatus(FeatureIdentifier.TokenVaultEditor)).toBe(FeatureStatus.Entitled)
    })

    it('feature status for deprecated feature', async () => {
      const featuresService = createService()

      sessionManager.isSignedIntoFirstPartyServer = jest.fn().mockReturnValue(true)

      expect(featuresService.getFeatureStatus(FeatureIdentifier.DeprecatedFileSafe as FeatureIdentifier)).toBe(
        FeatureStatus.NoUserSubscription,
      )

      await featuresService.updateOnlineRolesAndFetchFeatures('123', [RoleName.NAMES.CoreUser, RoleName.NAMES.PlusUser])

      expect(featuresService.getFeatureStatus(FeatureIdentifier.DeprecatedFileSafe as FeatureIdentifier)).toBe(
        FeatureStatus.Entitled,
      )
    })

    it('has paid subscription', async () => {
      const featuresService = createService()

      await featuresService.updateOnlineRolesAndFetchFeatures('123', [RoleName.NAMES.CoreUser])
      sessionManager.isSignedIntoFirstPartyServer = jest.fn().mockReturnValue(true)

      expect(featuresService.hasPaidAnyPartyOnlineOrOfflineSubscription()).toBeFalsy

      await featuresService.updateOnlineRolesAndFetchFeatures('123', [RoleName.NAMES.CoreUser, RoleName.NAMES.PlusUser])

      expect(featuresService.hasPaidAnyPartyOnlineOrOfflineSubscription()).toEqual(true)
    })

    it('has paid subscription should be true if offline repo and signed into third party server', async () => {
      const featuresService = createService()

      await featuresService.updateOnlineRolesAndFetchFeatures('123', [RoleName.NAMES.CoreUser])

      featuresService.hasOfflineRepo = jest.fn().mockReturnValue(true)
      sessionManager.isSignedIntoFirstPartyServer = jest.fn().mockReturnValue(false)

      expect(featuresService.hasPaidAnyPartyOnlineOrOfflineSubscription()).toEqual(true)
    })
  })

  describe('migrateFeatureRepoToUserSetting', () => {
    it('should extract key from extension repo url and update user setting', async () => {
      const extensionKey = '129b029707e3470c94a8477a437f9394'
      const extensionRepoItem = new SNFeatureRepo({
        uuid: '456',
        content_type: ContentType.ExtensionRepo,
        content: {
          url: `https://extensions.standardnotes.org/${extensionKey}`,
        },
      } as never)

      const featuresService = createService()
      await featuresService.migrateFeatureRepoToUserSetting([extensionRepoItem])
      expect(settingsService.updateSetting).toHaveBeenCalledWith(SettingName.create(SettingName.NAMES.ExtensionKey).getValue(), extensionKey, true)
    })
  })

  describe('downloadExternalFeature', () => {
    it('should not allow if identifier matches native identifier', async () => {
      apiService.downloadFeatureUrl = jest.fn().mockReturnValue({
        data: {
          identifier: 'org.standardnotes.bold-editor',
          name: 'Bold Editor',
          content_type: 'SN|Component',
          area: 'editor-editor',
          version: '1.0.0',
          url: 'http://localhost:8005/',
        },
      })

      const installUrl = 'http://example.com'
      crypto.base64Decode = jest.fn().mockReturnValue(installUrl)

      const featuresService = createService()
      const result = await featuresService.downloadExternalFeature(installUrl)
      expect(result).toBeUndefined()
    })

    it('should not allow if url matches native url', async () => {
      apiService.downloadFeatureUrl = jest.fn().mockReturnValue({
        data: {
          identifier: 'org.foo.bar',
          name: 'Bold Editor',
          content_type: 'SN|Component',
          area: 'editor-editor',
          version: '1.0.0',
          url: 'http://localhost:8005/org.standardnotes.bold-editor/index.html',
        },
      })

      const installUrl = 'http://example.com'
      crypto.base64Decode = jest.fn().mockReturnValue(installUrl)

      const featuresService = createService()
      const result = await featuresService.downloadExternalFeature(installUrl)
      expect(result).toBeUndefined()
    })
  })

  describe('sortRolesByHierarchy', () => {
    it('should sort given roles according to role hierarchy', () => {
      const featuresService = createService()

      const sortedRoles = featuresService.rolesBySorting([
        RoleName.NAMES.ProUser,
        RoleName.NAMES.CoreUser,
        RoleName.NAMES.PlusUser,
      ])

      expect(sortedRoles).toStrictEqual([RoleName.NAMES.CoreUser, RoleName.NAMES.PlusUser, RoleName.NAMES.ProUser])
    })
  })

  describe('hasMinimumRole', () => {
    it('should be false if core user checks for plus role', async () => {
      const featuresService = createService()

      await featuresService.updateOnlineRolesAndFetchFeatures('123', [RoleName.NAMES.CoreUser])

      const hasPlusUserRole = featuresService.hasMinimumRole(RoleName.NAMES.PlusUser)

      expect(hasPlusUserRole).toBe(false)
    })

    it('should be false if plus user checks for pro role', async () => {
      const featuresService = createService()

      sessionManager.isSignedIntoFirstPartyServer = jest.fn().mockReturnValue(true)

      await featuresService.updateOnlineRolesAndFetchFeatures('123', [RoleName.NAMES.PlusUser, RoleName.NAMES.CoreUser])

      const hasProUserRole = featuresService.hasMinimumRole(RoleName.NAMES.ProUser)

      expect(hasProUserRole).toBe(false)
    })

    it('should be true if pro user checks for core user', async () => {
      const featuresService = createService()

      sessionManager.isSignedIntoFirstPartyServer = jest.fn().mockReturnValue(true)

      await featuresService.updateOnlineRolesAndFetchFeatures('123', [RoleName.NAMES.ProUser, RoleName.NAMES.PlusUser])

      const hasCoreUserRole = featuresService.hasMinimumRole(RoleName.NAMES.CoreUser)

      expect(hasCoreUserRole).toBe(true)
    })

    it('should be true if pro user checks for pro user', async () => {
      const featuresService = createService()

      sessionManager.isSignedIntoFirstPartyServer = jest.fn().mockReturnValue(true)

      await featuresService.updateOnlineRolesAndFetchFeatures('123', [RoleName.NAMES.ProUser, RoleName.NAMES.PlusUser])

      const hasProUserRole = featuresService.hasMinimumRole(RoleName.NAMES.ProUser)

      expect(hasProUserRole).toBe(true)
    })
  })
})