Update Notification V2 controller tests to align with the new TypeORM…

… implementation.
safe-global · Oct 27, 2024 · f53fa31 · f53fa31
1 parent 5eae5f2
commit f53fa31
Showing 1 changed file with 16 additions and 4 deletions.
diff --git a/src/routes/notifications/v2/notifications.controller.spec.ts b/src/routes/notifications/v2/notifications.controller.spec.ts
@@ -177,7 +177,10 @@ describe('Notifications Controller V2 (Unit)', () => {
       expect(
         notificationsRepository.upsertSubscriptions,
       ).toHaveBeenNthCalledWith(1, {
-        signerAddress,
+        authPayload: {
+          signer_address: signerAddress,
+          chain_id: authPayloadDto.chain_id,
+        },
         upsertSubscriptionsDto,
       });
     });
@@ -247,7 +250,10 @@ describe('Notifications Controller V2 (Unit)', () => {
       expect(
         notificationsRepository.upsertSubscriptions,
       ).toHaveBeenNthCalledWith(1, {
-        signerAddress,
+        authPayload: {
+          signer_address: signerAddress,
+          chain_id: authPayloadDto.chain_id,
+        },
         upsertSubscriptionsDto,
       });
     });
@@ -315,7 +321,10 @@ describe('Notifications Controller V2 (Unit)', () => {
       expect(
         notificationsRepository.upsertSubscriptions,
       ).toHaveBeenNthCalledWith(1, {
-        signerAddress,
+        authPayload: {
+          signer_address: signerAddress,
+          chain_id: authPayloadDto.chain_id,
+        },
         upsertSubscriptionsDto,
       });
     });
@@ -696,7 +705,10 @@ describe('Notifications Controller V2 (Unit)', () => {
       expect(
         notificationsRepository.getSafeSubscription,
       ).toHaveBeenNthCalledWith(1, {
-        signerAddress,
+        authPayload: {
+          chain_id: authPayloadDto.chain_id,
+          signer_address: authPayloadDto.signer_address,
+        },
         deviceUuid,
         chainId,
         safeAddress,