Replace return with yield when docComments shows return value is a Generator

Author: stefandoorn

src/AspectMock/Intercept/BeforeMockTransformer.php

@@ -80,6 +80,10 @@ public function transform(StreamMetaData $metadata)
                     $beforeDefinition = $method->isStatic()
                         ? $this->beforeStatic
                         : $this->before;
+                    // replace return with yield when doccomment shows it returns a Generator
+                    if(stripos($method->getDocComment(), '@return \Generator') !== false) {
+                        $beforeDefinition = str_replace('return', 'yield', $beforeDefinition);
+                    }
                     $reflectedParams = $method->getParameters();
 
                     $params = [];