Factory: $withBodies cannot be used for internal classes/functions

nette · Oct 15, 2023 · 894a440 · 894a440
1 parent c7e7b82
commit 894a440
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/src/PhpGenerator/Factory.php b/src/PhpGenerator/Factory.php
@@ -37,8 +37,8 @@ public function fromClassReflection(
 		if ($materializeTraits !== null) {
 			trigger_error(__METHOD__ . '() parameter $materializeTraits has been removed (is always false).', E_USER_DEPRECATED);
 		}
-		if ($withBodies && $from->isAnonymous()) {
-			throw new Nette\NotSupportedException('The $withBodies parameter cannot be used for anonymous functions.');
+		if ($withBodies && ($from->isAnonymous() || $from->isInternal())) {
+			throw new Nette\NotSupportedException('The $withBodies parameter cannot be used for anonymous or internal classes.');
 		}
 
 		$enumIface = null;
@@ -183,8 +183,8 @@ public function fromFunctionReflection(\ReflectionFunction $from, bool $withBody
 		$function->setReturnType((string) $from->getReturnType());
 
 		if ($withBody) {
-			if ($from->isClosure()) {
-				throw new Nette\NotSupportedException('The $withBody parameter cannot be used for closures.');
+			if ($from->isClosure() || $from->isInternal()) {
+				throw new Nette\NotSupportedException('The $withBody parameter cannot be used for closures or internal functions.');
 			}
 
 			$function->setBody($this->getExtractor($from)->extractFunctionBody($from->name));

diff --git a/tests/PhpGenerator/ClassType.from.bodies.phpt b/tests/PhpGenerator/ClassType.from.bodies.phpt
@@ -11,8 +11,8 @@ require __DIR__ . '/fixtures/bodies.php';
 
 Assert::exception(
 	fn() => ClassType::from(PDO::class, withBodies: true),
-	Nette\InvalidStateException::class,
-	'Source code of PDO not found.',
+	Nette\NotSupportedException::class,
+	'The $withBodies parameter cannot be used for anonymous or internal classes.',
 );
 
 
@@ -23,7 +23,7 @@ Assert::exception(
 		}
 	}, withBodies: true),
 	Nette\NotSupportedException::class,
-	'The $withBodies parameter cannot be used for anonymous functions.',
+	'The $withBodies parameter cannot be used for anonymous or internal classes.',
 );