Prevent TypeParseException when dealing with nested template in assertion

src/Psalm/Internal/PhpVisitor/Reflector/FunctionLikeDocblockScanner.php

@@ -510,7 +510,9 @@ class FunctionLikeDocblockScanner
                     $self_fqcln,
                     null,
                     true
-                )
+                ),
+                null,
+                $function_template_types + $class_template_types
             );
         } catch (TypeParseTreeException $e) {
             $storage->docblock_issues[] = new InvalidDocblock(

tests/Template/FunctionTemplateAssertTest.php

@@ -767,6 +767,17 @@ class FunctionTemplateAssertTest extends TestCase
                         }
                     }'
             ],
+            'noCrashWhenAssertingTemplatedKeyOf' => [
+                '<?php
+                    /**
+                     * @template TArray as array
+                     * @template TCandidateKey as array-key
+                     * @param TArray $arr
+                     * @param array-key $key
+                     * @psalm-assert key-of<TArray> $key
+                     */
+                    function keyExists(array $arr, $key) : void {}'
+            ],
         ];
     }