check case of async func def

Author: rmorshea

flake8_idom_hooks/rules_of_hooks.py

@@ -49,6 +49,14 @@ def visit_FunctionDef(self, node: ast.FunctionDef) -> None:
             ):
                 self.generic_visit(node)
 
+    def visit_AsyncFunctionDef(self, node: ast.AsyncFunctionDef) -> None:
+        with set_current(
+            self,
+            function=node,
+            early_return=None,
+        ):
+            self.generic_visit(node)
+
     def visit_Call(self, node: ast.Call) -> None:
         with set_current(self, call=node):
             self.visit(node.func)

tests/cases/hook_usage.py

@@ -219,3 +219,8 @@ def regression_check():
     def effect():
         # this return caused false passitive early return error in use_effect usage
         return cleanup
+
+    @use_effect
+    async def effect():
+        # this return caused false passitive early return error in use_effect usage
+        return cleanup