apps/api/src/federation/utils/retry.spec.ts

@@ -160,21 +160,25 @@ describe("Retry Utility", () => {
       expect(operation).toHaveBeenCalledTimes(4);
     });
 
-    it("should verify exponential backoff timing", () => {
+    it("should verify exponential backoff timing", async () => {
       const operation = vi.fn().mockRejectedValue({
         code: "ECONNREFUSED",
         message: "Connection refused",
         name: "Error",
       });
 
-      // Just verify the function is called multiple times with retries
+      // Verify the function attempts multiple retries and eventually throws
-      const promise = withRetry(operation, {
+      await expect(
-        maxRetries: 2,
+        withRetry(operation, {
-        initialDelay: 10,
+          maxRetries: 2,
+          initialDelay: 10,
+        })
+      ).rejects.toMatchObject({
+        message: "Connection refused",
       });
 
-      // We don't await this - just verify the retry configuration exists
+      // Should be called 3 times (initial + 2 retries)
-      expect(promise).toBeInstanceOf(Promise);
+      expect(operation).toHaveBeenCalledTimes(3);
     });
   });
 });