feat: "in memory" behavior

config/in_memory.php

@@ -0,0 +1,18 @@
+<?php
+
+namespace Symfony\Component\DependencyInjection\Loader\Configurator;
+
+use Zenstruck\Foundry\InMemory\InMemoryFactoryRegistry;
+use Zenstruck\Foundry\InMemory\InMemoryRepositoryRegistry;
+
+return static function (ContainerConfigurator $container): void {
+    $container->services()
+        ->set('.zenstruck_foundry.in_memory.factory_registry', InMemoryFactoryRegistry::class)
+        ->decorate('.zenstruck_foundry.factory_registry')
+        ->arg('$decorated', service('.inner'));
+
+    $container->services()
+        ->set('.zenstruck_foundry.in_memory.repository_registry', InMemoryRepositoryRegistry::class)
+        ->arg('$inMemoryRepositories', abstract_arg('inMemoryRepositories'))
+    ;
+};

config/services.php

@@ -33,6 +33,7 @@
             service('.zenstruck_foundry.story_registry'),
             service('.zenstruck_foundry.persistence_manager')->nullOnInvalid(),
             service('event_dispatcher'),
+            service('.zenstruck_foundry.in_memory.repository_registry'),
         ])
         ->public()
     ;

src/Configuration.php

@@ -17,6 +17,8 @@
 use Zenstruck\Foundry\Exception\FoundryNotBooted;
 use Zenstruck\Foundry\Exception\PersistenceDisabled;
 use Zenstruck\Foundry\Exception\PersistenceNotAvailable;
+use Zenstruck\Foundry\InMemory\CannotEnableInMemory;
+use Zenstruck\Foundry\InMemory\InMemoryRepositoryRegistry;
 use Zenstruck\Foundry\Persistence\PersistenceManager;
 
 /**
@@ -43,16 +45,19 @@ final class Configuration
     /** @var \Closure():self|self|null */
     private static \Closure|self|null $instance = null;
 
+    private bool $inMemory = false;
+
     /**
      * @phpstan-param InstantiatorCallable $instantiator
      */
     public function __construct(
-        public readonly FactoryRegistry $factories,
+        public readonly FactoryRegistryInterface $factories,
         public readonly Faker\Generator $faker,
         callable $instantiator,
         public readonly StoryRegistry $stories,
         private readonly ?PersistenceManager $persistence = null,
         private readonly ?EventDispatcherInterface $eventDispatcher = null,
+        public readonly ?InMemoryRepositoryRegistry $inMemoryRepositoryRegistry = null,
     ) {
         $this->instantiator = $instantiator;
     }
@@ -131,4 +136,21 @@ public static function shutdown(): void
         StoryRegistry::reset();
         self::$instance = null;
     }
+
+    /**
+     * @throws CannotEnableInMemory
+     */
+    public function enableInMemory(): void
+    {
+        if (null === $this->inMemoryRepositoryRegistry) {
+            throw CannotEnableInMemory::noInMemoryRepositoryRegistry();
+        }
+
+        $this->inMemory = true;
+    }
+
+    public function isInMemoryEnabled(): bool
+    {
+        return $this->inMemory;
+    }
 }

src/Exception/CannotCreateFactory.php

@@ -0,0 +1,26 @@
+<?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\Exception;
+
+/**
+ * @author Nicolas PHILIPPE <[email protected]>
+ * @internal
+ */
+final class CannotCreateFactory extends \LogicException
+{
+    public static function argumentCountError(\ArgumentCountError $e): static
+    {
+        return new self('Factories with dependencies (services) cannot be created before foundry is booted.', previous: $e);
+    }
+}

src/Factory.php

@@ -12,6 +12,7 @@
 namespace Zenstruck\Foundry;
 
 use Faker;
+use Zenstruck\Foundry\Exception\CannotCreateFactory;
 
 /**
  * @author Kevin Bond <[email protected]>
@@ -53,7 +54,7 @@ final public static function new(array|callable $attributes = []): static
         try {
             $factory ??= new static(); // @phpstan-ignore new.static
         } catch (\ArgumentCountError $e) {
-            throw new \LogicException('Factories with dependencies (services) cannot be created before foundry is booted.', previous: $e);
+            throw CannotCreateFactory::argumentCountError($e);
         }
 
         return $factory

src/FactoryRegistry.php

@@ -11,12 +11,14 @@
 
 namespace Zenstruck\Foundry;
 
+use Zenstruck\Foundry\Exception\CannotCreateFactory;
+
 /**
  * @author Kevin Bond <[email protected]>
  *
  * @internal
  */
-final class FactoryRegistry
+final class FactoryRegistry implements FactoryRegistryInterface
 {
     /**
      * @param Factory<mixed>[] $factories
@@ -25,21 +27,18 @@ public function __construct(private iterable $factories)
     {
     }
 
-    /**
-     * @template T of Factory
-     *
-     * @param class-string<T> $class
-     *
-     * @return T|null
-     */
-    public function get(string $class): ?Factory
+    public function get(string $class): Factory
     {
         foreach ($this->factories as $factory) {
             if ($class === $factory::class) {
                 return $factory; // @phpstan-ignore return.type
             }
         }
 
-        return null;
+        try {
+            return new $class();
+        } catch (\ArgumentCountError $e) {
+            throw CannotCreateFactory::argumentCountError($e);
+        }
     }
 }

src/FactoryRegistryInterface.php

@@ -0,0 +1,29 @@
+<?php
+
+/*
+ * 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;
+
+/**
+ * @author Nicolas PHILIPPE <[email protected]>
+ *
+ * @internal
+ */
+interface FactoryRegistryInterface
+{
+    /**
+     * @template T of Factory
+     *
+     * @param class-string<T> $class
+     *
+     * @return T
+     */
+    public function get(string $class): Factory;
+}

src/InMemory/AsInMemoryTest.php

@@ -0,0 +1,36 @@
+<?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\InMemory;
+
+/**
+ * @author Nicolas PHILIPPE <[email protected]>
+ */
+#[\Attribute(\Attribute::TARGET_CLASS | \Attribute::TARGET_METHOD)]
+final class AsInMemoryTest
+{
+    /**
+     * @param class-string $class
+     * @internal
+     */
+    public static function shouldEnableInMemory(string $class, string $method): bool
+    {
+        $classReflection = new \ReflectionClass($class);
+
+        if ($classReflection->getAttributes(static::class)) {
+            return true;
+        }
+
+        return (bool) $classReflection->getMethod($method)->getAttributes(static::class);
+    }
+}

src/InMemory/CannotEnableInMemory.php

@@ -0,0 +1,27 @@
+<?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\InMemory;
+
+final class CannotEnableInMemory extends \LogicException
+{
+    public static function testIsNotAKernelTestCase(string $testName): self
+    {
+        return new self("{$testName}: Cannot use the #[AsInMemoryTest] attribute without extending KernelTestCase.");
+    }
+
+    public static function noInMemoryRepositoryRegistry(): self
+    {
+        return new self('Cannot enable "in memory": maybe not in a KernelTestCase?');
+    }
+}

src/InMemory/DependencyInjection/InMemoryCompilerPass.php

@@ -0,0 +1,55 @@
+<?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\InMemory\DependencyInjection;
+
+use Symfony\Component\DependencyInjection\Compiler\CompilerPassInterface;
+use Symfony\Component\DependencyInjection\Compiler\ServiceLocatorTagPass;
+use Symfony\Component\DependencyInjection\ContainerBuilder;
+use Symfony\Component\DependencyInjection\Reference;
+use Zenstruck\Foundry\InMemory\InMemoryRepository;
+
+/**
+ * @internal
+ * @author Nicolas PHILIPPE <[email protected]>
+ */
+final class InMemoryCompilerPass implements CompilerPassInterface
+{
+    public function process(ContainerBuilder $container): void
+    {
+        // create a service locator with all "in memory" repositories, indexed by target class
+        $inMemoryRepositoriesServices = $container->findTaggedServiceIds('foundry.in_memory.repository');
+        $inMemoryRepositoriesLocator = ServiceLocatorTagPass::register(
+            $container,
+            \array_combine(
+                \array_map(
+                    static function(string $serviceId) use ($container) {
+                        /** @var class-string<InMemoryRepository<object>> $inMemoryRepositoryClass */
+                        $inMemoryRepositoryClass = $container->getDefinition($serviceId)->getClass() ?? throw new \LogicException("Service \"{$serviceId}\" should have a class.");
+
+                        return $inMemoryRepositoryClass::_class();
+                    },
+                    \array_keys($inMemoryRepositoriesServices)
+                ),
+                \array_map(
+                    static fn(string $inMemoryRepositoryId) => new Reference($inMemoryRepositoryId),
+                    \array_keys($inMemoryRepositoriesServices)
+                ),
+            )
+        );
+
+        $container->findDefinition('.zenstruck_foundry.in_memory.repository_registry')
+            ->setArgument('$inMemoryRepositories', $inMemoryRepositoriesLocator)
+        ;
+    }
+}

src/InMemory/GenericInMemoryRepository.php

@@ -0,0 +1,61 @@
+<?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\InMemory;
+
+/**
+ * @template T of object
+ * @implements InMemoryRepository<T>
+ * @author Nicolas PHILIPPE <[email protected]>
+ *
+ * This class will be used when a specific "in-memory" repository does not exist for a given class.
+ */
+final class GenericInMemoryRepository implements InMemoryRepository
+{
+    /**
+     * @var list<T>
+     */
+    private array $elements = [];
+
+    /**
+     * @param class-string<T> $class
+     */
+    public function __construct(
+        private readonly string $class,
+    ) {
+    }
+
+    /**
+     * @param T $item
+     */
+    public function _save(object $item): void
+    {
+        if (!$item instanceof $this->class) {
+            throw new \InvalidArgumentException(\sprintf('Given object of class "%s" is not an instance of expected "%s"', $item::class, $this->class));
+        }
+
+        if (!\in_array($item, $this->elements, true)) {
+            $this->elements[] = $item;
+        }
+    }
+
+    public function _all(): array
+    {
+        return $this->elements;
+    }
+
+    public static function _class(): string
+    {
+        throw new \BadMethodCallException('This method should not be called on a GenericInMemoryRepository.');
+    }
+}