public function testAttributeSetRemoveGet(AttributeAware $entity) { $entity->setAttribute('test', 'value'); $this->assertEquals('value', $entity->getAttribute('test')); $entity->removeAttribute('test'); $this->assertEquals(null, $entity->getAttribute('test')); }