added CommentAware::removeComment()

nette · Apr 28, 2024 · f8a95dd · f8a95dd
1 parent 9afa824
commit f8a95dd
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/src/PhpGenerator/Traits/CommentAware.php b/src/PhpGenerator/Traits/CommentAware.php
@@ -39,4 +39,11 @@ public function addComment(string $val): static
 		$this->comment .= $this->comment ? "\n$val" : $val;
 		return $this;
 	}
+
+
+	public function removeComment(): static
+	{
+		$this->comment = null;
+		return $this;
+	}
 }
diff --git a/tests/PhpGenerator/ClassType.phpt b/tests/PhpGenerator/ClassType.phpt
@@ -59,7 +59,9 @@ $class->addProperty('handle')
 	->addComment('@var resource  orignal file handle');
 
 $class->addProperty('order')
-	->setValue(new Literal('RecursiveIteratorIterator::SELF_FIRST'));
+	->setValue(new Literal('RecursiveIteratorIterator::SELF_FIRST'))
+	->addComment('foo')
+	->removeComment();
 
 $class->addProperty('typed1')
 	->setType(Type::Array)