ClassType::getTraits() returns TraitUse[] instead of string[] (BC break)

Author: dg

src/PhpGenerator/ClassType.php

@@ -329,15 +329,8 @@ public function setTraits(array $traits): static
 	}
 
 
-	/** @return string[] */
+	/** @return TraitUse[] */
 	public function getTraits(): array
-	{
-		return array_keys($this->traits);
-	}
-
-
-	/** @internal */
-	public function getTraitResolutions(): array
 	{
 		return $this->traits;
 	}

src/PhpGenerator/Printer.php

@@ -135,7 +135,7 @@ public function printClass(ClassType $class, ?PhpNamespace $namespace = null): s
 			: fn($s) => $s;
 
 		$traits = [];
-		foreach ($class->getTraitResolutions() as $trait) {
+		foreach ($class->getTraits() as $trait) {
 			$resolutions = $trait->getResolutions();
 			$traits[] = Helpers::formatDocComment((string) $trait->getComment())
 				. 'use ' . $resolver($trait->getName())

tests/PhpGenerator/ClassType.addMember.phpt

@@ -19,7 +19,7 @@ $class = (new ClassType('Example'))
 Assert::same(['getHandle' => $method], $class->getMethods());
 Assert::same(['handle' => $property], $class->getProperties());
 Assert::same(['ROLE' => $const], $class->getConstants());
-Assert::same(['Foo\Bar'], $class->getTraits());
+Assert::same(['Foo\Bar' => $trait], $class->getTraits());
 Assert::same('', $method->getBody());
 
 

tests/PhpGenerator/ClassType.phpt

@@ -24,7 +24,6 @@ Assert::false($class->isInterface());
 Assert::false($class->isTrait());
 Assert::same([], $class->getExtends());
 Assert::same([], $class->getTraits());
-Assert::same([], $class->getTraitResolutions());
 
 $class
 	->setAbstract(true)
@@ -34,8 +33,8 @@ $class
 	->addComment("Description of class.\nThis is example\n /**/")
 	->addComment('@property-read Nette\Forms\Form $form');
 
-$class->addTrait('ObjectTrait');
-$class->addTrait('AnotherTrait')
+$trait1 = $class->addTrait('ObjectTrait');
+$trait2 = $class->addTrait('AnotherTrait')
 	->addResolution('sayHello as protected');
 
 $class->addConstant('ROLE', 'admin');
@@ -45,7 +44,7 @@ $class->addConstant('ACTIVE', false)
 Assert::false($class->isFinal());
 Assert::true($class->isAbstract());
 Assert::same('ParentClass', $class->getExtends());
-Assert::same(['ObjectTrait', 'AnotherTrait'], $class->getTraits());
+Assert::same(['ObjectTrait' => $trait1, 'AnotherTrait' => $trait2], $class->getTraits());
 Assert::count(2, $class->getConstants());
 Assert::type(Nette\PhpGenerator\Constant::class, $class->getConstants()['ROLE']);