Add withoutDefer and withDefer testing helpers (#53340)

Author: timacdonald

src/Illuminate/Foundation/Testing/Concerns/InteractsWithContainer.php

@@ -5,6 +5,7 @@
 use Closure;
 use Illuminate\Foundation\Mix;
 use Illuminate\Foundation\Vite;
+use Illuminate\Support\Defer\DeferredCallbackCollection;
 use Illuminate\Support\Facades\Facade;
 use Illuminate\Support\HtmlString;
 use Mockery;
@@ -25,6 +26,13 @@ trait InteractsWithContainer
      */
     protected $originalMix;
 
+    /**
+     * The original deferred callbacks collection.
+     *
+     * @var \Illuminate\Support\Defer\DeferredCallbackCollection|null
+     */
+    protected $originalDeferredCallbacksCollection;
+
     /**
      * Register an instance of an object in the container.
      *
@@ -234,4 +242,38 @@ protected function withMix()
 
         return $this;
     }
+
+    /**
+     * Execute deferred functions immediately.
+     *
+     * @return $this
+     */
+    protected function withoutDefer()
+    {
+        if ($this->originalDeferredCallbacksCollection == null) {
+            $this->originalDeferredCallbacksCollection = $this->app->make(DeferredCallbackCollection::class);
+        }
+
+        $this->swap(DeferredCallbackCollection::class, new class extends DeferredCallbackCollection
+        {
+            public function offsetSet(mixed $offset, mixed $value): void
+            {
+                $value();
+            }
+        });
+    }
+
+    /**
+     * Restore deferred functions.
+     *
+     * @return $this
+     */
+    protected function withDefer()
+    {
+        if ($this->originalDeferredCallbacksCollection) {
+            $this->app->instance(DeferredCallbackCollection::class, $this->originalDeferredCallbacksCollection);
+        }
+
+        return $this;
+    }
 }

tests/Foundation/Testing/Concerns/InteractsWithContainerTest.php

@@ -4,6 +4,7 @@
 
 use Illuminate\Foundation\Mix;
 use Illuminate\Foundation\Vite;
+use Illuminate\Support\Defer\DeferredCallbackCollection;
 use Orchestra\Testbench\TestCase;
 use stdClass;
 
@@ -73,6 +74,26 @@ public function testWithMixRestoresOriginalHandlerAndReturnsInstance()
         $this->assertSame($this, $instance);
     }
 
+    public function testWithoutDefer()
+    {
+        $called = [];
+        defer(function () use (&$called) {
+            $called[] = 1;
+        });
+        $this->assertSame([], $called);
+
+        $this->withoutDefer();
+        defer(function () use (&$called) {
+            $called[] = 2;
+        });
+        $this->assertSame([2], $called);
+
+        $this->withDefer();
+        $this->assertSame([2], $called);
+        $this->app[DeferredCallbackCollection::class]->invoke();
+        $this->assertSame([2, 1], $called);
+    }
+
     public function testForgetMock()
     {
         $this->mock(InstanceStub::class)