Fix StaticType::isSuperTypeOf() for subtypes of ObjectType

ondrejmirtes · ondrejmirtes · commit b439fed7b19f · 2023-04-04T17:38:05.000+02:00
diff --git a/phpstan-baseline.neon b/phpstan-baseline.neon
@@ -1450,7 +1450,7 @@ parameters:
 
 		-
 			message: "#^Doing instanceof PHPStan\\\\Type\\\\ObjectType is error\\-prone and deprecated\\. Use Type\\:\\:isObject\\(\\) or Type\\:\\:getObjectClassNames\\(\\) instead\\.$#"
-			count: 3
+			count: 2
 			path: src/Type/StaticType.php
 
 		-
diff --git a/src/Reflection/Native/NativeMethodReflection.php b/src/Reflection/Native/NativeMethodReflection.php
@@ -65,7 +65,10 @@ public function getPrototype(): ClassMemberReflection
 	{
 		try {
 			$prototypeMethod = $this->reflection->getPrototype();
-			$prototypeDeclaringClass = $this->reflectionProvider->getClass($prototypeMethod->getDeclaringClass()->getName());
+			$prototypeDeclaringClass = $this->declaringClass->getAncestorWithClassName($prototypeMethod->getDeclaringClass()->getName());
+			if ($prototypeDeclaringClass === null) {
+				$prototypeDeclaringClass = $this->reflectionProvider->getClass($prototypeMethod->getDeclaringClass()->getName());
+			}
 
 			$tentativeReturnType = null;
 			if ($prototypeMethod->getTentativeReturnType() !== null) {
diff --git a/src/Reflection/Php/PhpMethodReflection.php b/src/Reflection/Php/PhpMethodReflection.php
@@ -104,7 +104,10 @@ public function getPrototype(): ClassMemberReflection
 	{
 		try {
 			$prototypeMethod = $this->reflection->getPrototype();
-			$prototypeDeclaringClass = $this->reflectionProvider->getClass($prototypeMethod->getDeclaringClass()->getName());
+			$prototypeDeclaringClass = $this->declaringClass->getAncestorWithClassName($prototypeMethod->getDeclaringClass()->getName());
+			if ($prototypeDeclaringClass === null) {
+				$prototypeDeclaringClass = $this->reflectionProvider->getClass($prototypeMethod->getDeclaringClass()->getName());
+			}
 
 			$tentativeReturnType = null;
 			if ($prototypeMethod->getTentativeReturnType() !== null) {
diff --git a/src/Type/StaticType.php b/src/Type/StaticType.php
@@ -149,13 +149,6 @@ public function acceptsWithReason(Type $type, bool $strictTypes): AcceptsResult
 
 	public function isSuperTypeOf(Type $type): TrinaryLogic
 	{
-		if ($type instanceof ObjectType) {
-			$classReflection = $type->getClassReflection();
-			if ($classReflection !== null && $classReflection->isFinal()) {
-				$type = new StaticType($classReflection, $type->getSubtractedType());
-			}
-		}
-
 		if ($type instanceof self) {
 			return $this->getStaticObjectType()->isSuperTypeOf($type);
 		}
@@ -165,7 +158,15 @@ public function isSuperTypeOf(Type $type): TrinaryLogic
 		}
 
 		if ($type instanceof ObjectType) {
-			return $this->getStaticObjectType()->isSuperTypeOf($type)->and(TrinaryLogic::createMaybe());
+			$result = $this->getStaticObjectType()->isSuperTypeOf($type);
+			if ($result->yes()) {
+				$classReflection = $type->getClassReflection();
+				if ($classReflection !== null && $classReflection->isFinal()) {
+					return $result;
+				}
+			}
+
+			return $result->and(TrinaryLogic::createMaybe());
 		}
 
 		if ($type instanceof CompoundType) {
diff --git a/tests/PHPStan/Rules/Comparison/StrictComparisonOfDifferentTypesRuleTest.php b/tests/PHPStan/Rules/Comparison/StrictComparisonOfDifferentTypesRuleTest.php
@@ -963,4 +963,23 @@ public function testEnumTips(): void
 		]);
 	}
 
+	public function testBug9142(): void
+	{
+		if (PHP_VERSION_ID < 80100) {
+			$this->markTestSkipped('Test requires PHP 8.1.');
+		}
+
+		$this->checkAlwaysTrueStrictComparison = true;
+		$this->analyse([__DIR__ . '/data/bug-9142.php'], [
+			[
+				'Strict comparison using === between $this(Bug9142\MyEnum) and Bug9142\MyEnum::Three will always evaluate to false.',
+				18,
+			],
+			[
+				'Strict comparison using === between Bug9142\MyEnum and Bug9142\MyEnum::Three will always evaluate to false.',
+				31,
+			],
+		]);
+	}
+
 }
diff --git a/tests/PHPStan/Rules/Comparison/data/bug-9142.php b/tests/PHPStan/Rules/Comparison/data/bug-9142.php
@@ -0,0 +1,38 @@
+<?php // lint >= 8.1
+
+namespace Bug9142;
+
+enum MyEnum: string
+{
+
+	case One = 'one';
+	case Two = 'two';
+	case Three = 'three';
+
+	public function thisTypeWithSubtractedEnumCase(): int
+	{
+		if ($this === self::Three) {
+			return -1;
+		}
+
+		if ($this === self::Three) {
+			return 0;
+		}
+
+		return 1;
+	}
+
+	public function enumTypeWithSubtractedEnumCase(self $self): int
+	{
+		if ($self === self::Three) {
+			return -1;
+		}
+
+		if ($self === self::Three) {
+			return 0;
+		}
+
+		return 1;
+	}
+
+}

Original file line number	Diff line number	Diff line change
`@@ -1450,7 +1450,7 @@ parameters:`
`1450`	`1450`
`1451`	`1451`	`-`
`1452`	`1452`	`message: "#^Doing instanceof PHPStan\\\\Type\\\\ObjectType is error\\-prone and deprecated\\. Use Type\\:\\:isObject\\(\\) or Type\\:\\:getObjectClassNames\\(\\) instead\\.$#"`
`1453`		`- count: 3`
	`1453`	`+ count: 2`
`1454`	`1454`	`path: src/Type/StaticType.php`
`1455`	`1455`
`1456`	`1456`	`-`