supabase/apps/studio/components/interfaces/ProjectCreation/ProjectCreation.utils.test.ts

import { AWS_REGIONS } from 'shared-data'
import { describe, expect, it } from 'vitest'

import {
  HIGH_AVAILABILITY_INSTANCE_SIZE,
  HIGH_AVAILABILITY_POSTGRES_ENGINE,
  HIGH_AVAILABILITY_RELEASE_CHANNEL,
} from './ProjectCreation.constants'
import type { AvailableRegions } from './ProjectCreation.utils'
import {
  filterHighAvailabilityRegions,
  getAvailableRegions,
  getFreeTierGeneralRegionExperimentVariant,
  getHighAvailabilityRegionCode,
  getRegionSelectionType,
  resolveDefaultDbRegion,
} from './ProjectCreation.utils'

describe('resolveDefaultDbRegion', () => {
  const base = {
    cloudProvider: 'AWS_NIMBUS',
    isHighAvailabilityRestricted: false,
    highAvailabilityRegionName: undefined,
    isSmartRegionEnabled: false,
    recommendedSmartRegion: undefined,
    autoDefaultRegion: undefined,
    fixedDefaultRegion: AWS_REGIONS.EAST_US.displayName,
    environment: 'prod',
  } as const

  it('prefers the high availability region when restricted, even while it is still loading', () => {
    expect(
      resolveDefaultDbRegion({
        ...base,
        isHighAvailabilityRestricted: true,
        highAvailabilityRegionName: AWS_REGIONS.EAST_US.displayName,
      })
    ).toBe(AWS_REGIONS.EAST_US.displayName)
    expect(resolveDefaultDbRegion({ ...base, isHighAvailabilityRestricted: true })).toBeUndefined()
  })

  it('uses the recommended smart region when smart regions are enabled', () => {
    expect(
      resolveDefaultDbRegion({
        ...base,
        cloudProvider: 'AWS',
        isSmartRegionEnabled: true,
        recommendedSmartRegion: 'Americas',
        autoDefaultRegion: AWS_REGIONS.SOUTHEAST_ASIA.displayName,
      })
    ).toBe('Americas')
  })

  it('uses the geolocated region when the provider offers it', () => {
    expect(
      resolveDefaultDbRegion({
        ...base,
        cloudProvider: 'AWS',
        autoDefaultRegion: AWS_REGIONS.SOUTHEAST_ASIA.displayName,
      })
    ).toBe(AWS_REGIONS.SOUTHEAST_ASIA.displayName)
  })

  it('falls back to the fixed default when the provider does not offer the geolocated region', () => {
    expect(
      resolveDefaultDbRegion({ ...base, autoDefaultRegion: AWS_REGIONS.SOUTHEAST_ASIA.displayName })
    ).toBe(AWS_REGIONS.EAST_US.displayName)
  })

  it('falls back to the fixed default when no geolocated region resolved', () => {
    expect(resolveDefaultDbRegion(base)).toBe(AWS_REGIONS.EAST_US.displayName)
  })
})

describe('getAvailableRegions', () => {
  it.each(['local', 'staging', 'prod'])('returns all AWS regions for AWS on %s', (environment) => {
    expect(getAvailableRegions('AWS', environment)).toEqual(AWS_REGIONS)
    expect(getAvailableRegions('AWS_K8S', environment)).toEqual(AWS_REGIONS)
  })

  it.each([
    ['local', { SOUTHEAST_ASIA: AWS_REGIONS.SOUTHEAST_ASIA }],
    ['staging', { SOUTHEAST_ASIA: AWS_REGIONS.SOUTHEAST_ASIA }],
    ['prod', { EAST_US: AWS_REGIONS.EAST_US }],
  ])('returns the single AWS_NIMBUS region on %s', (environment, expectedRegions) => {
    expect(getAvailableRegions('AWS_NIMBUS', environment)).toEqual(expectedRegions)
  })
})

describe('High Availability project creation constraints', () => {
  it('pins the Alpha Postgres engine, release channel, and compute size', () => {
    expect(HIGH_AVAILABILITY_POSTGRES_ENGINE).toBe('17')
    expect(HIGH_AVAILABILITY_RELEASE_CHANNEL).toBe('ga')
    expect(HIGH_AVAILABILITY_INSTANCE_SIZE).toBe('large')
  })

  it.each([
    ['local', 'eu-central-1'],
    ['staging', 'us-east-1'],
    ['prod', 'us-east-1'],
  ])('resolves the %s region restriction', (environment, expectedRegion) => {
    expect(getHighAvailabilityRegionCode(environment)).toBe(expectedRegion)
  })

  it.each([
    ['local', undefined],
    ['staging', 'us-east-1'],
    ['prod', 'us-east-1'],
  ])(
    'applies the %s region restriction to high availability projects',
    (environment, expectedRegion) => {
      const regions = [{ code: 'us-east-1' }, { code: 'eu-central-1' }, { code: 'ap-southeast-1' }]

      expect(filterHighAvailabilityRegions(regions, true, environment)).toEqual(
        expectedRegion === undefined ? regions : [{ code: expectedRegion }]
      )
      expect(filterHighAvailabilityRegions(regions, false, environment)).toEqual(regions)
    }
  )
})

describe('getFreeTierGeneralRegionExperimentVariant', () => {
  const base = {
    isFreePlan: true,
    smartRegionEnabled: true,
    enrollmentFlag: true,
    selectionFlag: true,
  }

  it('returns test when in the experiment and the selection flag is on', () => {
    expect(getFreeTierGeneralRegionExperimentVariant(base)).toBe('test')
  })

  it('returns control when in the experiment and the selection flag is off', () => {
    expect(getFreeTierGeneralRegionExperimentVariant({ ...base, selectionFlag: false })).toBe(
      'control'
    )
  })

  it('returns undefined for a paid plan', () => {
    expect(
      getFreeTierGeneralRegionExperimentVariant({ ...base, isFreePlan: false })
    ).toBeUndefined()
  })

  it('returns undefined when smart regions are disabled (AWS_NIMBUS)', () => {
    expect(
      getFreeTierGeneralRegionExperimentVariant({ ...base, smartRegionEnabled: false })
    ).toBeUndefined()
  })

  it('returns undefined when the enrollment flag is off', () => {
    expect(
      getFreeTierGeneralRegionExperimentVariant({ ...base, enrollmentFlag: false })
    ).toBeUndefined()
  })

  it('returns undefined when the enrollment flag is off even if the selection flag is on', () => {
    expect(
      getFreeTierGeneralRegionExperimentVariant({
        ...base,
        enrollmentFlag: false,
        selectionFlag: true,
      })
    ).toBeUndefined()
  })
})

describe('getRegionSelectionType', () => {
  const availableRegions: AvailableRegions = {
    smartGroup: [{ code: 'americas', name: 'Americas', type: 'smartGroup' }],
    specific: [
      { code: 'us-east-1', name: 'East US (North Virginia)', provider: 'AWS', type: 'specific' },
    ],
  }

  it('classifies a smart group region as general', () => {
    expect(getRegionSelectionType({ dbRegion: 'Americas', availableRegions })).toBe('general')
  })

  it('classifies a specific region as specific', () => {
    expect(getRegionSelectionType({ dbRegion: 'East US (North Virginia)', availableRegions })).toBe(
      'specific'
    )
  })

  it('returns undefined when the region matches neither list', () => {
    expect(getRegionSelectionType({ dbRegion: 'Nonexistent', availableRegions })).toBeUndefined()
  })

  it('returns undefined when dbRegion is undefined', () => {
    expect(getRegionSelectionType({ dbRegion: undefined, availableRegions })).toBeUndefined()
  })

  it('returns undefined when availableRegions is undefined', () => {
    expect(
      getRegionSelectionType({ dbRegion: 'Americas', availableRegions: undefined })
    ).toBeUndefined()
  })
})