file-starter/src/server/__test_utils__/integration-test-utils.ts

import { OpenAPIHono } from '@hono/zod-openapi';
import { testClient } from 'hono/testing';
import { IntegrationTestDatabase } from './integration-test-db';
import { DataSource } from 'typeorm';
import { UserEntity } from '../modules/users/user.entity';
import { Role } from '../modules/users/role.entity';
import apiApp from '../api';

// /**
//  * 集成测试配置选项
//  */
// export interface IntegrationTestOptions {
//   setupDatabase?: boolean;
//   setupAuth?: boolean;
//   setupMiddlewares?: boolean;
// }

// /**
//  * 集成测试上下文
//  */
// export interface IntegrationTestContext {
//   app: OpenAPIHono;
//   client: ReturnType<typeof testClient<typeof apiApp>>;
//   dataSource: DataSource | null;
// }

// // /**
// //  * 创建集成测试应用实例
// //  */
// // export async function createIntegrationTestApp(
// //   routes: any[],
// //   options: IntegrationTestOptions = {}
// // ): Promise<OpenAPIHono> {
// //   // 使用主API应用，确保所有路由已注册
// //   return apiApp;
// // }

// /**
//  * 创建集成测试客户端（使用hono/testing的testClient）
//  */
// export function createIntegrationTestClient(
//   app: OpenAPIHono,
//   options: IntegrationTestOptions = {}
// ): ReturnType<typeof testClient<typeof apiApp>> {
//   const client = testClient(app);

//   // 设置默认认证头（如果需要）
//   if (options.setupAuth !== false) {
//     // testClient会自动处理header，这里不需要额外设置
//   }

//   return client;
// }

// /**
//  * 设置集成测试环境
//  */
// export async function setupIntegrationTestEnvironment(
//   routes: any[],
//   options: IntegrationTestOptions = {}
// ): Promise<IntegrationTestContext> {
//   const {
//     setupDatabase = true,
//     setupAuth = true,
//     setupMiddlewares = true
//   } = options;

//   // 创建测试应用
//   // const app = await createIntegrationTestApp(routes, options);

//   // 初始化数据库（如果需要）
//   let dataSource: DataSource | null = null;
//   if (setupDatabase) {
//     dataSource = await IntegrationTestDatabase.initialize();
//   }

//   // 创建API客户端
//   const client = createIntegrationTestClient(app, { setupAuth });

//   return {
//     app,
//     client,
//     dataSource
//   };
// }

// /**
//  * 清理集成测试环境
//  */
// export async function cleanupIntegrationTestEnvironment(): Promise<void> {
//   await IntegrationTestDatabase.clearAllData();
//   await IntegrationTestDatabase.cleanup();
// }

// /**
//  * 测试数据工厂函数
//  * 使用integration-test-db.ts中的TestDataFactory
//  */
// export { TestDataFactory } from './integration-test-db';

/**
 * 集成测试断言工具
 */
export class IntegrationTestAssertions {
  /**
   * 断言响应状态码
   */
  static expectStatus(response: { status: number }, expectedStatus: number): void {
    if (response.status !== expectedStatus) {
      throw new Error(`Expected status ${expectedStatus}, but got ${response.status}`);
    }
  }

  /**
   * 断言响应包含特定字段
   */
  static expectResponseToHave(response: { data: any }, expectedFields: Record<string, any>): void {
    for (const [key, value] of Object.entries(expectedFields)) {
      if (response.data[key] !== value) {
        throw new Error(`Expected field ${key} to be ${value}, but got ${response.data[key]}`);
      }
    }
  }

  /**
   * 断言响应包含特定结构
   */
  static expectResponseStructure(response: { data: any }, structure: Record<string, any>): void {
    for (const key of Object.keys(structure)) {
      if (!(key in response.data)) {
        throw new Error(`Expected response to have key: ${key}`);
      }
    }
  }

  /**
   * 断言用户存在于数据库中
   */
  static async expectUserToExist(username: string): Promise<void> {
    const dataSource = await IntegrationTestDatabase.getDataSource();
    if (!dataSource) {
      throw new Error('Database not initialized');
    }

    const userRepository = dataSource.getRepository(UserEntity);
    const user = await userRepository.findOne({ where: { username } });

    if (!user) {
      throw new Error(`Expected user ${username} to exist in database`);
    }
  }

  /**
   * 断言用户不存在于数据库中
   */
  static async expectUserNotToExist(username: string): Promise<void> {
    const dataSource = await IntegrationTestDatabase.getDataSource();
    if (!dataSource) {
      throw new Error('Database not initialized');
    }

    const userRepository = dataSource.getRepository(UserEntity);
    const user = await userRepository.findOne({ where: { username } });

    if (user) {
      throw new Error(`Expected user ${username} not to exist in database`);
    }
  }
}

// /**
//  * 集成测试生命周期钩子
//  */
// export function setupIntegrationTestHooks() {
//   beforeEach(async () => {
//     await IntegrationTestDatabase.initialize();
//   });

//   afterEach(async () => {
//     await IntegrationTestDatabase.clearAllData();
//   });

//   afterAll(async () => {
//     await IntegrationTestDatabase.cleanup();
//   });
// }