test: add functional test for union type

Author: nikophil

Diff for: tests/Fixture/Entity/EdgeCases/OneToManyWithUnionType/HasOneToManyWithUnionType.php

@@ -0,0 +1,29 @@
+<?php
+
+declare(strict_types=1);
+
+/*
+ * This file is part of the zenstruck/foundry package.
+ *
+ * (c) Kevin Bond <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Zenstruck\Foundry\Tests\Fixture\Entity\EdgeCases\OneToManyWithUnionType;
+
+use Doctrine\Common\Collections\Collection;
+use Doctrine\ORM\Mapping as ORM;
+use Zenstruck\Foundry\Tests\Fixture\Model\Base;
+
+#[ORM\Entity]
+class HasOneToManyWithUnionType extends Base
+{
+    public function __construct(
+        /** @var Collection<int,OwningSideEntity>|list<OwningSideEntity> */
+        #[ORM\OneToMany(targetEntity: OwningSideEntity::class, mappedBy: 'item')] // @phpstan-ignore doctrine.associationType, doctrine.associationType
+        public Collection|array $collection
+    ) {
+    }
+}

Diff for: tests/Fixture/Entity/EdgeCases/OneToManyWithUnionType/OwningSideEntity.php

@@ -0,0 +1,29 @@
+<?php
+
+declare(strict_types=1);
+
+/*
+ * This file is part of the zenstruck/foundry package.
+ *
+ * (c) Kevin Bond <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Zenstruck\Foundry\Tests\Fixture\Entity\EdgeCases\OneToManyWithUnionType;
+
+use Doctrine\ORM\Mapping as ORM;
+use Zenstruck\Foundry\Tests\Fixture\Model\Base;
+
+#[ORM\Entity]
+#[ORM\Table('union_type_owning_side_entity')]
+class OwningSideEntity extends Base
+{
+    public function __construct(
+        #[ORM\ManyToOne(inversedBy: 'collections')]
+        #[ORM\JoinColumn(nullable: false)]
+        public HasOneToManyWithUnionType $item,
+    ) {
+    }
+}

Diff for: tests/Integration/ORM/EdgeCasesRelationshipTest.php

@@ -13,6 +13,8 @@
 
 namespace Zenstruck\Foundry\Tests\Integration\ORM;
 
+use Doctrine\Common\Collections\ArrayCollection;
+use Doctrine\Common\Collections\Collection;
 use PHPUnit\Framework\Attributes\DataProvider;
 use PHPUnit\Framework\Attributes\RequiresPhpunit;
 use PHPUnit\Framework\Attributes\Test;
@@ -29,6 +31,7 @@
 use Zenstruck\Foundry\Tests\Fixture\Entity\EdgeCases\RichDomainMandatoryRelationship;
 use Zenstruck\Foundry\Tests\Fixture\Factories\Entity\EdgeCases\MultipleMandatoryRelationshipToSameEntity;
 use Zenstruck\Foundry\Tests\Integration\RequiresORM;
+use Zenstruck\Foundry\Tests\Fixture\Entity\EdgeCases\OneToManyWithUnionType;
 
 use function Zenstruck\Foundry\Persistence\persistent_factory;
 
@@ -158,6 +161,26 @@ public function indexed_one_to_many(): void
         self::assertNotNull($parent->getItems()->get('en')); // @phpstan-ignore argument.type
     }
 
+    /** @test */
+    #[Test]
+    public function object_with_union_type(): void
+    {
+        $owningSideFactory = persistent_factory(OneToManyWithUnionType\OwningSideEntity::class);
+        $hasOneToManyWithUnionTypeFactory = persistent_factory(OneToManyWithUnionType\HasOneToManyWithUnionType::class);
+
+        $object = $hasOneToManyWithUnionTypeFactory->create(
+            [
+                'collection' => $owningSideFactory->many(2),
+            ]
+        );
+
+        $owningSideFactory::assert()->count(2);
+        $hasOneToManyWithUnionTypeFactory::assert()->count(1);
+
+        self::assertCount(2, $object->collection);
+        self::assertInstanceOf(Collection::class, $object->collection);
+    }
+
     /**
      * @test
      */

Diff for: tests/Unit/Object/HydratorTest.php

@@ -14,6 +14,7 @@
 use Doctrine\Common\Collections\ArrayCollection;
 use Doctrine\Common\Collections\Collection;
 use Doctrine\Common\Collections\Selectable;
+use PHPUnit\Framework\Attributes\Test;
 use PHPUnit\Framework\TestCase;
 use Zenstruck\Foundry\Object\Hydrator;
 use Zenstruck\Foundry\Tests\Fixture\Object1;
@@ -26,6 +27,7 @@ class HydratorTest extends TestCase
     /**
      * @test
      */
+    #[Test]
     public function can_hydrate_scalar(): void
     {
         $value = 'Hello world';
@@ -42,6 +44,7 @@ public function can_hydrate_scalar(): void
     /**
      * @test
      */
+    #[Test]
     public function can_hydrate_scalar_array(): void
     {
         $value = ['foo', 'bar'];
@@ -58,6 +61,7 @@ public function can_hydrate_scalar_array(): void
     /**
      * @test
      */
+    #[Test]
     public function can_hydrate_object(): void
     {
         $object = new class {
@@ -79,6 +83,7 @@ public function __construct()
     /**
      * @test
      */
+    #[Test]
     public function can_hydrate_object_array(): void
     {
         $object = new class {
@@ -99,6 +104,7 @@ public function can_hydrate_object_array(): void
     /**
      * @test
      */
+    #[Test]
     public function can_hydrate_doctrine_collection(): void
     {
         $object = new class {
@@ -125,7 +131,8 @@ public function __construct()
     /**
      * @test
      */
-    public function can_hydrate_doctrine_collection_intersection(): void
+    #[Test]
+    public function can_hydrate_doctrine_collection_union(): void
     {
         $object = new class {
             /** @var Collection<array-key, Object1>|Selectable<array-key, Object1> */
@@ -151,7 +158,8 @@ public function __construct()
     /**
      * @test
      */
-    public function can_hydrate_doctrine_collection_union(): void
+    #[Test]
+    public function can_hydrate_doctrine_collection_intersection(): void
     {
         $object = new class {
             /** @var Collection<array-key, Object1>&Selectable<array-key, Object1> */