| public testOnlyExpectedReturnNodesAreReturned ( array $statements, integer $expectedReturns ) | ||
| $statements | array | |
| $expectedReturns | integer | |
public function testOnlyExpectedReturnNodesAreReturned(array $statements, $expectedReturns)
{
$visitor = new ReturnNodeVisitor();
$traverser = new NodeTraverser();
$traverser->addVisitor($visitor);
$traverser->traverse($statements);
$foundNodes = $visitor->getReturnNodes();
$this->assertCount($expectedReturns, $foundNodes);
$this->assertContainsOnlyInstancesOf(Node\Stmt\Return_::class, $foundNodes);
}