WIP: fix

Author: blueyed

src/_pytest/_code/code.py

@@ -244,12 +244,14 @@ def getsource(self, astcache=None) -> Optional["Source"]:
                 astnode = astcache.get(key, None)
         start = self.getfirstlinesource()
         try:
-            astnode, _, end = getstatementrange_ast(
+            astnode, ast_start, end = getstatementrange_ast(
                 self.lineno, source, astnode=astnode
             )
         except SyntaxError:
             end = self.lineno + 1
         else:
+            if ast_start - 1 < start:
+                start = ast_start
             if key is not None:
                 astcache[key] = astnode
         return source[start:end]

src/_pytest/_code/source.py

@@ -357,6 +357,8 @@ def get_statement_startend2(lineno: int, node: ast.AST) -> Tuple[int, Optional[i
     # AST's line numbers start indexing at 1
     values = []  # type: List[int]
     for x in ast.walk(node):
+        if isinstance(x, (ast.FunctionDef)):
+            values.extend(deco.lineno - 1 for deco in x.decorator_list)
         if isinstance(x, (ast.stmt, ast.ExceptHandler)):
             values.append(x.lineno - 1)
             for name in ("finalbody", "orelse"):