improved tests

Adyen · Jan 6, 2025 · 8dd954a · 8dd954a
1 parent 3d1f419
commit 8dd954a
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/packages/lib/src/components/Klarna/KlarnaPayments.test.tsx b/packages/lib/src/components/Klarna/KlarnaPayments.test.tsx
@@ -82,15 +82,15 @@ describe('KlarnaPayments', () => {
                     });
                 }
             });
-            const spy = jest.spyOn(klarna, 'handleAction');
+            const spy = jest.spyOn(klarna, 'updateWithAction');
 
             render(klarna.render());
             klarna.submit();
 
             await waitFor(() => expect(spy).toHaveBeenCalledWith(widgetAction), { interval: 100 });
         });
 
-        test('should NOT use updateWithAction when action type is NOT "sdk"', async () => {
+        test('should NOT use updateWithAction when action type is NOT "sdk"', done => {
             const action: PaymentAction = {
                 paymentMethodType: 'klarna_paynow',
                 type: 'redirect',
@@ -108,12 +108,15 @@ describe('KlarnaPayments', () => {
                     });
                 }
             });
-            const spy = jest.spyOn(klarna, 'handleAction');
+            const spy = jest.spyOn(klarna, 'updateWithAction');
 
             render(klarna.render());
             klarna.submit();
 
-            await waitFor(() => expect(spy).not.toHaveBeenCalled(), { interval: 100 });
+            setTimeout(() => {
+                expect(spy).not.toHaveBeenCalled();
+                done();
+            }, 500);
         });
     });
 });