Merge pull request #50 from facile-it/feature/reference

Support methods returning a reference

Author: xzhayon

src/Generator/Template/MethodTemplate.php

@@ -14,7 +14,7 @@ class MethodTemplate implements TemplateInterface
     use VisibilityTrait;
 
     private const TEMPLATE = '
-        %s %s function %s(%s)%s
+        %s %s function %s %s(%s)%s
         {
             %s $this->__call("%s", func_get_args());
         }
@@ -42,6 +42,7 @@ protected static function doGenerate(\ReflectionMethod $method): string
         $visibility = static::getVisibility($method);
 
         $static = $method->isStatic() ? 'static' : '';
+        $reference = $method->returnsReference() ? '&' : '';
 
         $parameters = $method->getParameters();
         $parametersCode = [];
@@ -66,6 +67,7 @@ protected static function doGenerate(\ReflectionMethod $method): string
             self::TEMPLATE,
             $visibility,
             $static,
+            $reference,
             $methodName,
             implode(',', $parametersCode),
             $returnType,

src/Tests/IncompleteAbstractTestClass.php

@@ -36,6 +36,11 @@ public function getCallable(): callable
         };
     }
 
+    public function & getReference(): array
+    {
+        return [];
+    }
+
     public function withArgument(int $argument): int
     {
         return $argument;

src/Tests/MokaMockingStrategyTestCase.php

@@ -40,6 +40,7 @@ protected function setUp(): void
             '$private' => mt_rand(),
             'isTrue' => (bool)random_int(0, 1),
             'getInt' => mt_rand(),
+            'getReference' => [],
             'withArgument' => mt_rand(),
             'throwException' => new \Exception((string)mt_rand())
         ];

tests/Generator/Template/MethodTemplateTest.php

@@ -15,6 +15,6 @@ public function testGenerate()
             new \ReflectionMethod(FooTestClass::class, 'getCallable')
         );
 
-        $this->assertRegExp('/public +function getCallable\(/', $code);
+        $this->assertRegExp('/public +function +getCallable\(/', $code);
     }
 }