fix ts type errors

huridocs · Jan 16, 2025 · 3c0f7d4 · 3c0f7d4
1 parent 2c62fd9
commit 3c0f7d4
Show file tree

Hide file tree

Showing 9 changed files with 15 additions and 15 deletions.
diff --git a/...-relationships-migration/specs/141-add-collections-for-v2-relationships-migration.spec.ts b/...-relationships-migration/specs/141-add-collections-for-v2-relationships-migration.spec.ts
@@ -55,12 +55,12 @@ describe('migration add collections for v2 relationships migration', () => {
   });
 
   it('should set unique index on the migration fields', async () => {
-    const relCollection = await db.collection('relationshipMigrationFields');
+    const relCollection = db.collection('relationshipMigrationFields');
     const indexInfo = await relCollection.indexInformation({ full: true });
     const uniqueIndex = indexInfo.find(
       (index: any) => index.name === 'sourceTemplate_1_relationType_1_targetTemplate_1'
     );
-    expect(uniqueIndex.unique).toBe(true);
+    expect(uniqueIndex?.unique).toBe(true);
   });
 
   it('should check if a reindex is needed', async () => {

diff --git a/app/api/odm/specs/DB.spec.ts b/app/api/odm/specs/DB.spec.ts
@@ -27,8 +27,8 @@ describe('DB', () => {
   beforeEach(async () => {
     const uri = config.DBHOST;
     await DB.connect(`${uri}_DB_spec_ts`);
-    db1 = DB.getConnection().useDb('db1').db;
-    db2 = DB.getConnection().useDb('db2').db;
+    db1 = DB.getConnection().getClient().db('db1');
+    db2 = DB.getConnection().getClient().db('db2');
   });
 
   afterAll(async () => {

diff --git a/app/api/odm/specs/model_multi_tenant.spec.ts b/app/api/odm/specs/model_multi_tenant.spec.ts
@@ -23,9 +23,9 @@ describe('ODM Model multi-tenant', () => {
 
   beforeAll(async () => {
     await testingDB.connect({ defaultTenant: false });
-    defaultDB = DB.connectionForDB(config.defaultTenant.dbName).db;
-    db1 = DB.connectionForDB('db1').db;
-    db2 = DB.connectionForDB('db2').db;
+    defaultDB = DB.getConnection().getClient().db(config.defaultTenant.dbName);
+    db1 = DB.getConnection().getClient().db('db1');
+    db2 = DB.getConnection().getClient().db('db2');
   });
 
   beforeEach(async () => {

diff --git a/app/api/services/pdfsegmentation/specs/PDFSegmentation.spec.ts b/app/api/services/pdfsegmentation/specs/PDFSegmentation.spec.ts
@@ -79,8 +79,8 @@ describe('PDFSegmentation', () => {
 
   beforeEach(async () => {
     segmentPdfs = new PDFSegmentation();
-    dbOne = DB.connectionForDB(tenantOne.dbName).db;
-    dbTwo = DB.connectionForDB(tenantTwo.dbName).db;
+    dbOne = DB.getConnection().getClient().db(tenantOne.dbName);
+    dbTwo = DB.getConnection().getClient().db(tenantTwo.dbName);
 
     tenants.tenants = { tenantOne };
     fileA = await fs.readFile(`app/api/services/pdfsegmentation/specs/uploads/${fixturesPdfNameA}`);

diff --git a/app/api/stats/routes.ts b/app/api/stats/routes.ts
@@ -6,7 +6,7 @@ import { DB } from 'api/odm';
 
 export default (app: Application) => {
   app.get('/api/stats', needsAuthorization(['admin']), async (_req, res, _next) => {
-    const { db } = DB.connectionForDB(tenants.current().dbName);
+    const db = DB.getConnection().getClient().db(tenants.current().dbName);
     const action = new RetrieveStatsService(db);
     const stats = await action.execute();
 

diff --git a/app/api/stats/specs/RetrieveStatsService.spec.ts b/app/api/stats/specs/RetrieveStatsService.spec.ts
@@ -10,7 +10,7 @@ describe('RetrieveStats', () => {
   let db: Db;
 
   beforeAll(async () => {
-    db = (await testingDB.connect()).db;
+    db = (await testingDB.connect()).db as Db;
   });
 
   beforeEach(async () => {

diff --git a/app/api/tenants/specs/tenantsContext.spec.ts b/app/api/tenants/specs/tenantsContext.spec.ts
@@ -26,7 +26,7 @@ describe('tenantsContext', () => {
     beforeAll(async () => {
       await testingDB.connect();
       testingEnvironment.setRequestId();
-      db = DB.connectionForDB(config.SHARED_DB).db;
+      db = DB.getConnection().getClient().db(config.SHARED_DB);
 
       await db.collection('tenants').deleteMany({});
       await db.collection('tenants').insertMany([

diff --git a/app/api/tenants/specs/tenantsModel.spec.ts b/app/api/tenants/specs/tenantsModel.spec.ts
@@ -17,7 +17,7 @@ describe('tenantsModel', () => {
   beforeAll(async () => {
     await testingDB.connect();
     testingEnvironment.setRequestId();
-    db = DB.connectionForDB(config.SHARED_DB).db;
+    db = DB.connectionForDB(config.SHARED_DB).getClient().db(config.SHARED_DB);
   });
 
   beforeEach(async () => {

diff --git a/app/api/utils/testing_db.ts b/app/api/utils/testing_db.ts
@@ -108,7 +108,7 @@ const testingDB: {
         .basename(expect.getState().testPath || '')
         .replace(/[.-]/g, '_')}`.substring(0, 63);
       await initMongoServer(this.dbName);
-      mongodb = mongooseConnection.db;
+      mongodb = mongooseConnection.getClient().db();
       this.mongodb = mongodb;
 
       if (options.defaultTenant) {
@@ -148,7 +148,7 @@ const testingDB: {
     await this.connect();
     let optionalMongo: Db | null = null;
     if (dbName) {
-      optionalMongo = DB.connectionForDB(dbName).db;
+      optionalMongo = DB.connectionForDB(dbName).getClient().db(dbName);
     }
     await fixturer.clearAllAndLoad(optionalMongo || mongodb, fixtures);
     await this.createIndices();