Merge pull request #71 from weirdan/fix-build-failures

Fix build failures
psalm · May 4, 2020 · f40da72 · f40da72
2 parents 5ea01f2 + 5590278
commit f40da72
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/src/Hooks/TestCaseHandler.php b/src/Hooks/TestCaseHandler.php
@@ -4,6 +4,7 @@
 
 namespace Psalm\PhpUnitPlugin\Hooks;
 
+use PhpParser\Comment\Doc;
 use PHPUnit\Framework\TestCase;
 use PhpParser\Node\Stmt\ClassLike;
 use PhpParser\Node\Stmt\ClassMethod;
@@ -502,7 +503,10 @@ private static function getSpecials(ClassMethod $method): array
 
         if ($docblock) {
             try {
-                $parsed_comment = DocComment::parse((string)$docblock->getReformattedText(), $docblock->getLine());
+                $parsed_comment = DocComment::parse(
+                    (string)$docblock->getReformattedText(),
+                    self::getCommentLine($docblock)
+                );
             } catch (DocblockParseException $e) {
                 return [];
             }
@@ -530,4 +534,13 @@ private static function queueClassLikeForScanning(
             $codebase->scanner->queueClassLikeForScanning($fq_class_name, $file_path);
         }
     }
+
+    private static function getCommentLine(Doc $docblock): int
+    {
+        if (method_exists($docblock, 'getStartLine')) {
+            return (int) $docblock->getStartLine();
+        }
+        /** @psalm-suppress DeprecatedMethod */
+        return $docblock->getLine();
+    }
 }