@lobehub/chat/src/server/routers/lambda/__tests__/integration/setup.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 { LobeChatDatabase } from '@/database/type';
import { uuid } from '@/utils/uuid';

/**
 * 创建测试上下文
 */
export const createTestContext = (userId?: string) => ({
  jwtPayload: { userId: userId || uuid() },
  userId: userId || uuid(),
});

/**
 * 创建测试用户
 */
export const createTestUser = async (serverDB: LobeChatDatabase, userId?: string) => {
  const id = userId || uuid();
  const { users } = await import('@/database/schemas');

  await serverDB.insert(users).values({ id });

  return id;
};

/**
 * 清理测试用户及其所有关联数据
 */
export const cleanupTestUser = async (serverDB: LobeChatDatabase, userId: string) => {
  const { users } = await import('@/database/schemas');
  const { eq } = await import('drizzle-orm');

  // 由于外键级联删除，只需删除用户即可
  await serverDB.delete(users).where(eq(users.id, userId));
};