QueryBuilderGetQueryDynamicReturnTypeExtension: handle regular set us…

…ages

src/Type/Doctrine/ArgumentsProcessor.php

@@ -15,7 +15,7 @@ class ArgumentsProcessor
 
 	/**
 	 * @param Arg[] $methodCallArgs
-	 * @return mixed[]
+	 * @return list<mixed>
 	 * @throws DynamicQueryBuilderArgumentException
 	 */
 	public function processArgs(

src/Type/Doctrine/QueryBuilder/QueryBuilderGetQueryDynamicReturnTypeExtension.php

@@ -24,6 +24,7 @@
 use PHPStan\Type\Type;
 use PHPStan\Type\TypeCombinator;
 use Throwable;
+use function array_slice;
 use function count;
 use function in_array;
 use function method_exists;
@@ -152,6 +153,18 @@ public function getTypeFromMethodCall(
 					continue;
 				}
 
+				if ($lowerMethodName === 'set') {
+					try {
+						$args = $this->argumentsProcessor->processArgs($scope, $methodName, array_slice($calledMethodCall->getArgs(), 0, 1));
+					} catch (DynamicQueryBuilderArgumentException $e) {
+						return $defaultReturnType;
+					}
+					if (count($args) === 1) {
+						$queryBuilder->set($args[0], $args[0]);
+						continue;
+					}
+				}
+
 				if (!method_exists($queryBuilder, $methodName)) {
 					continue;
 				}

tests/Rules/Doctrine/ORM/QueryBuilderDqlRuleTest.php

@@ -129,6 +129,16 @@ public function testAndOrVariants(): void
 		$this->analyse([__DIR__ . '/data/query-builder-dql-and-or-variants.php'], []);
 	}
 
+	public function testUpdateIssue(): void
+	{
+		$this->analyse([__DIR__ . '/data/query-builder-dql-update.php'], [
+			[
+				'Could not analyse QueryBuilder with dynamic arguments.',
+				31,
+			],
+		]);
+	}
+
 	public function testBranchingPerformance(): void
 	{
 		$this->analyse([__DIR__ . '/data/query-builder-branches-performance.php'], [

tests/Rules/Doctrine/ORM/data/query-builder-dql-update.php

@@ -0,0 +1,40 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Rules\Doctrine\ORM;
+
+use Doctrine\ORM\EntityManager;
+
+class TestUpdateQueryBuilder
+{
+
+	/** @var EntityManager */
+	private $entityManager;
+
+	public function __construct(EntityManager $entityManager)
+	{
+		$this->entityManager = $entityManager;
+	}
+
+	public function nonDynamicSet(int $int, string $title): void
+	{
+		$this->entityManager->createQueryBuilder()
+			->update(MyEntity::class, 'e')
+			->set('e.title', $title)
+			->andWhere('e.id = :int')
+			->setParameter('int', $int)
+			->getQuery()
+			->execute();
+	}
+
+	public function dynamicSet(int $int, string $field, string $title): void
+	{
+		$this->entityManager->createQueryBuilder()
+			->update(MyEntity::class, 'e')
+			->set('e.' . $field, $title)
+			->andWhere('e.id = :int')
+			->setParameter('int', $int)
+			->getQuery()
+			->execute();
+	}
+
+}