From 6b04399f130001e11cb3ba8e6cbfc87023eef9fe Mon Sep 17 00:00:00 2001
From: Sam Reed <sam@reedyboy.net>
Date: Tue, 17 Sep 2024 17:57:07 +0100
Subject: [PATCH] WillReturnStrategy.java: Simplify returnArgument (#1859)

Fixes https://github.com/kalessil/phpinspectionsea/issues/1858
---
 .../inspectors/phpUnit/strategy/WillReturnStrategy.java          | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/main/java/com/kalessil/phpStorm/phpInspectionsEA/inspectors/phpUnit/strategy/WillReturnStrategy.java b/src/main/java/com/kalessil/phpStorm/phpInspectionsEA/inspectors/phpUnit/strategy/WillReturnStrategy.java
index ea8fb4f5db..d55e3d3ae8 100644
--- a/src/main/java/com/kalessil/phpStorm/phpInspectionsEA/inspectors/phpUnit/strategy/WillReturnStrategy.java
+++ b/src/main/java/com/kalessil/phpStorm/phpInspectionsEA/inspectors/phpUnit/strategy/WillReturnStrategy.java
@@ -27,6 +27,7 @@ final public class WillReturnStrategy {
         methodsMapping.put("returnValueMap", "willReturnMap");
         methodsMapping.put("returnCallback", "willReturnCallback");
         methodsMapping.put("returnValue",    "willReturn");
+        methodsMapping.put("returnArgument", "willReturnArgument");
     }
 
     static public void apply(@NotNull String methodName, @NotNull MethodReference reference, @NotNull ProblemsHolder holder) {