[code-quality] Add DecorateWillReturnMapWithExpectsMockRector

Author: TomasVotruba

rules-tests/CodeQuality/Rector/Expression/DecorateWillReturnMapWithExpectsMockRector/DecorateWillReturnMapWithExpectsMockRectorTest.php

@@ -0,0 +1,28 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\PHPUnit\Tests\CodeQuality\Rector\Expression\DecorateWillReturnMapWithExpectsMockRector;
+
+use Iterator;
+use PHPUnit\Framework\Attributes\DataProvider;
+use Rector\Testing\PHPUnit\AbstractRectorTestCase;
+
+final class DecorateWillReturnMapWithExpectsMockRectorTest extends AbstractRectorTestCase
+{
+    #[DataProvider('provideData')]
+    public function test(string $filePath): void
+    {
+        $this->doTestFile($filePath);
+    }
+
+    public static function provideData(): Iterator
+    {
+        return self::yieldFilesFromDirectory(__DIR__ . '/Fixture');
+    }
+
+    public function provideConfigFilePath(): string
+    {
+        return __DIR__ . '/config/configured_rule.php';
+    }
+}

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

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

rules-tests/CodeQuality/Rector/Expression/DecorateWillReturnMapWithExpectsMockRector/config/configured_rule.php

@@ -0,0 +1,8 @@
+<?php
+
+declare(strict_types=1);
+
+use Rector\Config\RectorConfig;
+
+return RectorConfig::configure()
+    ->withRules([\Rector\PHPUnit\CodeQuality\Rector\Expression\DecorateWillReturnMapWithExpectsMockRector::class]);

rules/CodeQuality/Rector/Expression/DecorateWillReturnMapWithExpectsMockRector.php

@@ -0,0 +1,129 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\PHPUnit\CodeQuality\Rector\Expression;
+
+use PhpParser\Node;
+use Rector\PhpParser\Node\Value\ValueResolver;
+use Rector\Rector\AbstractRector;
+use Symplify\RuleDocGenerator\ValueObject\CodeSample\CodeSample;
+use Symplify\RuleDocGenerator\ValueObject\RuleDefinition;
+
+final class DecorateWillReturnMapWithExpectsMockRector extends AbstractRector
+{
+    public function __construct(
+        private readonly ValueResolver $valueResolver
+    ) {
+    }
+
+    public function getRuleDefinition(): RuleDefinition
+    {
+        return new RuleDefinition('Decorate willReturnMap() calls with expects on the mock object if missing', [
+            new CodeSample(
+                <<<'CODE_SAMPLE'
+use PHPUnit\Framework\TestCase;
+use PHPUnit\Framework\MockObject\MockObject;
+
+final class SomeTest extends TestCase
+{
+    private MockObject $someMock;
+
+    protected function setUp(): void
+    {
+        $this->someMock = $this->createMock(SomeClass::class);
+
+        $this->someMock->method("someMethod")
+            ->willReturnMap([
+                ["arg1", "arg2", "result1"],
+                ["arg3", "arg4", "result2"],
+            ]);
+    }
+}
+CODE_SAMPLE
+                ,
+                <<<'CODE_SAMPLE'
+use PHPUnit\Framework\TestCase;
+use PHPUnit\Framework\MockObject\MockObject;
+
+final class SomeTest extends TestCase
+{
+    private MockObject $someMock;
+
+    protected function setUp(): void
+    {
+        $this->someMock = $this->createMock(SomeClass::class);
+
+        $this->someMock->expects($this->exactly(2))
+            ->method("someMethod")
+            ->willReturnMap([
+                ["arg1", "arg2", "result1"],
+                ["arg3", "arg4", "result2"],
+            ]);
+    }
+}
+CODE_SAMPLE
+            ),
+
+        ]);
+    }
+
+    public function getNodeTypes(): array
+    {
+        return [Node\Stmt\Expression::class];
+    }
+
+    /**
+     * @param Node\Stmt\Expression $node
+     * @return Node\Stmt\Expression|null
+     */
+    public function refactor(Node $node)
+    {
+        if (! $node->expr instanceof Node\Expr\MethodCall) {
+            return null;
+        }
+
+        $methodCall = $node->expr;
+        if (! $this->isName($methodCall->name, 'willReturnMap')) {
+            return null;
+        }
+
+        $topmostCall = $this->resolveTopmostCall($methodCall);
+
+        // already covered
+        if ($this->isName($topmostCall->name, 'expects')) {
+            return null;
+        }
+
+        // count values in will map arg
+        $willReturnMapArg = $methodCall->getArgs()[0];
+        if (! $willReturnMapArg->value instanceof Node\Expr\Array_) {
+            return null;
+        }
+
+        $array = $willReturnMapArg->value;
+        $mapCount = count($array->items);
+
+        $topmostCall->var = new Node\Expr\MethodCall(
+            $topmostCall->var,
+            new Node\Identifier('expects'),
+            [new Node\Arg(new Node\Expr\MethodCall(
+                new Node\Expr\Variable('this'),
+                new Node\Identifier('exactly'),
+                [new Node\Arg(new Node\Scalar\Int_($mapCount))]
+            ))]
+        );
+
+        return $node;
+    }
+
+    private function resolveTopmostCall(Node\Expr\MethodCall $methodCall): Node\Expr\MethodCall
+    {
+        $currentCall = $methodCall;
+        while ($currentCall->var instanceof Node\Expr\MethodCall) {
+            $currentCall = $currentCall->var;
+        }
+
+        return $currentCall;
+    }
+}