@lobehub/chat/packages/utils/src/client/videoValidation.test.ts

Lobe Chat - an open-source, high-performance chatbot framework that supports speech synthesis, multimodal, and extensible Function Call plugin system. Supports one-click free deployment of your private ChatGPT/LLM web application.

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

import { validateVideoFileSize } from './videoValidation';

describe('validateVideoFileSize', () => {
  it('should return valid for non-video files', () => {
    const mockFile = new File(['test'], 'test.txt', { type: 'text/plain' });
    const result = validateVideoFileSize(mockFile);

    expect(result.isValid).toBe(true);
    expect(result.actualSize).toBeUndefined();
  });

  it('should return valid for video files under 20MB', () => {
    const mockVideoFile = new File(['x'.repeat(10 * 1024 * 1024)], 'video.mp4', {
      type: 'video/mp4',
    });
    const result = validateVideoFileSize(mockVideoFile);

    expect(result.isValid).toBe(true);
    expect(result.actualSize).toBe('10.0 MB');
  });

  it('should return invalid for video files over 20MB', () => {
    const mockLargeVideoFile = new File(['x'.repeat(25 * 1024 * 1024)], 'large-video.mp4', {
      type: 'video/mp4',
    });
    const result = validateVideoFileSize(mockLargeVideoFile);

    expect(result.isValid).toBe(false);
    expect(result.actualSize).toBe('25.0 MB');
  });

  it('should return invalid for video files exactly at 20MB limit plus 1 byte', () => {
    const mockBoundaryFile = new File(['x'.repeat(20 * 1024 * 1024 + 1)], 'boundary.mp4', {
      type: 'video/mp4',
    });
    const result = validateVideoFileSize(mockBoundaryFile);

    expect(result.isValid).toBe(false);
    expect(result.actualSize).toBe('20.0 MB');
  });

  it('should return valid for video files exactly at 20MB limit', () => {
    const mockBoundaryFile = new File(['x'.repeat(20 * 1024 * 1024)], 'boundary.mp4', {
      type: 'video/mp4',
    });
    const result = validateVideoFileSize(mockBoundaryFile);

    expect(result.isValid).toBe(true);
    expect(result.actualSize).toBe('20.0 MB');
  });
});