Added return type to setters

320da456 · Petr Besir Horáček · b202ac31 · 320da456
Commit 320da456 authored 6 years ago by Petr Besir Horáček
Hide whitespace changes
Inline Side-by-side

Showing

with 2 additions and 1 deletion
+2 -1
--- a/src/Generator/DataGenerator.php
+++ b/src/Generator/DataGenerator.php
@@ -280,7 +280,8 @@ class DataGenerator
 	{
 		$setter = $this->class->addMethod('set' . ucfirst($this->toCamelCase($name)))
 			->addBody("\$this->? = $?;\n", [$this->toCamelCase($name), $this->toCamelCase($name)])
-			->addBody('return $this;');
+			->addBody('return $this;')
+			->setReturnType('self');

 		$setter->addParameter($this->toCamelCase($name))
 			->setNullable($this->isPropertyNullable($propertyDefinition))