supabase/apps/studio/lib/upload.test.ts

import { createClient } from '@supabase/supabase-js'
import { beforeEach, describe, expect, it, vi } from 'vitest'

import { uploadAttachment } from './upload'

vi.mock('@supabase/supabase-js', () => ({
  createClient: vi.fn(),
}))

describe('uploadAttachment', () => {
  const mockUpload = vi.fn()
  const mockGetPublicUrl = vi.fn()

  const mockStorage = {
    from: vi.fn(() => ({
      upload: mockUpload,
      getPublicUrl: mockGetPublicUrl,
    })),
  }

  const mockSupabaseClient = {
    storage: mockStorage,
  }

  const mockFile = new File(['test'], 'test.png', { type: 'image/png' })

  beforeEach(() => {
    vi.clearAllMocks()
    ;(createClient as any).mockReturnValue(mockSupabaseClient)
  })

  it('uploads file and returns public URL when getUrl is true', async () => {
    mockUpload.mockResolvedValue({ data: { path: 'folder/test.png' }, error: null })
    mockGetPublicUrl.mockReturnValue({
      data: { publicUrl: 'https://cdn.supabase.io/folder/test.png' },
    })

    const url = await uploadAttachment('bucket', 'test.png', mockFile, true)

    expect(createClient).toHaveBeenCalled()
    expect(mockUpload).toHaveBeenCalledWith('test.png', mockFile, { cacheControl: '3600' })
    expect(mockGetPublicUrl).toHaveBeenCalledWith('folder/test.png')
    expect(url).toBe('https://cdn.supabase.io/folder/test.png')
  })

  it('returns undefined if upload succeeds but getUrl is false', async () => {
    mockUpload.mockResolvedValue({ data: { path: 'folder/test.png' }, error: null })

    const result = await uploadAttachment('bucket', 'test.png', mockFile, false)
    expect(result).toBeUndefined()
  })

  it('returns undefined and logs if upload fails', async () => {
    const errorSpy = vi.spyOn(console, 'error').mockImplementation(() => {})
    mockUpload.mockResolvedValue({ data: null, error: { message: 'Upload failed' } })

    const result = await uploadAttachment('bucket', 'test.png', mockFile)
    expect(result).toBeUndefined()
    expect(errorSpy).toHaveBeenCalledWith('Failed to upload:', { message: 'Upload failed' })

    errorSpy.mockRestore()
  })

  it('returns undefined if getPublicUrl returns no data', async () => {
    mockUpload.mockResolvedValue({ data: { path: 'folder/test.png' }, error: null })
    mockGetPublicUrl.mockReturnValue({ data: null })

    const result = await uploadAttachment('bucket', 'test.png', mockFile, true)
    expect(result).toBeUndefined()
  })
})