[typed-collections] Add AssertSameCountOnCollectionToAssertCountRector

Author: TomasVotruba

config/sets/typed-collections.php

@@ -28,6 +28,7 @@
 use Rector\Doctrine\TypedCollections\Rector\If_\RemoveIfInstanceofCollectionRector;
 use Rector\Doctrine\TypedCollections\Rector\If_\RemoveIsArrayOnCollectionRector;
 use Rector\Doctrine\TypedCollections\Rector\MethodCall\AssertNullOnCollectionToAssertEmptyRector;
+use Rector\Doctrine\TypedCollections\Rector\MethodCall\AssertSameCountOnCollectionToAssertCountRector;
 use Rector\Doctrine\TypedCollections\Rector\MethodCall\SetArrayToNewCollectionRector;
 use Rector\Doctrine\TypedCollections\Rector\New_\RemoveNewArrayCollectionWrapRector;
 use Rector\Doctrine\TypedCollections\Rector\NullsafeMethodCall\RemoveNullsafeOnCollectionRector;
@@ -82,7 +83,10 @@
 
         // cleanup
         RemoveNullsafeOnCollectionRector::class,
-        AssertNullOnCollectionToAssertEmptyRector::class,
+
+        // test assertions
         RemoveAssertNotNullOnCollectionRector::class,
+        AssertNullOnCollectionToAssertEmptyRector::class,
+        AssertSameCountOnCollectionToAssertCountRector::class,
     ]);
 };

rules-tests/TypedCollections/Rector/MethodCall/AssertSameCountOnCollectionToAssertCountRector/AssertSameCountOnCollectionToAssertCountRectorTest.php

@@ -0,0 +1,28 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\Doctrine\Tests\TypedCollections\Rector\MethodCall\AssertSameCountOnCollectionToAssertCountRector;
+
+use Iterator;
+use PHPUnit\Framework\Attributes\DataProvider;
+use Rector\Testing\PHPUnit\AbstractRectorTestCase;
+
+final class AssertSameCountOnCollectionToAssertCountRectorTest 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/TypedCollections/Rector/MethodCall/AssertSameCountOnCollectionToAssertCountRector/Fixture/assert_count.php.inc

@@ -0,0 +1,37 @@
+<?php
+
+namespace Rector\Doctrine\Tests\TypedCollections\Rector\MethodCall\AssertSameCountOnCollectionToAssertCountRector\Fixture;
+
+use Doctrine\Common\Collections\Collection;
+use PHPUnit\Framework\TestCase;
+
+final class AssertCount extends TestCase
+{
+    public Collection $items;
+
+    public function someMethod()
+    {
+        $this->assertSame(10, $this->items->count());
+    }
+}
+
+?>
+-----
+<?php
+
+namespace Rector\Doctrine\Tests\TypedCollections\Rector\MethodCall\AssertSameCountOnCollectionToAssertCountRector\Fixture;
+
+use Doctrine\Common\Collections\Collection;
+use PHPUnit\Framework\TestCase;
+
+final class AssertCount extends TestCase
+{
+    public Collection $items;
+
+    public function someMethod()
+    {
+        $this->assertCount(10, $this->items);
+    }
+}
+
+?>

rules-tests/TypedCollections/Rector/MethodCall/AssertSameCountOnCollectionToAssertCountRector/Fixture/skip_assert_count_on_somethign_else.php.inc

@@ -0,0 +1,15 @@
+<?php
+
+namespace Rector\Doctrine\Tests\TypedCollections\Rector\MethodCall\AssertSameCountOnCollectionToAssertCountRector\Fixture;
+
+use PHPUnit\Framework\TestCase;
+
+final class SkipAssertCountOnSomethingElse extends TestCase
+{
+    public array $items;
+
+    public function someMethod()
+    {
+        $this->assertSame(10, $this->items->count());
+    }
+}

rules-tests/TypedCollections/Rector/MethodCall/AssertSameCountOnCollectionToAssertCountRector/config/configured_rule.php

@@ -0,0 +1,10 @@
+<?php
+
+declare(strict_types=1);
+
+use Rector\Config\RectorConfig;
+use Rector\Doctrine\TypedCollections\Rector\MethodCall\AssertSameCountOnCollectionToAssertCountRector;
+
+return static function (RectorConfig $rectorConfig): void {
+    $rectorConfig->rule(AssertSameCountOnCollectionToAssertCountRector::class);
+};

rules/TypedCollections/Rector/Expression/RemoveAssertNotNullOnCollectionRector.php

@@ -4,11 +4,11 @@
 
 namespace Rector\Doctrine\TypedCollections\Rector\Expression;
 
-use PHPUnit\Framework\Assert;
 use PhpParser\Node;
 use PhpParser\Node\Expr\StaticCall;
 use PhpParser\Node\Stmt\Expression;
 use PhpParser\NodeVisitor;
+use PHPUnit\Framework\Assert;
 use Rector\Doctrine\TypedCollections\TypeAnalyzer\CollectionTypeDetector;
 use Rector\PHPUnit\Enum\PHPUnitClassName;
 use Rector\Rector\AbstractRector;

rules/TypedCollections/Rector/MethodCall/AssertSameCountOnCollectionToAssertCountRector.php

@@ -0,0 +1,103 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\Doctrine\TypedCollections\Rector\MethodCall;
+
+use PhpParser\Node;
+use PhpParser\Node\Arg;
+use PhpParser\Node\Expr\MethodCall;
+use PhpParser\Node\Identifier;
+use Rector\Doctrine\TypedCollections\TypeAnalyzer\CollectionTypeDetector;
+use Rector\PHPUnit\NodeAnalyzer\TestsNodeAnalyzer;
+use Rector\Rector\AbstractRector;
+use Symplify\RuleDocGenerator\ValueObject\CodeSample\CodeSample;
+use Symplify\RuleDocGenerator\ValueObject\RuleDefinition;
+
+/**
+ * @see \Rector\Doctrine\Tests\TypedCollections\Rector\MethodCall\AssertSameCountOnCollectionToAssertCountRector\AssertSameCountOnCollectionToAssertCountRectorTest
+ */
+final class AssertSameCountOnCollectionToAssertCountRector extends AbstractRector
+{
+    public function __construct(
+        private readonly TestsNodeAnalyzer $testsNodeAnalyzer,
+        private readonly CollectionTypeDetector $collectionTypeDetector
+    ) {
+    }
+
+    public function getRuleDefinition(): RuleDefinition
+    {
+        return new RuleDefinition(
+            'Change $this->assertSame(5, $collection->count()) to $this->assertCount(5, $collection) in tests',
+            [
+                new CodeSample(
+                    <<<'CODE_SAMPLE'
+use Doctrine\Common\Collections\Collection;
+
+final class SomeClass extends \PHPUnit\Framework\TestCase
+{
+    private Collection $items;
+
+    public function test(): void
+    {
+        $this->assertSame(5, $this->items->count());
+    }
+}
+CODE_SAMPLE
+                    ,
+                    <<<'CODE_SAMPLE'
+use Doctrine\Common\Collections\Collection;
+
+final class SomeClass extends \PHPUnit\Framework\TestCase
+{
+    private Collection $items;
+
+    public function test(): void
+    {
+        $this->assertCount(5, $this->items);
+    }
+}
+CODE_SAMPLE
+                )]
+        );
+    }
+
+    public function getNodeTypes(): array
+    {
+        return [MethodCall::class];
+
+    }
+
+    /**
+     * @param MethodCall $node
+     */
+    public function refactor(Node $node): MethodCall|null
+    {
+        if ($node->isFirstClassCallable()) {
+            return null;
+        }
+
+        if (! $this->isName($node->name, 'assertSame')) {
+            return null;
+        }
+
+        if (! $this->testsNodeAnalyzer->isInTestClass($node)) {
+            return null;
+        }
+
+        $comparedArg = $node->getArgs()[1]
+            ->value;
+
+        if ($comparedArg instanceof MethodCall && $this->isName(
+            $comparedArg->name,
+            'count'
+        ) && $this->collectionTypeDetector->isCollectionType($comparedArg->var)) {
+            $node->name = new Identifier('assertCount');
+            $node->args[1] = new Arg($comparedArg->var);
+
+            return $node;
+        }
+
+        return null;
+    }
+}