Fix TypeError if no type given for param or property

Author: codeliner

src/Code/MethodGenerator.php

@@ -185,9 +185,13 @@ public function generate(): ClassMethod
         $docBlockTypes = [];
 
         foreach ($this->getParameters() as $parameter) {
-            $type = $parameter->getType()->isNullable()
-                ? $parameter->getType()->type() . '|null'
-                : $parameter->getType()->type();
+            if(null === $parameter->getType()) {
+                $type = 'mixed';
+            } else {
+                $type = $parameter->getType()->isNullable()
+                    ? $parameter->getType()->type() . '|null'
+                    : $parameter->getType()->type();
+            }
 
             if ($typeHint = $parameter->getTypeDocBlockHint()) {
                 $type = $typeHint;

src/Code/ParameterGenerator.php

@@ -88,7 +88,7 @@ public function setType($type): self
         return $this;
     }
 
-    public function getType(): TypeGenerator
+    public function getType(): ?TypeGenerator
     {
         return $this->type;
     }

src/Code/PropertyGenerator.php

@@ -82,7 +82,7 @@ public function setType(string $type): self
         return $this;
     }
 
-    public function getType(): TypeGenerator
+    public function getType(): ?TypeGenerator
     {
         return $this->type;
     }

tests/Code/MethodGeneratorTest.php

@@ -56,6 +56,33 @@ public function setType(?string $type);
         $this->assertSame($expectedOutput, $this->printer->prettyPrintFile([$method->generate()]));
     }
 
+    /**
+     * @test
+     */
+    public function it_generates_method_with_mixed_type_doc_block(): void
+    {
+        $method = new MethodGenerator(
+            'setType',
+            [
+                new ParameterGenerator('type'),
+            ]
+        );
+        $method->setDocBlockComment('Sets an awesome type');
+
+        $expectedOutput = <<<'EOF'
+<?php
+
+/**
+ * Sets an awesome type
+ *
+ * @var mixed $type
+ */
+public function setType($type);
+EOF;
+
+        $this->assertSame($expectedOutput, $this->printer->prettyPrintFile([$method->generate()]));
+    }
+
     /**
      * @test
      */

tests/Code/ParameterGeneratorTest.php

@@ -64,6 +64,22 @@ public function it_generates_type(string $type): void
 <?php
 
 $type \$myParameter
+PHP;
+
+        $this->assertSame($expectedOutput, $this->printer->prettyPrintFile([$parameter->generate()]));
+    }
+
+    /**
+     * @test
+     */
+    public function it_works_without_type()
+    {
+        $parameter = new ParameterGenerator('myParameter');
+
+        $expectedOutput = <<<PHP
+<?php
+
+\$myParameter
 PHP;
 
         $this->assertSame($expectedOutput, $this->printer->prettyPrintFile([$parameter->generate()]));