1 files changed, 39 insertions, 18 deletions
diff --git a/tests/phpunit/includes/api/generateRandomImages.php b/tests/phpunit/includes/api/generateRandomImages.php
index aca52acbed42..f3a14e5bd1a7 100644
--- a/tests/phpunit/includes/api/generateRandomImages.php
+++ b/tests/phpunit/includes/api/generateRandomImages.php
@@ -1,26 +1,47 @@
 <?php
+/**
+ * Bootstrapping for test image file generation
+ *
+ * @file
+ */
 
+// Evaluate the include path relative to this file
+$IP = dirname( dirname( dirname( dirname( dirname( __FILE__ ) ) ) ) );
+
+// Start up MediaWiki in command-line mode
+require_once( "$IP/maintenance/Maintenance.php" );
 require("RandomImageGenerator.php");
 
-$getOptSpec = array(
-	'dictionaryFile::',
-	'minWidth::',
-	'maxWidth::',
-	'minHeight::',
-	'maxHeight::',
-	'shapesToDraw::',
-	'shape::',
+class GenerateRandomImages extends Maintenance {
+
+	public function execute() {
+
+		$getOptSpec = array(
+			'dictionaryFile::',
+			'minWidth::',
+			'maxWidth::',
+			'minHeight::',
+			'maxHeight::',
+			'shapesToDraw::',
+			'shape::',
+
+			'number::',
+			'format::'
+		);
+		$options = getopt( null, $getOptSpec );
+
+		$format = isset( $options['format'] ) ? $options['format'] : 'jpg';
+		unset( $options['format'] );
+
+		$number = isset( $options['number'] ) ? intval( $options['number'] ) : 10;
+		unset( $options['number'] );
 
-	'number::',
-	'format::'
-);
-$options = getopt( null, $getOptSpec );
+		$randomImageGenerator = new RandomImageGenerator( $options );
+		$randomImageGenerator->writeImages( $number, $format );
+	}
+}
 
-$format = isset( $options['format'] ) ? $options['format'] : 'jpg';
-unset( $options['format'] );
+$maintClass = 'GenerateRandomImages';
+require( RUN_MAINTENANCE_IF_MAIN );
 
-$number = isset( $options['number'] ) ? intval( $options['number'] ) : 10;
-unset( $options['number'] );
 
-$randomImageGenerator = new RandomImageGenerator( $options );
-$randomImageGenerator->writeImages( $number, $format );