Testcase for #52

Author: gossi

src/model/parts/TraitsPart.php

@@ -109,7 +109,7 @@ public function removeTrait($trait) {
 
 		$index = array_search($name, $this->traits);
 		if ($index !== false) {
-			unset($this->traits[$name]);
+			unset($this->traits[$index]);
 
 			if ($trait instanceof PhpTrait) {
 				$qname = $trait->getQualifiedName();

tests/model/ClassTest.php

@@ -13,7 +13,7 @@
  * @group model
  */
 class ClassTest extends \PHPUnit_Framework_TestCase {
-	
+
 	use ModelAssertions;
 	use ValueTests;
 
@@ -163,6 +163,12 @@ public function testTraits() {
 		$this->assertTrue($class->hasUseStatement('other\name\space\Trait'));
 		$this->assertSame($class, $class->removeTrait('other\name\space\Trait'));
 		$this->assertTrue($class->hasUseStatement('other\name\space\Trait'));
+
+		$class = new PhpClass('my\name\space\Class');
+		$class->addTrait('my\trait');
+		$this->assertEquals(1, count($class->getTraits()));
+		$class->removeTrait('my\trait');
+		$this->assertEquals(0, count($class->getTraits()));
 	}
 
 	public function testProperties() {
@@ -227,5 +233,5 @@ public function testDescripion() {
 		$class->setDescription(['multiline', 'description']);
 		$this->assertEquals("multiline\ndescription", $class->getDescription());
 	}
-	
+
 }