Use ARRAY_CONTAINS to query roles

Jira ticket: CAMS-283
US-Trustee-Program · Sep 19, 2024 · fe6b6da · fe6b6da
1 parent cd0ac3e
commit fe6b6da
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/backend/functions/lib/adapters/gateways/cosmos/cosmos.repository.ts b/backend/functions/lib/adapters/gateways/cosmos/cosmos.repository.ts
@@ -5,6 +5,7 @@ import { CosmosConfig } from '../../types/database';
 import { ServerConfigError } from '../../../common-errors/server-config-error';
 import { isPreExistingDocumentError } from './cosmos.helper';
 import { DocumentRepository } from '../../../use-cases/gateways.types';
+import { UnknownError } from '../../../common-errors/unknown-error';
 
 export class CosmosDbRepository<T> implements DocumentRepository<T> {
   protected cosmosDbClient;
@@ -35,7 +36,7 @@ export class CosmosDbRepository<T> implements DocumentRepository<T> {
           originalError,
         });
       } else {
-        throw originalError;
+        throw new UnknownError(this.moduleName, { originalError });
       }
     }
   }

diff --git a/backend/functions/lib/adapters/gateways/offices.cosmosdb.repository.ts b/backend/functions/lib/adapters/gateways/offices.cosmosdb.repository.ts
@@ -29,14 +29,18 @@ export class OfficesCosmosDbRepository implements OfficesRepository {
     context: ApplicationContext,
     officeCode: string,
   ): Promise<AttorneyUser[]> {
-    const query = `SELECT * FROM c WHERE c.officeCode = @officeCode and c.documentType = 'OFFICE_STAFF' and ${CamsRole.TrialAttorney} IN c.roles`;
+    const query = `SELECT * FROM c WHERE c.officeCode = @officeCode and c.documentType = 'OFFICE_STAFF' and ARRAY_CONTAINS(c.roles, @role)`;
     const querySpec = {
       query,
       parameters: [
         {
           name: '@officeCode',
           value: officeCode,
         },
+        {
+          name: '@role',
+          value: CamsRole.TrialAttorney,
+        },
       ],
     };
     return await this.officeStaffRepo.query(context, querySpec);