Chris@13: <?php declare(strict_types=1);
Chris@0: 
Chris@0: namespace PhpParser\Builder;
Chris@0: 
Chris@0: use PhpParser\Comment;
Chris@0: use PhpParser\Node\Name;
Chris@0: use PhpParser\Node\Stmt;
Chris@0: 
Chris@17: class TraitTest extends \PHPUnit\Framework\TestCase
Chris@0: {
Chris@0:     protected function createTraitBuilder($class) {
Chris@0:         return new Trait_($class);
Chris@0:     }
Chris@0: 
Chris@0:     public function testStmtAddition() {
Chris@0:         $method1 = new Stmt\ClassMethod('test1');
Chris@0:         $method2 = new Stmt\ClassMethod('test2');
Chris@0:         $method3 = new Stmt\ClassMethod('test3');
Chris@13:         $prop = new Stmt\Property(Stmt\Class_::MODIFIER_PUBLIC, [
Chris@0:             new Stmt\PropertyProperty('test')
Chris@13:         ]);
Chris@0:         $use = new Stmt\TraitUse([new Name('OtherTrait')]);
Chris@0:         $trait = $this->createTraitBuilder('TestTrait')
Chris@0:             ->setDocComment('/** Nice trait */')
Chris@0:             ->addStmt($method1)
Chris@0:             ->addStmts([$method2, $method3])
Chris@0:             ->addStmt($prop)
Chris@0:             ->addStmt($use)
Chris@0:             ->getNode();
Chris@0:         $this->assertEquals(new Stmt\Trait_('TestTrait', [
Chris@0:             'stmts' => [$use, $prop, $method1, $method2, $method3]
Chris@0:         ], [
Chris@0:             'comments' => [
Chris@0:                 new Comment\Doc('/** Nice trait */')
Chris@0:             ]
Chris@0:         ]), $trait);
Chris@0:     }
Chris@0: 
Chris@0:     public function testInvalidStmtError() {
Chris@17:         $this->expectException(\LogicException::class);
Chris@17:         $this->expectExceptionMessage('Unexpected node of type "Stmt_Echo"');
Chris@0:         $this->createTraitBuilder('Test')
Chris@13:             ->addStmt(new Stmt\Echo_([]))
Chris@0:         ;
Chris@0:     }
Chris@0: }