Ability to test chained job via closure (#49337)

laravel · Dec 12, 2023 · 5654bdf · 5654bdf
1 parent b41612c
commit 5654bdf
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/src/Illuminate/Support/Testing/Fakes/BusFake.php b/src/Illuminate/Support/Testing/Fakes/BusFake.php
@@ -421,6 +421,18 @@ protected function assertDispatchedWithChainOfObjects($command, $expectedChain,
                             ! $chain[$index]($chainedBatch->toPendingBatch())) {
                             return false;
                         }
+                    } elseif ($chain[$index] instanceof Closure) {
+                        [$expectedType, $callback] = [$this->firstClosureParameterType($chain[$index]), $chain[$index]];
+
+                        $chainedJob = unserialize($serializedChainedJob);
+
+                        if (! $chainedJob instanceof $expectedType) {
+                            throw new RuntimeException('The chained job was expected to be of type '.$expectedType.', '.$chainedJob::class.' chained.');
+                        }
+
+                        if (! $callback($chainedJob)) {
+                            return false;
+                        }
                     } elseif (is_string($chain[$index])) {
                         if ($chain[$index] != get_class(unserialize($serializedChainedJob))) {
                             return false;

diff --git a/tests/Support/SupportTestingBusFakeTest.php b/tests/Support/SupportTestingBusFakeTest.php
@@ -534,6 +534,16 @@ public function testAssertChained()
             ChainedJobStub::class,
         ]);
 
+        $this->fake->chain([
+            new ChainedJobStub(123),
+            new ChainedJobStub(456),
+        ])->dispatch();
+
+        $this->fake->assertChained([
+            fn (ChainedJobStub $job) => $job->id === 123,
+            fn (ChainedJobStub $job) => $job->id === 456,
+        ]);
+
         Container::setInstance(null);
     }
 
@@ -777,6 +787,13 @@ class BusJobStub
 class ChainedJobStub
 {
     use Queueable;
+
+    public $id;
+
+    public function __construct($id = null)
+    {
+        $this->id = $id;
+    }
 }
 
 class OtherBusJobStub