Fix call with no class yet

Jan0707 · Feb 10, 2025 · 34c99d9 · 34c99d9
1 parent 8642383
commit 34c99d9
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/src/Type/ObjectProphecy/WillExtendOrImplementDynamicReturnTypeExtension.php b/src/Type/ObjectProphecy/WillExtendOrImplementDynamicReturnTypeExtension.php
@@ -70,20 +70,19 @@ public function getTypeFromMethodCall(
         $argumentType = $scope->getType($args[0]->value);
 
         $templateObjectType = $calledOnType->getTemplateType(Prophecy\ObjectProphecy::class, 'T');
-        $classObjectType = $argumentType->getClassStringObjectType();
 
-        if ($templateObjectType->isObject()->no()) {
-            return $returnType;
+        $objects = [];
+        if ($templateObjectType->isObject()->no()
+            || \count($templateObjectType->getObjectClassNames()) !== 0) {
+            $objects[] = $templateObjectType;
         }
 
+        $classObjectType = $argumentType->getClassStringObjectType();
+
         if ($classObjectType->isObject()->no()) {
             return $returnType;
         }
 
-        $objects = [];
-        if (\count($templateObjectType->getObjectClassNames()) !== 0) {
-            $objects[] = $templateObjectType;
-        }
         $objects[] = $classObjectType;
 
         return new Type\Generic\GenericObjectType(