skip setup method in DecorateWillReturnMapWithExpectsMockRector

Author: TomasVotruba

rules-tests/CodeQuality/Rector/Expression/DecorateWillReturnMapWithExpectsMockRector/Fixture/skip_from_setup.php.inc

@@ -0,0 +1,14 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\PHPUnit\Tests\CodeQuality\Rector\Expression\DecorateWillReturnMapWithExpectsMockRector\Fixture;
+
+final class SkipFromSetup extends \PHPUnit\Framework\TestCase
+{
+    protected function setUp(): void
+    {
+        $someMock = $this->createMock(\stdClass::class);
+        $someMock->method('some')->willReturnMap([1, 2]);
+    }
+}

rules/CodeQuality/Rector/Expression/DecorateWillReturnMapWithExpectsMockRector.php

@@ -12,7 +12,9 @@
 use PhpParser\Node\Identifier;
 use PhpParser\Node\Scalar\Int_;
 use PhpParser\Node\Stmt\Expression;
+use Rector\PHPStan\ScopeFetcher;
 use Rector\Rector\AbstractRector;
+use Rector\ValueObject\MethodName;
 use Symplify\RuleDocGenerator\ValueObject\CodeSample\CodeSample;
 use Symplify\RuleDocGenerator\ValueObject\RuleDefinition;
 
@@ -92,6 +94,13 @@ public function refactor(Node $node)
             return null;
         }
 
+        $scope = ScopeFetcher::fetch($node);
+
+        // allowed as can be flexible
+        if ($scope->getFunctionName() === MethodName::SET_UP) {
+            return null;
+        }
+
         $topmostCall = $this->resolveTopmostCall($methodCall);
 
         // already covered