1 files changed, 156 insertions, 0 deletions
diff --git a/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Utils/Runners/JobQueueRunner.php b/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Utils/Runners/JobQueueRunner.php
new file mode 100644
index 00000000..29e124db
--- /dev/null
+++ b/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Utils/Runners/JobQueueRunner.php
@@ -0,0 +1,156 @@
+<?php
+
+namespace SMW\Tests\Utils\Runners;
+
+use Job;
+use JobQueueGroup;
+use SMW\Connection\ConnectionProvider as IConnectionProvider;
+use SMW\Tests\TestEnvironment;
+use SMW\Tests\Utils\Connection\ConnectionProvider;
+
+/**
+ * Partly copied from the MW 1.19 RunJobs maintenance script
+ *
+ * @group SMW
+ * @group SMWExtension
+ *
+ * @license GNU GPL v2+
+ * @since 1.9.2
+ */
+class JobQueueRunner {
+
+	protected $type = null;
+	protected $status = [];
+	protected $connectionProvider = null;
+
+	/**
+	 * @var TestEnvironment
+	 */
+	private $testEnvironment;
+
+	/**
+	 * @since 1.9.2
+	 *
+	 * @param string|null $type
+	 * @param IConnectionProvider|null $connectionProvider
+	 */
+	public function __construct( $type = null, IConnectionProvider $connectionProvider = null ) {
+		$this->type = $type;
+		$this->connectionProvider = $connectionProvider;
+
+		if ( $this->connectionProvider === null ) {
+			$this->connectionProvider = new ConnectionProvider();
+		}
+
+		$this->testEnvironment = new TestEnvironment();
+	}
+
+	/**
+	 * @since 2.1
+	 *
+	 * @param string $type
+	 *
+	 * @return JobQueueRunner
+	 */
+	public function setType( $type ) {
+		$this->type = $type;
+		return $this;
+	}
+
+	/**
+	 * @since 2.1
+	 *
+	 * @param IConnectionProvider $connectionProvider
+	 *
+	 * @return JobQueueRunner
+	 */
+	public function setConnectionProvider( IConnectionProvider $connectionProvider ) {
+		$this->connectionProvider = $connectionProvider;
+		return $this;
+	}
+
+	/**
+	 * @since 1.9.2
+	 */
+	public function run() {
+
+		$conds = '';
+		$connection = $this->connectionProvider->getConnection();
+
+		if ( $this->type !== null ) {
+			$conds = "job_cmd = " . $connection->addQuotes( $this->type );
+		}
+
+		while ( $connection->selectField( 'job', 'job_id', $conds, __METHOD__ ) ) {
+
+			$job = $this->type === null ? $this->pop() : $this->pop_type( $this->type );
+
+			if ( !$job ) {
+				break;
+			}
+
+			wfWaitForSlaves();
+
+			$this->status[] = [
+				'type'   => $job->command,
+				'status' => $job->run()
+			];
+		}
+
+		$this->testEnvironment->executePendingDeferredUpdates();
+	}
+
+	/**
+	 * @since  2.0
+	 */
+	public function deleteAllJobs() {
+
+		$conditions = '*';
+		$connection = $this->connectionProvider->getConnection();
+
+		if ( $this->type !== null ) {
+			$conditions = "job_cmd = " . $connection->addQuotes( $this->type );
+		}
+
+		$connection->delete(
+			'job',
+			$conditions,
+			__METHOD__
+		);
+	}
+
+	/**
+	 * @since 1.9.2
+	 *
+	 * @return array
+	 */
+	public function getStatus() {
+		return $this->status;
+	}
+
+	/**
+	 * @see https://gerrit.wikimedia.org/r/#/c/162009/
+	 */
+	private function pop() {
+		$offset = 0;
+
+		if ( class_exists( 'JobQueueGroup' ) ) {
+			return JobQueueGroup::singleton()->pop();
+		}
+
+		return Job::pop( $offset );
+	}
+
+	/**
+	 * @see https://gerrit.wikimedia.org/r/#/c/162009/
+	 */
+	public function pop_type( $type ) {
+
+		if ( class_exists( 'JobQueueGroup' ) ) {
+			return JobQueueGroup::singleton()->get( $type )->pop();
+		}
+
+		return Job::pop_type( $type );
+	}
+
+}