feat(Data): Add DataUnion attribute

Author: roxblnfk

src/Data/DataUnion.php

@@ -0,0 +1,33 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Testo\Data;
+
+use Testo\Data\Internal\DataProviderAttribute;
+use Testo\Data\Internal\DataProviderInterceptor;
+use Testo\Pipeline\Attribute\FallbackInterceptor;
+use Testo\Pipeline\Attribute\Interceptable;
+
+/**
+ * Concatenates multiple data providers into a single sequence.
+ *
+ * Primarily useful as a nested attribute inside {@see DataCross} or {@see DataZip}
+ * to combine multiple providers before crossing or zipping with others.
+ *
+ * @api
+ */
+#[\Attribute(\Attribute::TARGET_METHOD | \Attribute::TARGET_FUNCTION | \Attribute::IS_REPEATABLE)]
+#[FallbackInterceptor(DataProviderInterceptor::class)]
+final class DataUnion implements Interceptable, DataProviderAttribute
+{
+    /**
+     * @param array<DataProviderAttribute> $providers Data providers to concatenate.
+     */
+    public readonly array $providers;
+
+    public function __construct(DataProviderAttribute ...$providers)
+    {
+        $this->providers = $providers;
+    }
+}

src/Data/Internal/DataProviderInterceptor.php

@@ -12,6 +12,7 @@
 use Testo\Data\DataCross;
 use Testo\Data\DataProvider;
 use Testo\Data\DataSet;
+use Testo\Data\DataUnion;
 use Testo\Data\DataZip;
 use Testo\Data\MultipleResult;
 use Testo\Event\Test\TestBatchFinished;
@@ -171,6 +172,7 @@ private static function extractDataSets(TestInfo $info, object $attr): iterable
             $attr instanceof DataSet => [($attr->name ?? 0) => $attr->arguments],
             $attr instanceof DataCross => self::fromDataCross($info, $attr),
             $attr instanceof DataZip => self::fromDataZip($info, $attr),
+            $attr instanceof DataUnion => self::fromDataUnion($info, $attr),
             default => throw new \RuntimeException('Unknown Data Provider Attribute type.'),
         };
     }
@@ -192,7 +194,7 @@ private static function fromDataProvider(TestInfo $info, DataProvider $attribute
                 $m = $class->getMethod($provider);
                 $provider = match (true) {
                     $m->isStatic() => $m->getClosure(null),
-                    default => static fn () => $m->getClosure($info->caseInfo->instance->getInstance()),
+                    default => static fn() => $m->getClosure($info->caseInfo->instance->getInstance()),
                 };
             }
 
@@ -216,7 +218,7 @@ private static function fromDataProvider(TestInfo $info, DataProvider $attribute
     private static function fromDataZip(TestInfo $info, DataZip $attr): iterable
     {
         $generators = \array_map(
-            static fn ($providerAttr): DeferredGenerator => DeferredGenerator::fromHandler(
+            static fn($providerAttr): DeferredGenerator => DeferredGenerator::fromHandler(
                 static function () use ($info, $providerAttr) {
                     yield from self::extractDataSets($info, $providerAttr);
                 },
@@ -294,4 +296,14 @@ private static function cartesianProduct(
             );
         }
     }
+
+    /**
+     * Extract data sets from a DataUnion attribute (concatenation).
+     */
+    private static function fromDataUnion(TestInfo $info, DataUnion $attr): iterable
+    {
+        foreach ($attr->providers as $providerAttr) {
+            yield from self::extractDataSets($info, $providerAttr);
+        }
+    }
 }

tests/Data/Self/DataUnion.php

@@ -0,0 +1,90 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Tests\Data\Self;
+
+use Testo\Application\Attribute\Test;
+use Testo\Assert;
+use Testo\Data\DataProvider;
+use Testo\Data\DataSet;
+
+final class DataUnion
+{
+    public static function adminsProvider(): array
+    {
+        return [
+            'root' => ['root', true],
+            'admin' => ['admin', true],
+        ];
+    }
+
+    public static function usersProvider(): iterable
+    {
+        yield 'alice' => ['alice', false];
+        yield 'bob' => ['bob', false];
+    }
+
+    #[Test]
+    #[\Testo\Data\DataCross(
+        new \Testo\Data\DataUnion(
+            new DataProvider('adminsProvider'),
+            new DataProvider('usersProvider'),
+        ),
+        new DataSet(['read'], 'read'),
+    )]
+    public function crossWithUnion(string $user, bool $isAdmin, string $permission): void
+    {
+        // (2 admins + 2 users) × 1 permission = 4 combinations
+        Assert::same('read', $permission);
+        Assert::true(\in_array([$user, $isAdmin], [
+            ['root', true],
+            ['admin', true],
+            ['alice', false],
+            ['bob', false],
+        ], true));
+    }
+
+    #[Test]
+    #[\Testo\Data\DataZip(
+        new \Testo\Data\DataUnion(
+            new DataProvider('adminsProvider'),
+            new DataProvider('usersProvider'),
+        ),
+        new DataProvider('permissionsProvider'),
+    )]
+    public function zipWithUnion(string $user, bool $isAdmin, string $permission): void
+    {
+        // zip stops at shortest: 4 users | 3 permissions = 3 iterations
+        Assert::true(\in_array([$user, $isAdmin, $permission], [
+            ['root', true, 'read'],
+            ['admin', true, 'write'],
+            ['alice', false, 'delete'],
+        ], true));
+    }
+
+    public static function permissionsProvider(): array
+    {
+        return [
+            'read' => ['read'],
+            'write' => ['write'],
+            'delete' => ['delete'],
+        ];
+    }
+
+    #[Test]
+    #[\Testo\Data\DataUnion(
+        new DataProvider('adminsProvider'),
+        new DataProvider('usersProvider'),
+    )]
+    public function simpleUnion(string $user, bool $isAdmin): void
+    {
+        // 2 admins + 2 users = 4 iterations
+        Assert::true(\in_array([$user, $isAdmin], [
+            ['root', true],
+            ['admin', true],
+            ['alice', false],
+            ['bob', false],
+        ], true));
+    }
+}