Merge pull request #965 from afflux/fix-964-asyncdef-line

handle decorators for AsyncFunctionDefs. Closes #964
nedbat · Mar 26, 2020 · aaa9f42 · aaa9f42
2 parents f318f93 + 0db15a2
commit aaa9f42
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/coverage/parser.py b/coverage/parser.py
@@ -611,6 +611,8 @@ def _line__Dict(self, node):
 
     _line__FunctionDef = _line_decorated
 
+    _line__AsyncFunctionDef = _line_decorated
+
     def _line__List(self, node):
         if node.elts:
             return self.line_for_node(node.elts[0])

diff --git a/tests/test_arcs.py b/tests/test_arcs.py
@@ -1557,6 +1557,23 @@ async def go():
             arcz_missing=".2 23 3.",
         )
 
+    def test_async_decorator(self):
+        if env.PYBEHAVIOR.trace_decorated_def:
+            arcz = ".1 14 45 5.  .2 2.  -46 6-4"
+        else:
+            arcz = ".1 14 4.     .2 2.  -46 6-4"
+        self.check_coverage("""\
+            def wrap(f):        # 1
+                return f
+
+            @wrap               # 4
+            async def go():
+                return
+            """,
+            arcz=arcz,
+            arcz_missing='-46 6-4',
+        )
+
 
 class ExcludeTest(CoverageTest):
     """Tests of exclusions to indicate known partial branches."""