added support for class constants (#25)

nette · Jun 8, 2017 · ced817c · ced817c
1 parent 8605fd1
commit ced817c
Show file tree

Hide file tree

Showing 5 changed files with 28 additions and 1 deletion.
diff --git a/src/PhpGenerator/Factory.php b/src/PhpGenerator/Factory.php
@@ -46,6 +46,7 @@ public function fromClassReflection(\ReflectionClass $from): ClassType
 			}
 		}
 		$class->setMethods($methods);
+		$class->setConstants($from->getConstants());
 		return $class;
 	}
 

diff --git a/tests/PhpGenerator/ClassType.from.expect b/tests/PhpGenerator/ClassType.from.expect
@@ -64,3 +64,9 @@ class Class3
 	public $prop1;
 
 }
+
+class Class4
+{
+	const THE_CONSTANT = 9;
+
+}
diff --git a/tests/PhpGenerator/ClassType.from.php71.expect b/tests/PhpGenerator/ClassType.from.php71.expect
@@ -15,4 +15,11 @@ class ClassA
 	{
 	}
 
-}
+}
+
+class ClassB
+{
+	const THE_PRIVATE_CONSTANT = 9;
+	const THE_PUBLIC_CONSTANT = 9;
+
+}
diff --git a/tests/PhpGenerator/ClassType.from.php71.phpt b/tests/PhpGenerator/ClassType.from.php71.phpt
@@ -23,7 +23,14 @@ class ClassA
 	function func3(): void {}
 }
 
+class ClassB
+{
+	private const THE_PRIVATE_CONSTANT = 9;
+	public const THE_PUBLIC_CONSTANT = 9;
+}
+
 
 $res[] = ClassType::from(ClassA::class);
+$res[] = ClassType::from(ClassB::class);
 
 Assert::matchFile(__DIR__ . '/ClassType.from.php71.expect', implode("\n", $res));
diff --git a/tests/PhpGenerator/ClassType.from.phpt b/tests/PhpGenerator/ClassType.from.phpt
@@ -69,12 +69,18 @@ class Class3
 	public $prop1;
 }
 
+class Class4
+{
+	const THE_CONSTANT = 9;
+}
+
 $res[] = ClassType::from(Interface1::class);
 $res[] = ClassType::from(Interface2::class);
 $res[] = ClassType::from(Class1::class);
 $res[] = ClassType::from(new Class2);
 $obj = new Class3;
 $obj->prop2 = 1;
 $res[] = (new Factory)->fromClassReflection(new \ReflectionObject($obj));
+$res[] = ClassType::from(Class4::class);
 
 Assert::matchFile(__DIR__ . '/ClassType.from.expect', implode("\n", $res));