Test IDENTITY

Author: janedbal

Diff for: tests/Platform/Entity/PlatformEntity.php

@@ -23,6 +23,15 @@ class PlatformEntity
 	#[ORM\Column(type: 'string', nullable: false)]
 	public $id;
 
+	/**
+	 * @ORM\ManyToOne(targetEntity=PlatformRelatedEntity::class)
+	 * @ORM\JoinColumn(name="related_entity_id", referencedColumnName="id", nullable=false)
+	 * @var PlatformRelatedEntity
+	 */
+	#[ORM\ManyToOne(targetEntity: PlatformRelatedEntity::class)]
+	#[ORM\JoinColumn(name: 'related_entity_id', referencedColumnName: 'id', nullable: false)]
+	public $related_entity;
+
 	/**
 	 * @ORM\Column(type="string", name="col_string", nullable=false)
 	 * @var string

Diff for: tests/Platform/Entity/PlatformRelatedEntity.php

@@ -0,0 +1,25 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Platform\Entity;
+
+use Doctrine\ORM\Mapping as ORM;
+
+/**
+ * @ORM\Table(name="test_related")
+ * @ORM\Entity
+ */
+#[ORM\Table(name: 'test_related')]
+#[ORM\Entity]
+class PlatformRelatedEntity
+{
+
+	/**
+	 * @ORM\Id
+	 * @ORM\Column(type="integer", nullable=false)
+	 * @var int
+	 */
+	#[ORM\Id]
+	#[ORM\Column(type: 'integer', nullable: false)]
+	public $id;
+
+}

Diff for: tests/Platform/QueryResultTypeWalkerFetchTypeMatrixTest.php

@@ -22,6 +22,7 @@
 use PHPStan\Doctrine\Driver\DriverType;
 use PHPStan\Php\PhpVersion;
 use PHPStan\Platform\Entity\PlatformEntity;
+use PHPStan\Platform\Entity\PlatformRelatedEntity;
 use PHPStan\Testing\PHPStanTestCase;
 use PHPStan\Type\Accessory\AccessoryNumericStringType;
 use PHPStan\Type\BooleanType;
@@ -3509,7 +3510,21 @@ public static function provideCases(): iterable
 			'stringify' => self::STRINGIFY_DEFAULT,
 		];
 
-		// TODO test IDENTITY
+		yield 'IDENTITY(t.related_entity)' => [
+			'data' => self::dataDefault(),
+			'select' => 'SELECT IDENTITY(t.related_entity) FROM %s t',
+			'mysql' => self::int(),
+			'sqlite' => self::int(),
+			'pdo_pgsql' => self::int(),
+			'pgsql' => self::int(),
+			'mssql' => self::mixed(),
+			'mysqlResult' => 1,
+			'sqliteResult' => 1,
+			'pdoPgsqlResult' => 1,
+			'pgsqlResult' => 1,
+			'mssqlResult' => 1,
+			'stringify' => self::STRINGIFY_DEFAULT,
+		];
 	}
 
 	/**
@@ -3653,8 +3668,14 @@ private function getQuery(
 		$schemaTool->dropSchema($classes);
 		$schemaTool->createSchema($classes);
 
+		$relatedEntity = new PlatformRelatedEntity();
+		$relatedEntity->id = 1;
+		$entityManager->persist($relatedEntity);
+
 		foreach ($data as $rowData) {
 			$entity = new PlatformEntity();
+			$entity->related_entity = $relatedEntity;
+
 			foreach ($rowData as $column => $value) {
 				$entity->$column = $value; // @phpstan-ignore-line Intentionally dynamic
 			}