[WIP] UnitOfWork::getEntityChangeSet tests update

Author: n-starshinov-dats

src/Type/Doctrine/UnitOfWorkGetEntityChangeSetDynamicReturnTypeExtension.php

@@ -12,11 +12,10 @@
 use PHPStan\Type\Constant\ConstantArrayTypeBuilder;
 use PHPStan\Type\Constant\ConstantIntegerType;
 use PHPStan\Type\Constant\ConstantStringType;
-use PHPStan\Type\Doctrine\DescriptorNotRegisteredException;
 use PHPStan\Type\DynamicMethodReturnTypeExtension;
 use PHPStan\Type\MixedType;
-use PHPStan\Type\Type;
 use PHPStan\Type\ObjectType;
+use PHPStan\Type\Type;
 use PHPStan\Type\TypeCombinator;
 use function count;
 use function is_array;
@@ -109,7 +108,7 @@ private function createChangeSetType(ClassMetadata $metadata): Type
 
 			$builder->setOffsetValueType(
 				new ConstantStringType($fieldName),
-				$fieldBuilder->getArray()
+				$fieldBuilder->getArray(),
 			);
 		}
 
@@ -131,7 +130,7 @@ private function createChangeSetType(ClassMetadata $metadata): Type
 
 				$builder->setOffsetValueType(
 					new ConstantStringType($fieldName),
-					$fieldBuilder->getArray()
+					$fieldBuilder->getArray(),
 				);
 				continue;
 			}
@@ -146,7 +145,7 @@ private function createChangeSetType(ClassMetadata $metadata): Type
 
 			$builder->setOffsetValueType(
 				new ConstantStringType($fieldName),
-				$fieldBuilder->getArray()
+				$fieldBuilder->getArray(),
 			);
 		}
 
@@ -180,7 +179,7 @@ private function getDefaultReturnType(MethodReflection $methodReflection, Scope
 		return ParametersAcceptorSelector::selectFromArgs(
 			$scope,
 			$methodCall->getArgs(),
-			$methodReflection->getVariants()
+			$methodReflection->getVariants(),
 		)->getReturnType();
 	}
 

tests/Type/Doctrine/UnitOfWorkGetEntityChangeSetDynamicReturnTypeExtensionTest.php

@@ -33,4 +33,3 @@ public static function getAdditionalConfigFiles(): array
 	}
 
 }
-

tests/Type/Doctrine/data/UnitOfWorkChangeSet/Entities/RelatedEntity.php

@@ -1,26 +1,3 @@
 <?php declare(strict_types = 1);
 
-use Cache\Adapter\PHPArray\ArrayCachePool;
-use Doctrine\DBAL\DriverManager;
-use Doctrine\ORM\EntityManager;
-use Doctrine\ORM\Mapping\Driver\StaticPHPDriver;
-use Doctrine\ORM\ORMSetup;
-
-$config = ORMSetup::createConfiguration(
-	true,
-	__DIR__,
-	new ArrayCachePool()
-);
-
-$config->setMetadataDriverImpl(new StaticPHPDriver([__DIR__ . '/Entities']));
-
-$config->setProxyNamespace('PHPStan\\Doctrine\\UnitOfWorkChangeSetProxies');
-$config->setAutoGenerateProxyClasses(true);
-
-return new EntityManager(
-	DriverManager::getConnection([
-		'driver' => 'pdo_sqlite',
-		'memory' => true,
-	]),
-	$config
-);
+return require __DIR__ . '/../QueryResult/entity-manager.php';

tests/Type/Doctrine/data/UnitOfWorkChangeSet/Entities/SimpleEntity.php

@@ -3,15 +3,24 @@
 namespace UnitOfWorkChangeSet;
 
 use Doctrine\ORM\UnitOfWork;
-use UnitOfWorkChangeSet\Entities\SimpleEntity;
+use QueryResult\Entities\Many;
+use QueryResult\Entities\Simple;
 use function PHPStan\Testing\assertType;
 
 final class UnitOfWorkChangeSetAssertions
 {
-	public function simpleField(UnitOfWork $unitOfWork, SimpleEntity $entity): void
+	public function simpleField(UnitOfWork $unitOfWork, Simple $entity): void
 	{
 		assertType(
-			'array{foo: array{int, int}, nullableFoo: array{int|null, int|null}, related: array{UnitOfWorkChangeSet\\Entities\\RelatedEntity|null, UnitOfWorkChangeSet\\Entities\\RelatedEntity|null}, relatedCollection: array{Doctrine\\ORM\\PersistentCollection, Doctrine\\ORM\\PersistentCollection}}',
+			'array{intColumn: array{int, int}, floatColumn: array{float, float}, decimalColumn: array{numeric-string&uppercase-string, numeric-string&uppercase-string}, stringColumn: array{string, string}, stringNullColumn: array{string|null, string|null}, mixedColumn: array{mixed, mixed}}',
+			$unitOfWork->getEntityChangeSet($entity)
+		);
+	}
+
+	public function associations(UnitOfWork $unitOfWork, Many $entity): void
+	{
+		assertType(
+			'array{intColumn: array{int, int}, stringColumn: array{string, string}, stringNullColumn: array{string|null, string|null}, datetimeColumn: array{DateTime, DateTime}, datetimeImmutableColumn: array{DateTimeImmutable, DateTimeImmutable}, simpleArrayColumn: array{list<string>, list<string>}, one: array{QueryResult\\Entities\\One, QueryResult\\Entities\\One}, oneNull: array{QueryResult\\Entities\\One|null, QueryResult\\Entities\\One|null}, oneDefaultNullability: array{QueryResult\\Entities\\One|null, QueryResult\\Entities\\One|null}, compoundPk: array{QueryResult\\Entities\\CompoundPk|null, QueryResult\\Entities\\CompoundPk|null}, compoundPkAssoc: array{QueryResult\\Entities\\CompoundPkAssoc|null, QueryResult\\Entities\\CompoundPkAssoc|null}}',
 			$unitOfWork->getEntityChangeSet($entity)
 		);
 	}