✅ Test refactoring, make test more stable (#374)

- Move copy-pasted code to User API - Clean the database after each test - Refactor documents search test
linagora · Feb 25, 2024 · e0aca9d · e0aca9d
1 parent 488aafe
commit e0aca9d
Show file tree

Hide file tree

Showing 20 changed files with 719 additions and 909 deletions.
diff --git a/...end/node/src/core/platform/services/database/services/orm/connectors/postgres/postgres.ts b/...end/node/src/core/platform/services/database/services/orm/connectors/postgres/postgres.ts
@@ -165,7 +165,18 @@ export class PostgresConnector extends AbstractConnector<PostgresConnectionOptio
   }
 
   async drop(): Promise<this> {
-    logger.info("Drop database is not implemented for security reasons.");
+    const query = `
+        DO $$ 
+        DECLARE 
+          tablename text;
+        BEGIN 
+          FOR tablename IN (SELECT table_name FROM information_schema.tables WHERE table_schema = 'public') 
+          LOOP 
+            EXECUTE 'DELETE FROM  "' || tablename || '" CASCADE'; 
+          END LOOP; 
+        END $$;`;
+    logger.debug(`service.database.orm.postgres.drop - Query: "${query}"`);
+    await this.client.query(query);
     return this;
   }
 

diff --git a/tdrive/backend/node/src/services/documents/services/index.ts b/tdrive/backend/node/src/services/documents/services/index.ts
@@ -1,10 +1,6 @@
 import SearchRepository from "../../../core/platform/services/search/repository";
 import { getLogger, logger, TdriveLogger } from "../../../core/platform/framework";
-import {
-  CrudException,
-  ExecutionContext,
-  ListResult,
-} from "../../../core/platform/framework/api/crud-service";
+import { CrudException, ListResult } from "../../../core/platform/framework/api/crud-service";
 import Repository, {
   comparisonType,
   inType,

diff --git a/tdrive/backend/node/src/services/files/services/index.ts b/tdrive/backend/node/src/services/files/services/index.ts
@@ -311,7 +311,5 @@ export class FileServiceImpl {
 }
 
 function getFilePath(entity: File): string {
-  return `${gr.platformServices.storage.getHomeDir()}/files/${entity.company_id}/${
-    entity.user_id || "anonymous"
-  }/${entity.id}`;
+  return `/files/${entity.company_id}/${entity.user_id || "anonymous"}/${entity.id}`;
 }
diff --git a/tdrive/backend/node/src/services/user/services/users/service.ts b/tdrive/backend/node/src/services/user/services/users/service.ts
@@ -33,10 +33,7 @@ import { getPublicUserRoom, getUserRoom } from "../../realtime";
 import NodeCache from "node-cache";
 import gr from "../../../global-resolver";
 import { formatUser } from "../../../../utils/users";
-import { getDefaultDriveItem } from "../../../documents/utils";
-import { CompanyExecutionContext } from "../../../documents/types";
 import { TYPE as DriveFileType, DriveFile } from "../../../documents/entities/drive-file";
-import config from "config";
 
 export class UserServiceImpl {
   version: "1";