public function testSetAndGetAndAddProperties() { $code = new PhpClass(); $code->setPropertyCollection(new PropertyCollection(array(new Property(array('name' => 'test1', 'value' => 1))))); $this->assertInstanceOf('\\ClassGeneration\\PropertyCollection', $code->getPropertyCollection()); $this->assertInstanceOf('\\ClassGeneration\\Property', $code->getPropertyCollection()->current()); $this->assertCount(1, $code->getPropertyCollection()); $this->assertEquals('test1', $code->getPropertyCollection()->current()->getName()); $code->addProperty(new Property(array('name' => 'test2', 'value' => 1))); $this->assertCount(2, $code->getPropertyCollection()); $this->assertEquals('test2', $code->getPropertyCollection()->last()->getName()); }