Fix PDOException::getCode return type

phpstan · Feb 14, 2022 · 64e7ed5 · 64e7ed5
1 parent 832a9a1
commit 64e7ed5
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 11 deletions.
diff --git a/src/Type/Php/ThrowableReturnTypeExtension.php b/src/Type/Php/ThrowableReturnTypeExtension.php
@@ -50,15 +50,15 @@ public function getTypeFromMethodCall(MethodReflection $methodReflection, Method
 					continue 2;
 				}
 			}
-			if ($pdoException->isSuperTypeOf($classType)->yes()) {
-				$types[] = new StringType();
-				continue;
-			}
 
-			if (in_array(strtolower($class), [
-				'throwable',
-				'exception',
-			], true)) {
+			if (
+				$pdoException->isSuperTypeOf($classType)->yes()
+				|| in_array(strtolower($class), [
+					'throwable',
+					'exception',
+					'runtimeexception',
+				], true)
+			) {
 				$types[] = new BenevolentUnionType([new IntegerType(), new StringType()]);
 				continue;
 			}

diff --git a/tests/PHPStan/Analyser/data/bug-6001.php b/tests/PHPStan/Analyser/data/bug-6001.php
@@ -11,10 +11,11 @@ public function doFoo(\Throwable $t): void
 	{
 		assertType('(int|string)', (new \Exception())->getCode());
 		assertType('(int|string)', $t->getCode());
-		assertType('int', (new \RuntimeException())->getCode());
-		assertType('string', (new \PDOException())->getCode());
+		assertType('int|string', (new \RuntimeException())->getCode());
+		assertType('int', (new \LogicException())->getCode());
+		assertType('int|string', (new \PDOException())->getCode());
 		assertType('int', (new MyException())->getCode());
-		assertType('string', (new SubPDOException())->getCode());
+		assertType('int|string', (new SubPDOException())->getCode());
 		assertType('1|2|3', (new ExceptionWithMethodTag())->getCode());
 	}