Method & Property: setVisibility() checks argument

Author: dg

src/PhpGenerator/ClassType.php

@@ -515,7 +515,7 @@ public function addMethod($name)
 	{
 		$method = new Method;
 		if ($this->type === 'interface') {
-			$method->setVisibility('')->setBody(FALSE);
+			$method->setVisibility(NULL)->setBody(FALSE);
 		} else {
 			$method->setVisibility('public');
 		}

src/PhpGenerator/Method.php

@@ -32,7 +32,7 @@ class Method extends Nette\Object
 	/** @var bool */
 	private $static = FALSE;
 
-	/** @var string  public|protected|private or none */
+	/** @var string|NULL  public|protected|private */
 	private $visibility;
 
 	/** @var bool */
@@ -252,11 +252,14 @@ public function isStatic()
 
 
 	/**
-	 * @param  string  public|protected|private
+	 * @param  string|NULL  public|protected|private
 	 * @return self
 	 */
 	public function setVisibility($val)
 	{
+		if (!in_array($val, array('public', 'protected', 'private', NULL), TRUE)) {
+			throw new Nette\InvalidArgumentException('Argument must be public|protected|private|NULL.');
+		}
 		$this->visibility = (string) $val;
 		return $this;
 	}

src/PhpGenerator/Property.php

@@ -114,6 +114,9 @@ public function isStatic()
 	 */
 	public function setVisibility($val)
 	{
+		if (!in_array($val, array('public', 'protected', 'private'), TRUE)) {
+			throw new Nette\InvalidArgumentException('Argument must be public|protected|private.');
+		}
 		$this->visibility = (string) $val;
 		return $this;
 	}