From d386f9d533b80e87c3d799c8e19f89a01b6b8367 Mon Sep 17 00:00:00 2001
From: Wojciech Maj <kontakt@wojtekmaj.pl>
Date: Mon, 12 Jun 2023 20:59:26 +0200
Subject: [PATCH] Add more tests

---
 src/AsyncButton.spec.tsx | 52 ++++++++++++++++++++++++++++++++++++----
 1 file changed, 48 insertions(+), 4 deletions(-)

diff --git a/src/AsyncButton.spec.tsx b/src/AsyncButton.spec.tsx
index b056347..165a581 100644
--- a/src/AsyncButton.spec.tsx
+++ b/src/AsyncButton.spec.tsx
@@ -262,7 +262,7 @@ describe('<AsyncButton /> component', () => {
     <MyCustomComponent as={5} />;
   });
 
-  it('should allow HTMLButtonElement event handlers to be passed by default', () => {
+  it('should allow sync HTMLButtonElement event handlers to be passed by default', () => {
     function onClick(event: React.MouseEvent<HTMLButtonElement>) {
       event.preventDefault();
     }
@@ -271,6 +271,15 @@ describe('<AsyncButton /> component', () => {
     <AsyncButton {...defaultProps} onClick={onClick} />;
   });
 
+  it('should allow async HTMLButtonElement event handlers to be passed by default', () => {
+    async function onClick(event: React.MouseEvent<HTMLButtonElement>) {
+      event.preventDefault();
+    }
+
+    // @ts-expect-no-error
+    <AsyncButton {...defaultProps} onClick={onClick} />;
+  });
+
   it('should not allow HTMLAnchorElement event handlers to be passed by default', () => {
     function onClick(event: React.MouseEvent<HTMLAnchorElement>) {
       event.preventDefault();
@@ -280,7 +289,7 @@ describe('<AsyncButton /> component', () => {
     <AsyncButton {...defaultProps} onClick={onClick} />;
   });
 
-  it('should allow HTMLButtonElement event handlers to be passed given as="button"', () => {
+  it('should allow sync HTMLButtonElement event handlers to be passed given as="button"', () => {
     function onClick(event: React.MouseEvent<HTMLButtonElement>) {
       event.preventDefault();
     }
@@ -289,6 +298,15 @@ describe('<AsyncButton /> component', () => {
     <AsyncButton {...defaultProps} as="button" onClick={onClick} />;
   });
 
+  it('should allow async HTMLButtonElement event handlers to be passed given as="button"', () => {
+    async function onClick(event: React.MouseEvent<HTMLButtonElement>) {
+      event.preventDefault();
+    }
+
+    // @ts-expect-no-error
+    <AsyncButton {...defaultProps} as="button" onClick={onClick} />;
+  });
+
   it('should not allow HTMLAnchorElement event handlers to be passed given as="button"', () => {
     function onClick(event: React.MouseEvent<HTMLAnchorElement>) {
       event.preventDefault();
@@ -302,7 +320,7 @@ describe('<AsyncButton /> component', () => {
     <button onClick={onClick}></button>;
   });
 
-  it('should allow HTMLAnchorElement event handlers to be passed given as="a"', () => {
+  it('should allow sync HTMLAnchorElement event handlers to be passed given as="a"', () => {
     function onClick(event: React.MouseEvent<HTMLAnchorElement>) {
       event.preventDefault();
     }
@@ -311,6 +329,15 @@ describe('<AsyncButton /> component', () => {
     <AsyncButton {...defaultProps} as="a" onClick={onClick} />;
   });
 
+  it('should allow async HTMLAnchorElement event handlers to be passed given as="a"', () => {
+    async function onClick(event: React.MouseEvent<HTMLAnchorElement>) {
+      event.preventDefault();
+    }
+
+    // @ts-expect-no-error
+    <AsyncButton {...defaultProps} as="a" onClick={onClick} />;
+  });
+
   it('should not allow HTMLButtonElement event handlers to be passed given as="a"', () => {
     function onClick(event: React.MouseEvent<HTMLButtonElement>) {
       event.preventDefault();
@@ -326,7 +353,7 @@ describe('<AsyncButton /> component', () => {
     </a>;
   });
 
-  it('should allow HTMLButtonElement event handlers to be passed given as={MyButton} that handles onClick', () => {
+  it('should allow sync HTMLButtonElement event handlers to be passed given as={MyButton} that handles onClick', () => {
     function MyButton({
       onClick,
     }: {
@@ -343,6 +370,23 @@ describe('<AsyncButton /> component', () => {
     <AsyncButton {...defaultProps} as={MyButton} onClick={onClick} />;
   });
 
+  it('should allow async HTMLButtonElement event handlers to be passed given as={MyButton} that handles onClick', () => {
+    function MyButton({
+      onClick,
+    }: {
+      onClick: (event: React.MouseEvent<HTMLButtonElement>) => void;
+    }) {
+      return <button onClick={onClick} type="submit"></button>;
+    }
+
+    async function onClick(event: React.MouseEvent<HTMLButtonElement>) {
+      event.preventDefault();
+    }
+
+    // @ts-expect-no-error
+    <AsyncButton {...defaultProps} as={MyButton} onClick={onClick} />;
+  });
+
   it('should allow HTMLAnchorElement event handlers to be passed given as={MyAnchor} that handles onClick', () => {
     function MyAnchor({
       onClick,
-- 
2.54.0