check case of async func def

reactive-python · Oct 7, 2022 · 833cd52 · 833cd52 · Archmonger · Oct 7, 2022
1 parent 6c5abe7
commit 833cd52
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/flake8_idom_hooks/rules_of_hooks.py b/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)

diff --git a/tests/cases/hook_usage.py b/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