fix: edge case with Doctrine Middleware & early kernel boot (#993)

Author: HypeMC

src/ZenstruckFoundryBundle.php

@@ -37,7 +37,7 @@ final class ZenstruckFoundryBundle extends AbstractBundle implements CompilerPas
 {
     public function boot(): void
     {
-        if ($this->container && !Configuration::isBooted()) {
+        if ($this->container) {
             Configuration::boot($this->container->get('.zenstruck_foundry.configuration')); // @phpstan-ignore argument.type
         }
     }

tests/Fixture/ResetDatabase/DoctrineMiddleware.php

@@ -0,0 +1,31 @@
+<?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\ResetDatabase;
+
+use Doctrine\DBAL\Driver;
+use Doctrine\DBAL\Driver\Middleware;
+use Symfony\Component\HttpKernel\KernelInterface;
+
+final class DoctrineMiddleware implements Middleware
+{
+    public function __construct(
+        private KernelInterface $kernel, // @phpstan-ignore property.onlyWritten
+    ) {
+    }
+
+    public function wrap(Driver $driver): Driver
+    {
+        return $driver;
+    }
+}

tests/Fixture/ResetDatabase/ResetDatabaseTestKernel.php

@@ -64,6 +64,10 @@ protected function configureContainer(ContainerBuilder $c, LoaderInterface $load
 
         $c->register(GlobalInvokableService::class);
 
+        $c->register(DoctrineMiddleware::class)
+            ->setAutowired(true)
+            ->setAutoconfigured(true);
+
         if (self::hasORM()) {
             $c->register(OrmResetterDecorator::class)->setAutowired(true)->setAutoconfigured(true);
         }

tests/Integration/ResetDatabase/EarlyBootedKernelTest.php

@@ -0,0 +1,50 @@
+<?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\Integration\ResetDatabase;
+
+use Doctrine\DBAL\Connection;
+use Doctrine\DBAL\Driver\Middleware;
+use PHPUnit\Framework\Attributes\BeforeClass;
+use PHPUnit\Framework\Attributes\RequiresPhpunit;
+use PHPUnit\Framework\Attributes\Test;
+use Zenstruck\Foundry\Test\ResetDatabase;
+use Zenstruck\Foundry\Tests\Fixture\ResetDatabase\DoctrineMiddleware;
+
+/**
+ * @requires PHPUnit >=11.3
+ */
+#[RequiresPhpunit('>=11.3')]
+final class EarlyBootedKernelTest extends ResetDatabaseTestCase
+{
+    /**
+     * Needs to happen before {@see ResetDatabase::_resetDatabaseBeforeFirstTest()}.
+     */
+    #[BeforeClass(10)]
+    public static function before(): void
+    {
+        self::bootKernel();
+    }
+
+    #[Test]
+    public function connection_uses_doctrine_middleware(): void
+    {
+        /** @var Connection $connection */
+        $connection = self::getContainer()->get(Connection::class);
+
+        self::assertContains(
+            DoctrineMiddleware::class,
+            \array_map(static fn(Middleware $middleware) => $middleware::class, $connection->getConfiguration()->getMiddlewares())
+        );
+    }
+}