/** * @test */ function it_adds_a_trait_with_alias() { $assembler = new TraitAssembler('\\Namespace\\MyTrait', 'TraitAlias'); $context = $this->createContext(); $assembler->assemble($context); $code = $context->getClass()->generate(); $expected = <<<CODE namespace MyNamespace; use Namespace\\MyTrait as TraitAlias; class MyType { use TraitAlias; } CODE; $this->assertEquals($expected, $code); }