import { describe, expect, it, vi } from 'vitest'
import { resolveWorktreeAddBaseRef, worktreeBaseRefFamily } from './base-ref'

describe('resolveWorktreeAddBaseRef', () => {
  it('leaves fully qualified refs unchanged', async () => {
    const refExists = vi.fn()

    await expect(resolveWorktreeAddBaseRef('refs/heads/main', refExists)).resolves.toBe(
      'refs/heads/main'
    )

    expect(refExists).not.toHaveBeenCalled()
  })

  it('leaves provider review refs unchanged', async () => {
    const refExists = vi.fn()

    await expect(resolveWorktreeAddBaseRef('refs/pull/123/head', refExists)).resolves.toBe(
      'refs/pull/123/head'
    )
    await expect(
      resolveWorktreeAddBaseRef('refs/merge-requests/456/head', refExists)
    ).resolves.toBe('refs/merge-requests/456/head')

    expect(refExists).not.toHaveBeenCalled()
  })

  it('qualifies a bare local branch name', async () => {
    const refExists = vi.fn(async (ref: string) => ref === 'refs/heads/main')

    await expect(resolveWorktreeAddBaseRef('main', refExists)).resolves.toBe('refs/heads/main')

    expect(refExists).toHaveBeenCalledWith('refs/heads/main')
  })

  it('prefers a remote-tracking ref for remote-display names', async () => {
    const refExists = vi.fn(async (ref: string) => ref === 'refs/remotes/origin/main')

    await expect(resolveWorktreeAddBaseRef('origin/main', refExists)).resolves.toBe(
      'refs/remotes/origin/main'
    )

    expect(refExists).toHaveBeenCalledWith('refs/remotes/origin/main')
  })

  it('qualifies a slash-containing local branch when no matching remote ref exists', async () => {
    const refExists = vi.fn(async (ref: string) => ref === 'refs/heads/release/main')

    await expect(resolveWorktreeAddBaseRef('release/main', refExists)).resolves.toBe(
      'refs/heads/release/main'
    )

    expect(refExists.mock.calls.map((call) => call[0])).toEqual([
      'refs/remotes/release/main',
      'refs/heads/release/main'
    ])
  })

  it('keeps unresolvable revisions untouched', async () => {
    const refExists = vi.fn(async () => false)

    await expect(resolveWorktreeAddBaseRef('abc1234', refExists)).resolves.toBe('abc1234')
  })
})

describe('worktreeBaseRefFamily', () => {
  it('gives a local branch and its remote-tracking copies the same family', () => {
    expect(worktreeBaseRefFamily('refs/heads/main')).toBe('main')
    expect(worktreeBaseRefFamily('refs/remotes/origin/main')).toBe('main')
    expect(worktreeBaseRefFamily('refs/remotes/upstream/main')).toBe('main')
  })

  it('keeps the full branch path for slash-containing branches', () => {
    expect(worktreeBaseRefFamily('refs/heads/release/24.1')).toBe('release/24.1')
    expect(worktreeBaseRefFamily('refs/remotes/origin/release/24.1')).toBe('release/24.1')
  })

  it('separates different branches', () => {
    expect(worktreeBaseRefFamily('refs/heads/main')).not.toBe(
      worktreeBaseRefFamily('refs/heads/release')
    )
  })

  it('has no family for anything that is not a branch ref', () => {
    expect(worktreeBaseRefFamily('abc1234')).toBeNull()
    expect(worktreeBaseRefFamily('main')).toBeNull()
    expect(worktreeBaseRefFamily('refs/tags/v1.0.0')).toBeNull()
    expect(worktreeBaseRefFamily('refs/pull/123/head')).toBeNull()
    expect(worktreeBaseRefFamily('refs/remotes/origin/HEAD')).toBeNull()
    expect(worktreeBaseRefFamily('refs/remotes/origin')).toBeNull()
  })
})