mini-starter/packages/unified-file-module/tests/utils/integration-test-db.ts

import { DataSource } from "typeorm";
import { UnifiedFile } from "../../src/entities";

/**
 * 测试数据工厂类
 */
export class TestDataFactory {
  /**
   * 创建测试文件数据
   */
  static createFileData(overrides: Partial<UnifiedFile> = {}): Partial<UnifiedFile> {
    const timestamp = Date.now();
    return {
      fileName: `testfile_${timestamp}.txt`,
      mimeType: "text/plain",
      fileSize: 1024,
      filePath: `/uploads/testfile_${timestamp}.txt`,
      description: `Test file ${timestamp}`,
      createdAt: new Date(),
      ...overrides
    };
  }

  /**
   * 在数据库中创建测试文件
   */
  static async createTestFile(dataSource: DataSource, overrides: Partial<UnifiedFile> = {}): Promise<UnifiedFile> {
    const fileData = this.createFileData(overrides);
    const fileRepository = dataSource.getRepository(UnifiedFile);

    const file = fileRepository.create(fileData);
    return await fileRepository.save(file);
  }
}