Add Conditionable to Batched and Chained jobs (#49310)

* Add Conditionable to Batched and Chained jobs

* styleci

* styleci

---------

Co-authored-by: Brett Bailey <[email protected]>

Author: bretto36

src/Illuminate/Bus/PendingBatch.php

@@ -8,11 +8,14 @@
 use Illuminate\Contracts\Events\Dispatcher as EventDispatcher;
 use Illuminate\Support\Arr;
 use Illuminate\Support\Collection;
+use Illuminate\Support\Traits\Conditionable;
 use Laravel\SerializableClosure\SerializableClosure;
 use Throwable;
 
 class PendingBatch
 {
+    use Conditionable;
+
     /**
      * The IoC container instance.
      *

src/Illuminate/Foundation/Bus/PendingChain.php

@@ -5,10 +5,13 @@
 use Closure;
 use Illuminate\Contracts\Bus\Dispatcher;
 use Illuminate\Queue\CallQueuedClosure;
+use Illuminate\Support\Traits\Conditionable;
 use Laravel\SerializableClosure\SerializableClosure;
 
 class PendingChain
 {
+    use Conditionable;
+
     /**
      * The class name of the job being dispatched.
      *

tests/Integration/Queue/JobChainingTest.php

@@ -3,9 +3,11 @@
 namespace Illuminate\Tests\Integration\Queue;
 
 use Illuminate\Bus\Batchable;
+use Illuminate\Bus\PendingBatch;
 use Illuminate\Bus\Queueable;
 use Illuminate\Contracts\Queue\ShouldQueue;
 use Illuminate\Foundation\Bus\Dispatchable;
+use Illuminate\Foundation\Bus\PendingChain;
 use Illuminate\Foundation\Testing\DatabaseMigrations;
 use Illuminate\Queue\InteractsWithQueue;
 use Illuminate\Support\Carbon;
@@ -387,6 +389,43 @@ public function testChainBatchFailureNotAllowed()
         $this->assertEquals(['c1', 'c2', 'b1', 'b3'], JobRunRecorder::$results);
         $this->assertEquals(['batch failed', 'chain failed'], JobRunRecorder::$failures);
     }
+
+    public function testChainConditionable()
+    {
+        $chain = Bus::chain([])
+            ->onConnection('sync1')
+            ->when(true, function (PendingChain $chain) {
+                $chain->onConnection('sync2');
+            });
+
+        $this->assertEquals('sync2', $chain->connection);
+
+        $chain = Bus::chain([])
+            ->onConnection('sync1')
+            ->when(false, function (PendingChain $chain) {
+                $chain->onConnection('sync2');
+            });
+
+        $this->assertEquals('sync1', $chain->connection);
+    }
+
+    public function testBatchConditionable()
+    {
+        $batch = Bus::batch([])
+            ->onConnection('sync1')
+            ->when(true, function (PendingBatch $batch) {
+                $batch->onConnection('sync2');
+            });
+
+        $this->assertEquals('sync2', $batch->connection());
+        $batch = Bus::batch([])
+            ->onConnection('sync1')
+            ->when(false, function (PendingBatch $batch) {
+                $batch->onConnection('sync2');
+            });
+
+        $this->assertEquals('sync1', $batch->connection());
+    }
 }
 
 class JobChainingTestFirstJob implements ShouldQueue