Handle errors from Cosmos better

Co-authored-by: Fritz Madden <[email protected]> Co-authored-by: Arthur Morrow <[email protected]>
US-Trustee-Program · Oct 16, 2024 · ee97959 · ee97959
1 parent a6d7077
commit ee97959
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 7 deletions.
diff --git a/backend/functions/lib/adapters/gateways/user-session-cache.cosmosdb.repository.ts b/backend/functions/lib/adapters/gateways/user-session-cache.cosmosdb.repository.ts
@@ -5,6 +5,8 @@ import { CamsSession } from '../../../../../common/src/cams/session';
 import { UserSessionCacheRepository } from './user-session-cache.repository';
 import { UnauthorizedError } from '../../common-errors/unauthorized-error';
 import { CamsJwtClaims } from '../../../../../common/src/cams/jwt';
+import { getCamsError } from '../../common-errors/error-utilities';
+import { isConflictError } from '../../use-cases/user-session/user-session';
 
 const MODULE_NAME: string = 'COSMOS_DB_REPOSITORY_USER_SESSION_CACHE';
 const CONTAINER_NAME: string = 'user-session-cache';
@@ -38,7 +40,14 @@ export class UserSessionCacheCosmosDbRepository implements UserSessionCacheRepos
       ],
     };
 
-    const cached = await this.repo.query(context, querySpec);
+    let cached;
+    try {
+      cached = await this.repo.query(context, querySpec);
+    } catch (error) {
+      if (isConflictError(error)) throw error;
+      throw getCamsError(error, MODULE_NAME);
+    }
+
     if (cached && cached.length === 1) {
       return toCamsSession(cached[0]);
     } else {

diff --git a/backend/functions/lib/use-cases/user-session/user-session.ts b/backend/functions/lib/use-cases/user-session/user-session.ts
@@ -85,16 +85,19 @@ export class UserSessionUseCase {
       await sessionCacheRepository.put(context, session);
 
       return session;
-    } catch (originalError) {
-      if (isConflictError(originalError)) {
+    } catch (error) {
+      const isConflict = error.originalError
+        ? isConflictError(error.originalError)
+        : isConflictError(error);
+      if (isConflict) {
         return await sessionCacheRepository.get(context, token);
       }
 
-      throw isCamsError(originalError)
-        ? originalError
+      throw isCamsError(error)
+        ? error
         : new UnauthorizedError(MODULE_NAME, {
-            message: originalError.message,
-            originalError,
+            message: error.message,
+            originalError: error,
           });
     }
   }