Test the finished callback of batch runner. Refs drush-ops#5009

weitzman · May 8, 2022 · dc83ea1 · dc83ea1
1 parent a4e57fe
commit dc83ea1
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/tests/functional/BatchTest.php b/tests/functional/BatchTest.php
@@ -16,6 +16,8 @@ public function testBatch()
             'include' => __DIR__,
         ];
         $this->drush('unit-batch', [], $options);
-        $this->assertStringContainsString('!!! ArrayObject does its job.', $this->getErrorOutput());
+        $error_output = $this->getErrorOutput();
+        $this->assertStringContainsString('!!! ArrayObject does its job.', $error_output);
+        $this->assertStringContainsString('Result count is 5', $error_output);
     }
 }
diff --git a/tests/unish/Batch/UnitBatchOperations.php b/tests/unish/Batch/UnitBatchOperations.php
@@ -12,13 +12,17 @@ public static function operate(&$context)
 
         for ($i = 0; $i < 5; $i++) {
             Drush::logger()->info("Iteration $i");
+            // Store some results for post-processing in the 'finished' callback.
+            // The contents of 'results' will be available as $results in the
+            // 'finished' function.
+            $context['results'][] = $i;
+
         }
         $context['finished'] = 1;
     }
 
-    public static function finish()
+    public static function finish($success, $results, $operations)
     {
-        // Restore php limits.
-        // TODO.
+        Drush::logger()->success("Result count is " . count($results));
     }
 }