Capture comments in Variable assignments #8

Author: oscarotero

src/PhpNodeVisitor.php

@@ -45,6 +45,7 @@ public function enterNode(Node $node)
             case 'Stmt_Echo':
             case 'Stmt_Return':
             case 'Expr_Print':
+            case 'Expr_Assign':
                 $this->bufferComments = $node;
                 return null;
         }

tests/PhpFunctionsScannerTest.php

@@ -164,7 +164,9 @@ public function testPhpFunctionsExtractor()
         $this->assertSame(30, $function->getLine());
         $this->assertSame(30, $function->getLastLine());
         $this->assertSame($file, $function->getFilename());
-        $this->assertCount(0, $function->getComments());
+        $this->assertCount(1, $function->getComments());
+        $comments = $function->getComments();
+        $this->assertSame('Related comment 5', array_shift($comments));
     }
 
     public function _testPhpFunctionsScannerWithDisabledComments()

tests/assets/functions.php

@@ -26,5 +26,5 @@
     'arg12'
     /* No Related comment 4 */
 );
-
-fn13("Translatable string","",["context"=>"Context string", 'foo']);
+//Related comment 5
+$bar = fn13("Translatable string","",["context"=>"Context string", 'foo']);