Fix coverage for nested functions and classes

Author: kelunik

src/CodeCoverage/Report/Node/File.php

@@ -351,6 +351,8 @@ public function getNumTestedFunctions()
      */
     protected function calculateStatistics()
     {
+        $classStack = $functionStack = [];
+
         if ($this->cacheTokens) {
             $tokens = PHP_Token_Stream_CachingFactory::get($this->getPath());
         } else {
@@ -367,6 +369,10 @@ protected function calculateStatistics()
             if (isset($this->startLines[$lineNumber])) {
                 // Start line of a class.
                 if (isset($this->startLines[$lineNumber]['className'])) {
+                    if (isset($currentClass)) {
+                        $classStack[] = &$currentClass;
+                    }
+
                     $currentClass = &$this->startLines[$lineNumber];
                 } // Start line of a trait.
                 elseif (isset($this->startLines[$lineNumber]['traitName'])) {
@@ -376,6 +382,10 @@ protected function calculateStatistics()
                     $currentMethod = &$this->startLines[$lineNumber];
                 } // Start line of a function.
                 elseif (isset($this->startLines[$lineNumber]['functionName'])) {
+                    if (isset($currentFunction)) {
+                        $functionStack[] = &$currentFunction;
+                    }
+
                     $currentFunction = &$this->startLines[$lineNumber];
                 }
             }
@@ -424,6 +434,13 @@ protected function calculateStatistics()
                 // End line of a class.
                 if (isset($this->endLines[$lineNumber]['className'])) {
                     unset($currentClass);
+
+                    if ($classStack) {
+                        end($classStack);
+                        $key = key($classStack);
+                        $currentClass = &$classStack[$key];
+                        unset($classStack[$key]);
+                    }
                 } // End line of a trait.
                 elseif (isset($this->endLines[$lineNumber]['traitName'])) {
                     unset($currentTrait);
@@ -433,6 +450,13 @@ protected function calculateStatistics()
                 } // End line of a function.
                 elseif (isset($this->endLines[$lineNumber]['functionName'])) {
                     unset($currentFunction);
+
+                    if ($functionStack) {
+                        end($functionStack);
+                        $key = key($functionStack);
+                        $currentFunction = &$functionStack[$key];
+                        unset($functionsStack[$key]);
+                    }
                 }
             }
         }