include message as well (#428)

Author: TomasVotruba

rules-tests/TypedCollections/Rector/Expression/RemoveAssertNotNullOnCollectionRector/Fixture/also_include_message.php.inc

@@ -0,0 +1,34 @@
+<?php
+
+namespace Rector\Doctrine\Tests\TypedCollections\Rector\Expression\RemoveAssertNotNullOnCollectionRector\Fixture;
+
+use Doctrine\Common\Collections\Collection;
+
+final class AlsoIncludeMessage
+{
+    public Collection $items;
+
+    public function someMethod()
+    {
+        \PHPUnit\Framework\Assert::assertNotNull($this->items, 'some message');
+    }
+}
+
+?>
+-----
+<?php
+
+namespace Rector\Doctrine\Tests\TypedCollections\Rector\Expression\RemoveAssertNotNullOnCollectionRector\Fixture;
+
+use Doctrine\Common\Collections\Collection;
+
+final class AlsoIncludeMessage
+{
+    public Collection $items;
+
+    public function someMethod()
+    {
+    }
+}
+
+?>

rules-tests/TypedCollections/Rector/Expression/RemoveAssertNotNullOnCollectionRector/Fixture/skip_non_collection.php.inc

@@ -0,0 +1,13 @@
+<?php
+
+namespace Rector\Doctrine\Tests\TypedCollections\Rector\Expression\RemoveAssertNotNullOnCollectionRector\Fixture;
+
+final class SkipNonCollection
+{
+    public array $items;
+
+    public function someMethod()
+    {
+        \PHPUnit\Framework\Assert::assertNotNull($this->items);
+    }
+}

rules/TypedCollections/Rector/Expression/RemoveAssertNotNullOnCollectionRector.php

@@ -81,10 +81,6 @@ public function refactor(Node $node): ?int
             return null;
         }
 
-        if (count($staticCall->args) !== 1) {
-            return null;
-        }
-
         $firstArg = $staticCall->getArgs()[0];
         if (! $this->collectionTypeDetector->isCollectionType($firstArg->value)) {
             return null;