Преглед на файлове

♻️ refactor(test): optimize integration test database initialization logic

- 调整数据源初始化逻辑,避免重复调用initializeDataSource
- 仅在AppDataSource未初始化时调用initialize()方法,提高测试效率
yourname преди 1 месец
родител
ревизия
cea9c64035
променени са 1 файла, в които са добавени 5 реда и са изтрити 1 реда
  1. 5 1
      packages/user-module/tests/utils/integration-test-db.ts

+ 5 - 1
packages/user-module/tests/utils/integration-test-db.ts

@@ -21,8 +21,12 @@ export class IntegrationTestDatabase {
    * 获取当前数据源
    * 获取当前数据源
    */
    */
   static async getDataSource(): Promise<DataSource> {
   static async getDataSource(): Promise<DataSource> {
+    if (!AppDataSource) {
+      initializeDataSource([UserEntity, Role]);
+    }
+
     if (!AppDataSource.isInitialized) {
     if (!AppDataSource.isInitialized) {
-      await initializeDataSource([UserEntity, Role]);
+      await AppDataSource.initialize();
     }
     }
     return AppDataSource;
     return AppDataSource;
   }
   }