@lobehub/chat/src/app/(backend)/webapi/chat/google/route.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.

// @vitest-environment edge-runtime
import { describe, expect, it, vi } from 'vitest';

import { POST as UniverseRoute } from '../[provider]/route';
import { POST, preferredRegion, runtime } from './route';

// 模拟 '../[provider]/route'
vi.mock('../[provider]/route', () => ({
  POST: vi.fn().mockResolvedValue('mocked response'),
}));

describe('Configuration tests', () => {
  it('should have runtime set to "edge"', () => {
    expect(runtime).toBe('edge');
  });

  it('should contain specific regions in preferredRegion', () => {
    expect(preferredRegion).not.contain(['hkg1']);
    expect(preferredRegion).not.contain(['dub1']);
    expect(preferredRegion).not.contain(['cdg1']);
    expect(preferredRegion).not.contain(['fra1']);
    expect(preferredRegion).not.contain(['lhr1']);
    expect(preferredRegion).not.contain(['arn1']);
  });
});

describe('Google POST function tests', () => {
  it('should call UniverseRoute with correct parameters', async () => {
    const mockRequest = new Request('https://example.com', { method: 'POST' });
    await POST(mockRequest);
    expect(UniverseRoute).toHaveBeenCalledWith(mockRequest, {
      params: Promise.resolve({ provider: 'google' }),
    });
  });
});