Refactored unit test

Adyen · Nov 29, 2024 · baa4961 · baa4961
1 parent 39e5985
commit baa4961
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 8 deletions.
diff --git a/packages/lib/src/components/Klarna/KlarnaPayments.test.tsx b/packages/lib/src/components/Klarna/KlarnaPayments.test.tsx
@@ -22,18 +22,15 @@ describe('KlarnaPayments', () => {
         expect(screen.queryByRole('button', { name: 'Continue to Pay By Bank' })).toBeFalsy();
     });
 
-    test('should call setStatus if elementRef is a drop-in', async () => {
+    test('should call setStatus if elementRef is a drop-in', () => {
         const KlarnaPaymentsEle = new KlarnaPayments(global.core, {
             ...coreProps,
             ...{ paymentData: '', paymentMethodType: '', sdkData: undefined, useKlarnaWidget: false, showPayButton: false }
         });
         KlarnaPaymentsEle.elementRef = new Dropin(global.core);
         render(KlarnaPaymentsEle.render());
         const spy = jest.spyOn(KlarnaPaymentsEle.elementRef, 'setStatus');
-        // @ts-ignore to test
-        await waitFor(() => KlarnaPaymentsEle.componentRef);
-        // @ts-ignore to test
-        KlarnaPaymentsEle.componentRef.props.onLoaded();
+        KlarnaPaymentsEle.onLoaded();
         expect(spy).toHaveBeenCalled();
     });
 
@@ -55,7 +52,7 @@ describe('KlarnaPayments', () => {
         // @ts-ignore to test
         await waitFor(() => KlarnaPaymentsEle.componentRef);
         // @ts-ignore to test
-        KlarnaPaymentsEle.componentRef.props.onComplete();
+        KlarnaPaymentsEle.componentRef.onComplete();
         expect(onAdditionalDetailsMock).toHaveBeenCalled();
     });
 });
diff --git a/packages/lib/src/components/Klarna/components/KlarnaContainer/KlarnaContainer.tsx b/packages/lib/src/components/Klarna/components/KlarnaContainer/KlarnaContainer.tsx
@@ -19,7 +19,7 @@ export function KlarnaContainer(props) {
 
     klarnaContainerRef.current.setAction = setAction;
     klarnaContainerRef.current.setStatus = setStatus;
-    klarnaContainerRef.current.props = props; // Needed for unit tests
+    klarnaContainerRef.current.onComplete = props.onComplete; // Needed for unit tests
 
     if (action.sdkData) {
         return (

diff --git a/packages/lib/src/components/Klarna/types.ts b/packages/lib/src/components/Klarna/types.ts
@@ -64,5 +64,5 @@ export interface KlarnaAction {
 export interface KlarnaContainerRef extends ComponentMethodsRef {
     initWidget?: () => void;
     setAction?(action: KlarnaAction): void;
-    props?: any;
+    onComplete?: (state: any) => void;
 }