Nest.js E2E测试中连接CosmosDB失败问题求助

可能的原因及对应解决方案 #

1. 确认测试环境变量的加载 #

虽然你提到环境变量在测试外正常，但E2E测试可能未正确读取测试专用配置文件。默认ConfigModule.forRoot()加载根目录.env，而测试环境通常需要.env.test，可明确指定：

ConfigModule.forRoot({
  validationSchema: schema,
  isGlobal: true,
  envFilePath: '.env.test', // 指定测试环境的env文件
}),

同时确保.env.test中AZURE_COSMOS_DB_TEST_NAME对应的数据库存在，AZURE_COSMOS_DB_ENDPOINT和AZURE_COSMOS_DB_KEY与测试环境完全匹配。

2. 检查CosmosDB的网络与权限配置 #

如果使用Azure云端CosmosDB，需确认：

• 测试机器IP已加入CosmosDB账户的防火墙白名单
• 所用密钥拥有测试数据库的读写权限（建议先用主键测试）
• 本地测试可临时关闭CosmosDB防火墙限制（仅测试环境）

3. 调整模块初始化的等待逻辑 #

E2E测试可能在CosmosDB连接建立前就启动，导致连接失败。可手动等待客户端初始化完成：

beforeAll(async () => {
  const moduleFixture: TestingModule = await Test.createTestingModule({
    imports: [
      ConfigModule.forRoot({
        validationSchema: schema,
        isGlobal: true,
        envFilePath: '.env.test',
      }),
      AzureCosmosDbModule.forRootAsync({
        imports: [ConfigModule],
        useFactory: async (cfg: ConfigService) => ({
          endpoint: cfg.get<string>('AZURE_COSMOS_DB_ENDPOINT'),
          dbName: cfg.get<string>('AZURE_COSMOS_DB_TEST_NAME'),
          key: cfg.get<string>('AZURE_COSMOS_DB_KEY'),
          options: {
            connectionPolicy: {
              retryOptions: {
                maxRetryAttemptsOnThrottledRequests: 5,
                maxRetryWaitTimeInSeconds: 10,
              },
            },
          },
        }),
        inject: [ConfigService],
      }),
      CatsModule,
    ],
  }).compile();

  app = moduleFixture.createNestApplication();
  app.setGlobalPrefix('api');

  // 手动获取Cosmos客户端并确认数据库存在
  const cosmosClient = moduleFixture.get(CosmosClient);
  const cfg = moduleFixture.get(ConfigService);
  await cosmosClient.databases.createIfNotExists({ id: cfg.get<string>('AZURE_COSMOS_DB_TEST_NAME') });

  await app.init();
});

4. 改用模拟（Mock）CosmosDB服务（推荐） #

如果E2E测试核心是验证API逻辑而非数据库连接，可模拟AzureCosmosDbModule提供的服务，避免依赖真实数据库：

beforeAll(async () => {
  const moduleFixture: TestingModule = await Test.createTestingModule({
    imports: [
      ConfigModule.forRoot({
        validationSchema: schema,
        isGlobal: true,
      }),
      CatsModule,
    ],
    providers: [
      // 模拟实体对应的Repository
      {
        provide: getRepositoryToken(Cat), // 替换为你的实体类
        useValue: {
          find: jest.fn().mockResolvedValue([]),
          create: jest.fn().mockResolvedValue({ id: 'test-id', name: 'test-cat' }),
          // 根据测试需求添加更多模拟方法
        },
      },
    ],
  }).compile();

  app = moduleFixture.createNestApplication();
  app.setGlobalPrefix('api');
  await app.init();
});

内容的提问来源于stack exchange，提问作者sushiPlague