From e7d4d697acb317c9ff8037e9bd612d1518d25143 Mon Sep 17 00:00:00 2001
From: Jack Worman <jworman@healthcall.com>
Date: Sun, 18 Dec 2022 13:13:36 -0600
Subject: [PATCH] Reflection class stub updates

---
 stubs/Php80.phpstub                         | 10 +++--
 stubs/Reflection.phpstub                    | 25 ++++++-----
 tests/ReflectionTest.php                    | 47 +++++++++++++++++++++
 tests/Traits/ValidCodeAnalysisTestTrait.php | 10 ++++-
 4 files changed, 77 insertions(+), 15 deletions(-)
 create mode 100644 tests/ReflectionTest.php

diff --git a/stubs/Php80.phpstub b/stubs/Php80.phpstub
index 5d3c9fcf4cd..e26dffef45b 100644
--- a/stubs/Php80.phpstub
+++ b/stubs/Php80.phpstub
@@ -89,15 +89,17 @@ class ReflectionClass implements Reflector {
     public function getDocComment(): string|false {}
 
     /**
-     * @param ReflectionClass|class-string $class
-     *
+     * @template J as object
+     * @param self<J>|class-string<J> $class
+     * @psalm-assert-if-true self<T&J> $this
      * @psalm-pure
      */
     public function isSubclassOf(self|string $class): bool {}
 
     /**
-     * @param self|class-string $interface
-     *
+     * @template J as object
+     * @param self<J>|interface-string<J> $interface
+     * @psalm-assert-if-true self<T&J> $this
      * @psalm-pure
      */
     public function implementsInterface(self|string $interface): bool {}
diff --git a/stubs/Reflection.phpstub b/stubs/Reflection.phpstub
index 02bfa3859c5..9264e58dca4 100644
--- a/stubs/Reflection.phpstub
+++ b/stubs/Reflection.phpstub
@@ -147,15 +147,13 @@ class ReflectionClass implements Reflector {
     public function getReflectionConstants(?int $filter = null): array {}
 
     /**
-     * @return array<class-string, self>
-     *
+     * @return array<interface-string, self>
      * @psalm-pure
      */
     public function getInterfaces(): array {}
 
     /**
-     * @return list<class-string>
-     *
+     * @return list<interface-string>
      * @psalm-pure
      */
     public function getInterfaceNames(): array {}
@@ -192,7 +190,12 @@ class ReflectionClass implements Reflector {
      */
     public function getModifiers(): bool {}
 
-    /** @psalm-pure */
+    /**
+     * @template J as object
+     * @param J $object
+     * @psalm-assert-if-true T&J $object
+     * @psalm-pure
+     */
     public function isInstance(object $object): bool {}
 
     /**
@@ -222,8 +225,9 @@ class ReflectionClass implements Reflector {
     public function getParentClass() {}
 
     /**
-     * @param ReflectionClass|class-string $class
-     *
+     * @template J as object
+     * @param self<J>|class-string<J> $class
+     * @psalm-assert-if-true self<T&J> $this
      * @psalm-pure
      */
     public function isSubclassOf($class): bool {}
@@ -245,8 +249,9 @@ class ReflectionClass implements Reflector {
     public function isIterable(): bool {}
 
     /**
-     * @param self|class-string $interface
-     *
+     * @template J as object
+     * @param self<J>|interface-string<J> $interface
+     * @psalm-assert-if-true self<T&J> $this
      * @psalm-pure
      */
     public function implementsInterface($interface): bool {}
@@ -275,7 +280,7 @@ class ReflectionClass implements Reflector {
     public function getShortName(): string {}
 
     /**
-     * @return ?array<string>
+     * @return ?list<trait-string>
      * @psalm-ignore-nullable-return
      * @psalm-pure
      */
diff --git a/tests/ReflectionTest.php b/tests/ReflectionTest.php
new file mode 100644
index 00000000000..fb6e54cb387
--- /dev/null
+++ b/tests/ReflectionTest.php
@@ -0,0 +1,47 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Psalm\Tests;
+
+use Psalm\Tests\Traits\ValidCodeAnalysisTestTrait;
+
+class ReflectionTest extends TestCase
+{
+    use ValidCodeAnalysisTestTrait;
+
+    public function providerValidCodeParse(): iterable
+    {
+        yield 'ReflectionClass::isSubclassOf' => [
+            'code' => <<<'PHP'
+                <?php
+                $a = new ReflectionClass(stdClass::class);
+                if (!$a->isSubclassOf(Iterator::class)) {
+                    throw new Exception();
+                }
+                PHP,
+            'assertions' => ['$a===' => 'ReflectionClass<stdClass&Iterator>'],
+        ];
+        yield 'ReflectionClass::implementsInterface' => [
+            'code' => <<<'PHP'
+                <?php
+                $a = new ReflectionClass(stdClass::class);
+                if (!$a->implementsInterface(Iterator::class)) {
+                    throw new Exception();
+                }
+                PHP,
+            'assertions' => ['$a===' => 'ReflectionClass<stdClass&Iterator>'],
+        ];
+        yield 'ReflectionClass::isInstance' => [
+            'code' => <<<'PHP'
+                <?php
+                $a = new stdClass();
+                $b = new ReflectionClass(Iterator::class);
+                if (!$b->isInstance($a)) {
+                    throw new Exception();
+                }
+                PHP,
+            'assertions' => ['$a===' => 'Iterator&stdClass'],
+        ];
+    }
+}
diff --git a/tests/Traits/ValidCodeAnalysisTestTrait.php b/tests/Traits/ValidCodeAnalysisTestTrait.php
index f5f710ecbaa..ea9e70a4661 100644
--- a/tests/Traits/ValidCodeAnalysisTestTrait.php
+++ b/tests/Traits/ValidCodeAnalysisTestTrait.php
@@ -18,7 +18,15 @@
 trait ValidCodeAnalysisTestTrait
 {
     /**
-     * @return iterable<string,array{code:string,assertions?:array<string,string>,ignored_issues?:list<string>,php_version?:string}>
+     * @return iterable<
+     *     string,
+     *     array{
+     *         code: string,
+     *         assertions?: array<string, string>,
+     *         ignored_issues?: list<string>,
+     *         php_version?: string,
+     *     }
+     * >
      */
     abstract public function providerValidCodeParse(): iterable;