[fix] Keep original message in AssertEqualsOrAssertSameFloatParameterToSpecificMethodsTypeRector (#406)

Author: TomasVotruba

rules-tests/CodeQuality/Rector/MethodCall/AssertEqualsOrAssertSameFloatParameterToSpecificMethodsTypeRector/Fixture/keep_delta_message.php.inc

@@ -0,0 +1,33 @@
+<?php
+
+namespace Rector\PHPUnit\Tests\CodeQuality\Rector\MethodCall\AssertEqualsOrAssertSameFloatParameterToSpecificMethodsTypeRector\Fixture;
+
+use PHPUnit\Framework\TestCase;
+
+final class KeepDeltaMessage extends TestCase
+{
+    public function test()
+    {
+        $value = 10.20001;
+        $this->assertSame(10.20, $value, 'Some message');
+    }
+}
+
+?>
+-----
+<?php
+
+namespace Rector\PHPUnit\Tests\CodeQuality\Rector\MethodCall\AssertEqualsOrAssertSameFloatParameterToSpecificMethodsTypeRector\Fixture;
+
+use PHPUnit\Framework\TestCase;
+
+final class KeepDeltaMessage extends TestCase
+{
+    public function test()
+    {
+        $value = 10.20001;
+        $this->assertEqualsWithDelta(10.20, $value, PHP_FLOAT_EPSILON, 'Some message');
+    }
+}
+
+?>

rules/CodeQuality/Rector/MethodCall/AssertEqualsOrAssertSameFloatParameterToSpecificMethodsTypeRector.php

@@ -38,16 +38,12 @@ public function getRuleDefinition(): RuleDefinition
                     // code before
                     <<<'CODE_SAMPLE'
 $this->assertSame(10.20, $value);
-$this->assertEquals(10.20, $value);
 $this->assertEquals(10.200, $value);
-$this->assertSame(10, $value);
 CODE_SAMPLE
                     ,
                     <<<'CODE_SAMPLE'
 $this->assertEqualsWithDelta(10.20, $value, PHP_FLOAT_EPSILON);
-$this->assertEqualsWithDelta(10.20, $value, PHP_FLOAT_EPSILON);
 $this->assertEqualsWithDelta(10.200, $value, PHP_FLOAT_EPSILON);
-$this->assertSame(10, $value);
 CODE_SAMPLE
                 ),
             ]
@@ -82,11 +78,17 @@ public function refactor(Node $node): ?Node
             return null;
         }
 
+        $customMessageArg = $args[2] ?? null;
+
         $newMethodCall = $this->assertCallFactory->createCallWithName($node, 'assertEqualsWithDelta');
         $newMethodCall->args[0] = $args[0];
         $newMethodCall->args[1] = $args[1];
         $newMethodCall->args[2] = new Arg(new ConstFetch(new Name('PHP_FLOAT_EPSILON')));
 
+        if ($customMessageArg instanceof Arg) {
+            $newMethodCall->args[] = $customMessageArg;
+        }
+
         return $newMethodCall;
     }