Added return

Author: Slamdunk

src/StaticAnalysis/ExecutableLinesFindingVisitor.php

@@ -44,7 +44,7 @@ public function enterNode(Node $node): void
         if ($node instanceof Node\Stmt\Function_ ||
             $node instanceof Node\Stmt\ClassMethod
         ) {
-            $this->setLineBranch($node->getStartLine(), $node->getEndLine(), ++$this->nextBranch);
+            $this->setLineBranch($node->getStartLine(), $node->getEndLine() - 1, ++$this->nextBranch);
 
             return;
         }
@@ -138,8 +138,30 @@ public function enterNode(Node $node): void
             return;
         }
 
+        if ($node instanceof Node\Stmt\Return_) {
+            $returnBranch  = $this->executableLinesGroupedByBranch[$node->getStartLine()];
+            $returnEndLine = $node->getEndLine();
+            $nextBranch    = null;
+
+            foreach ($this->executableLinesGroupedByBranch as $line => $branch) {
+                if ($line <= $returnEndLine || $branch !== $returnBranch) {
+                    continue;
+                }
+
+                if (null === $nextBranch) {
+                    $nextBranch = ++$this->nextBranch;
+                }
+
+                $this->executableLinesGroupedByBranch[$line] = $nextBranch;
+            }
+
+            return;
+        }
+
         if ($node instanceof Node\Stmt\Declare_) {
             $this->unsets[] = range($node->getStartLine(), $node->getEndLine());
+
+            return;
         }
 
         if ($node instanceof Node\Identifier) {

tests/_files/source_for_branched_exec_lines.php

@@ -13,7 +13,7 @@
 function simple()   // +1
 {                   // 0
     return 1;       // 0
-}                   // 0
+}
 
 $var2 = 1;          // -1
 
@@ -31,7 +31,7 @@ function withIf()   // +3
         $var += 2;  // +1
     }               // -1
     return $var;    // 0
-}                   // 0
+}
 
 class MyClass
 {
@@ -48,7 +48,7 @@ function            // 0
         if (false) {    // 0
             $var += 2;  // +1
         }               // -1
-    }                   // 0
+    }
     public function withForeach()           // +2
     {                                       // 0
         $var = 1;                           // 0
@@ -73,7 +73,7 @@ public function withForeach()           // +2
         {                                   // 0
             $var += 2;                      // +4
         }                                   // -4
-    }                                       // 0
+    }
     public function withWhile()             // +5
     {                                       // 0
         $var = 1;                           // 0
@@ -96,7 +96,7 @@ public function withWhile()             // +5
         {                                   // 0
             ++$var;                         // +4
         }                                   // -4
-    }                                       // 0
+    }
     public function withIfElseifElse()      // +5
     {                                       // 0
         $var = 1;                           // 0
@@ -140,7 +140,7 @@ public function withIfElseifElse()      // +5
         {                                   // 0
             ++$var;                         // +12
         }                                   // -12
-    }                                       // 0
+    }
     public function withFor()               // +13
     {                                       // 0
         $var = 1;                           // 0
@@ -163,7 +163,7 @@ public function withFor()               // +13
         {                                   // 0
             $var += 2;                      // +4
         }                                   // -4
-    }                                       // 0
+    }
     public function withDoWhile()           // +5
     {                                       // 0
         $var = 1;                           // 0
@@ -188,7 +188,7 @@ public function withDoWhile()           // +5
             $var                            // 0
         )                                   // 0
         ;                                   // 0
-    }                                       // 0
+    }
     public function withSwitch()            // +1
     {                                       // 0
         $var = 1;                           // 0
@@ -218,7 +218,7 @@ public function withSwitch()            // +1
             default:                        // -7
                 ++$var;                     // +8
         endswitch;                          // -8
-    }                                       // 0
+    }
     public function withMatch()             // +9
     {                                       // 0
         $var = 1;                           // 0
@@ -248,5 +248,21 @@ public function withMatch()             // +9
             ,                               // -6
         }                                   // 0
         ;                                   // 0
-    }                                       // 0
+    }
+    public function withReturn()            // +7
+    {                                       // 0
+        $var = 1;                           // 0
+        if (false) {                        // 0
+            ++$var;                         // +1
+            return                          // 0
+                $var                        // 0
+            ;                               // 0
+            ++$var;                         // +1
+            if (false) {                    // 0
+                ++$var;                     // +1
+            }                               // -1
+        }                                   // -2
+        return;                             // 0
+        ++$var;                             // +4
+    }
 }