ConstantsAware: added hasConstant() (#133)

nette · Jul 30, 2023 · d34e60d · d34e60d
1 parent bdb3fc6
commit d34e60d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/PhpGenerator/Traits/ConstantsAware.php b/src/PhpGenerator/Traits/ConstantsAware.php
@@ -62,4 +62,10 @@ public function removeConstant(string $name): static
 		unset($this->consts[$name]);
 		return $this;
 	}
+
+
+	public function hasConstant(string $name): bool
+	{
+		return isset($this->consts[$name]);
+	}
 }
diff --git a/tests/PhpGenerator/ClassType.phpt b/tests/PhpGenerator/ClassType.phpt
@@ -41,6 +41,8 @@ $class->addConstant('ROLE', 'admin');
 $class->addConstant('ACTIVE', false)
 	->setFinal()
 	->setType('bool');
+Assert::true($class->hasConstant('ROLE'));
+Assert::false($class->hasConstant('xxx'));
 
 Assert::false($class->isFinal());
 Assert::true($class->isAbstract());