supabase/apps/studio/components/v2/Navigation/ProjectBranchSelector.utils.test.ts

import { describe, expect, it } from 'vitest'

import { getProjectBranchSelectorState } from './ProjectBranchSelector.utils'

describe('getProjectBranchSelectorState', () => {
  it('returns main branch when branching disabled', () => {
    const result = getProjectBranchSelectorState({
      selectedBranch: undefined,
      isBranchingEnabled: false,
      selectedOrganization: undefined,
    })
    expect(result.isMainBranch).toBe(true)
    expect(result.branchDisplayName).toBe('main')
  })

  it('returns main when branching enabled and selected branch is default', () => {
    const result = getProjectBranchSelectorState({
      selectedBranch: {
        id: '1',
        name: 'main',
        project_ref: 'ref',
        is_default: true,
        created_at: '',
      } as any,
      isBranchingEnabled: true,
      selectedOrganization: undefined,
    })
    expect(result.isMainBranch).toBe(true)
    expect(result.branchDisplayName).toBe('main')
  })

  it('returns non-main when branching enabled and selected branch is not default', () => {
    const result = getProjectBranchSelectorState({
      selectedBranch: {
        id: '2',
        name: 'feature-x',
        project_ref: 'ref-2',
        is_default: false,
        created_at: '',
      } as any,
      isBranchingEnabled: true,
      selectedOrganization: undefined,
    })
    expect(result.isMainBranch).toBe(false)
    expect(result.branchDisplayName).toBe('feature-x')
  })

  it('returns main when selected branch undefined and branching enabled', () => {
    const result = getProjectBranchSelectorState({
      selectedBranch: undefined,
      isBranchingEnabled: true,
      selectedOrganization: undefined,
    })
    expect(result.branchDisplayName).toBe('main')
  })

  it('returns organization href when org has slug', () => {
    const result = getProjectBranchSelectorState({
      selectedBranch: undefined,
      isBranchingEnabled: false,
      selectedOrganization: { slug: 'my-org', name: 'My Org' } as any,
    })
    expect(result.organizationHref).toBe('/org/my-org')
    expect(result.selectedOrgInitial).toBe('M')
  })

  it('returns organizations fallback when no org slug', () => {
    const result = getProjectBranchSelectorState({
      selectedBranch: undefined,
      isBranchingEnabled: false,
      selectedOrganization: undefined,
    })
    expect(result.organizationHref).toBe('/organizations')
    expect(result.selectedOrgInitial).toBe('O')
  })

  it('handles org name with leading/trailing spaces for initial', () => {
    const result = getProjectBranchSelectorState({
      selectedBranch: undefined,
      isBranchingEnabled: false,
      selectedOrganization: { slug: 'x', name: '  Acme  ' } as any,
    })
    expect(result.selectedOrgInitial).toBe('A')
  })
})