From 989c3ada2ed9f47cece73510fd9f8bec6f3ad4c1 Mon Sep 17 00:00:00 2001
From: adrew <klimichkartorgnusov@gmail.com>
Date: Sat, 11 Dec 2021 22:51:18 +0300
Subject: [PATCH] Fix type inference of closure params

---
 .../Expression/Call/ArgumentsAnalyzer.php     |  4 +---
 tests/CallableTest.php                        | 22 +++++++++++++++++++
 tests/FunctionCallTest.php                    |  2 --
 3 files changed, 23 insertions(+), 5 deletions(-)

diff --git a/src/Psalm/Internal/Analyzer/Statements/Expression/Call/ArgumentsAnalyzer.php b/src/Psalm/Internal/Analyzer/Statements/Expression/Call/ArgumentsAnalyzer.php
index ac083fb1b..bf2b05532 100644
--- a/src/Psalm/Internal/Analyzer/Statements/Expression/Call/ArgumentsAnalyzer.php
+++ b/src/Psalm/Internal/Analyzer/Statements/Expression/Call/ArgumentsAnalyzer.php
@@ -181,8 +181,6 @@ class ArgumentsAnalyzer
 
             if (($arg->value instanceof PhpParser\Node\Expr\Closure
                     || $arg->value instanceof PhpParser\Node\Expr\ArrowFunction)
-                && $template_result
-                && $template_result->lower_bounds
                 && $param
                 && !$arg->value->getDocComment()
             ) {
@@ -191,7 +189,7 @@ class ArgumentsAnalyzer
                     $args,
                     $method_id,
                     $context,
-                    $template_result,
+                    $template_result ?? new TemplateResult([], []),
                     $argument_offset,
                     $arg,
                     $param
diff --git a/tests/CallableTest.php b/tests/CallableTest.php
index 83b42fff4..c0abc4459 100644
--- a/tests/CallableTest.php
+++ b/tests/CallableTest.php
@@ -63,6 +63,28 @@ class CallableTest extends TestCase
                 'error_levels' => [],
                 '7.4',
             ],
+            'inferArgFromClassContext' => [
+                '<?php
+                    final class Calc
+                    {
+                        /**
+                         * @param Closure(int, int): int $_fn
+                         */
+                        public function __invoke(Closure $_fn): void
+                        {
+                            return $_fn(42, 42);
+                        }
+                    }
+
+                    $calc = new Calc();
+
+                    $a = $calc(fn($a, $b) => $a + $b);',
+                'assertions' => [
+                    '$a' => 'int',
+                ],
+                'error_levels' => [],
+                '7.4',
+            ],
             'varReturnType' => [
                 '<?php
                     $add_one = function(int $a) : int {
diff --git a/tests/FunctionCallTest.php b/tests/FunctionCallTest.php
index 2b425b3ec..e13296489 100644
--- a/tests/FunctionCallTest.php
+++ b/tests/FunctionCallTest.php
@@ -1377,8 +1377,6 @@ class FunctionCallTest extends TestCase
                 '<?php
                     /**
                      * @param string[] $ids
-                     * @psalm-suppress MissingClosureReturnType
-                     * @psalm-suppress MixedArgumentTypeCoercion
                      */
                     function(array $ids): array {
                         return \preg_replace_callback(