public function testSetAndGetAndAddUse() { $code = new PhpClass(); $code->setUseCollection(new UseCollection(array(new UseClass(array('className' => 'ClassGeneration\\PhpClass'))))); $this->assertInstanceOf('\\ClassGeneration\\UseCollection', $code->getUseCollection()); $this->assertCount(1, $code->getUseCollection()); $this->assertEquals('ClassGeneration\\PhpClass', $code->getUseCollection()->current()->getClassName()); $code->addUse(new UseClass(array('className' => 'ClassGeneration\\PropertyCollection', 'alias' => 'Properties'))); $this->assertCount(2, $code->getUseCollection()); $this->assertEquals('use ClassGeneration\\PropertyCollection as Properties;' . PHP_EOL, $code->getUseCollection()->last()->toString()); }