cherry-studio/packages/shared/__tests__/utils.test.ts

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

import { isBase64ImageDataUrl, isDataUrl, parseDataUrl } from '../utils'

describe('parseDataUrl', () => {
  it('parses a standard base64 image data URL', () => {
    const result = parseDataUrl('data:image/png;base64,iVBORw0KGgo=')
    expect(result).toEqual({
      mediaType: 'image/png',
      isBase64: true,
      data: 'iVBORw0KGgo='
    })
  })

  it('parses a base64 data URL with additional parameters', () => {
    const result = parseDataUrl('data:image/jpeg;name=foo;base64,/9j/4AAQ')
    expect(result).toEqual({
      mediaType: 'image/jpeg',
      isBase64: true,
      data: '/9j/4AAQ'
    })
  })

  it('parses a plain text data URL (non-base64)', () => {
    const result = parseDataUrl('data:text/plain,Hello%20World')
    expect(result).toEqual({
      mediaType: 'text/plain',
      isBase64: false,
      data: 'Hello%20World'
    })
  })

  it('parses a data URL with empty media type', () => {
    const result = parseDataUrl('data:;base64,SGVsbG8=')
    expect(result).toEqual({
      mediaType: undefined,
      isBase64: true,
      data: 'SGVsbG8='
    })
  })

  it('returns null for non-data URLs', () => {
    const result = parseDataUrl('https://example.com/image.png')
    expect(result).toBeNull()
  })

  it('returns null for malformed data URL without comma', () => {
    const result = parseDataUrl('data:image/png;base64')
    expect(result).toBeNull()
  })

  it('handles empty string', () => {
    const result = parseDataUrl('')
    expect(result).toBeNull()
  })

  it('handles large base64 data without performance issues', () => {
    // Simulate a 4K image base64 string (about 1MB)
    const largeData = 'A'.repeat(1024 * 1024)
    const dataUrl = `data:image/png;base64,${largeData}`

    const start = performance.now()
    const result = parseDataUrl(dataUrl)
    const duration = performance.now() - start

    expect(result).not.toBeNull()
    expect(result?.mediaType).toBe('image/png')
    expect(result?.isBase64).toBe(true)
    expect(result?.data).toBe(largeData)
    // Should complete in under 10ms (string operations are fast)
    expect(duration).toBeLessThan(10)
  })

  it('parses SVG data URL', () => {
    const result = parseDataUrl('data:image/svg+xml;base64,PHN2Zz4=')
    expect(result).toEqual({
      mediaType: 'image/svg+xml',
      isBase64: true,
      data: 'PHN2Zz4='
    })
  })

  it('parses JSON data URL', () => {
    const result = parseDataUrl('data:application/json,{"key":"value"}')
    expect(result).toEqual({
      mediaType: 'application/json',
      isBase64: false,
      data: '{"key":"value"}'
    })
  })
})

describe('isDataUrl', () => {
  it('returns true for valid data URLs', () => {
    expect(isDataUrl('data:image/png;base64,ABC')).toBe(true)
    expect(isDataUrl('data:text/plain,hello')).toBe(true)
    expect(isDataUrl('data:,simple')).toBe(true)
  })

  it('returns false for non-data URLs', () => {
    expect(isDataUrl('https://example.com')).toBe(false)
    expect(isDataUrl('file:///path/to/file')).toBe(false)
    expect(isDataUrl('')).toBe(false)
  })

  it('returns false for malformed data URLs', () => {
    expect(isDataUrl('data:')).toBe(false)
    expect(isDataUrl('data:image/png')).toBe(false)
  })
})

describe('isBase64ImageDataUrl', () => {
  it('returns true for base64 image data URLs', () => {
    expect(isBase64ImageDataUrl('data:image/png;base64,ABC')).toBe(true)
    expect(isBase64ImageDataUrl('data:image/jpeg;base64,/9j/')).toBe(true)
    expect(isBase64ImageDataUrl('data:image/gif;base64,R0lG')).toBe(true)
    expect(isBase64ImageDataUrl('data:image/webp;base64,UklG')).toBe(true)
  })

  it('returns false for non-base64 image data URLs', () => {
    expect(isBase64ImageDataUrl('data:image/svg+xml,<svg></svg>')).toBe(false)
  })

  it('returns false for non-image data URLs', () => {
    expect(isBase64ImageDataUrl('data:text/plain;base64,SGVsbG8=')).toBe(false)
    expect(isBase64ImageDataUrl('data:application/json,{}')).toBe(false)
  })

  it('returns false for regular URLs', () => {
    expect(isBase64ImageDataUrl('https://example.com/image.png')).toBe(false)
    expect(isBase64ImageDataUrl('file:///image.png')).toBe(false)
  })

  it('returns false for malformed data URLs', () => {
    expect(isBase64ImageDataUrl('data:image/png')).toBe(false)
    expect(isBase64ImageDataUrl('')).toBe(false)
  })
})