mini-starter/tests/utils/server/integration-test-utils.ts

import { IntegrationTestDatabase } from './integration-test-db';
import { UserEntity } from '@/server/modules/users/user.entity';
import { ActivityEntity } from '@/server/modules/activities/activity.entity';
import { RouteEntity } from '@/server/modules/routes/route.entity';



/**
 * 集成测试断言工具
 */
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`);
    }
  }

  /**
   * 断言活动存在于数据库中
   */
  static async expectActivityToExist(activityId: number): Promise<void> {
    const dataSource = await IntegrationTestDatabase.getDataSource();
    if (!dataSource) {
      throw new Error('Database not initialized');
    }

    const activityRepository = dataSource.getRepository(ActivityEntity);
    const activity = await activityRepository.findOne({ where: { id: activityId } });

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

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

    const activityRepository = dataSource.getRepository(ActivityEntity);
    const activity = await activityRepository.findOne({ where: { id: activityId } });

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

  /**
   * 断言路线存在于数据库中
   */
  static async expectRouteToExist(routeId: number): Promise<void> {
    const dataSource = await IntegrationTestDatabase.getDataSource();
    if (!dataSource) {
      throw new Error('Database not initialized');
    }

    const routeRepository = dataSource.getRepository(RouteEntity);
    const route = await routeRepository.findOne({ where: { id: routeId } });

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

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

    const routeRepository = dataSource.getRepository(RouteEntity);
    const route = await routeRepository.findOne({ where: { id: routeId } });

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