1 files changed, 92 insertions, 0 deletions
diff --git a/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Utils/PageRefresher.php b/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Utils/PageRefresher.php
new file mode 100644
index 00000000..05c74776
--- /dev/null
+++ b/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Utils/PageRefresher.php
@@ -0,0 +1,92 @@
+<?php
+
+namespace SMW\Tests\Utils;
+
+use RuntimeException;
+use SMW\ApplicationFactory;
+use SMW\ContentParser;
+use SMW\DIWikiPage;
+use SMW\MediaWiki\Jobs\UpdateJob;
+use Title;
+use WikiPage;
+
+/**
+ *
+ * @group SMW
+ * @group SMWExtension
+ *
+ * @license GNU GPL v2+
+ * @since 2.0
+ */
+class PageRefresher {
+
+	/**
+	 * @since 2.0
+	 *
+	 * @param mixed $title
+	 *
+	 * @return PageRefresher
+	 */
+	public function doRefresh( $title ) {
+
+		if ( $title instanceof WikiPage || $title instanceof DIWikiPage ) {
+			$title = $title->getTitle();
+		}
+
+		if ( !$title instanceof Title ) {
+			throw new RuntimeException( 'Expected a title instance' );
+		}
+
+		$contentParser = new ContentParser( $title );
+
+		$parserData = ApplicationFactory::getInstance()->newParserData(
+			$title,
+			$contentParser->parse()->getOutput()
+		);
+
+		$parserData->updateStore();
+
+		return $this;
+	}
+
+	/**
+	 * @since 2.1
+	 *
+	 * @param array $pages
+	 *
+	 * @return PageRefresher
+	 */
+	public function doRefreshPoolOfPages( array $pages ) {
+		foreach ( $pages as $page ) {
+			$this->doRefreshByUpdateJob( $page );
+		}
+	}
+
+	/**
+	 * @since 2.1
+	 *
+	 * @param mixed $title
+	 *
+	 * @return PageRefresher
+	 */
+	public function doRefreshByUpdateJob( $title ) {
+
+		if ( $title instanceof WikiPage || $title instanceof DIWikiPage ) {
+			$title = $title->getTitle();
+		}
+
+		if ( is_string( $title ) ) {
+			$title = Title::newFromText( $title );
+		}
+
+		if ( !$title instanceof Title ) {
+			throw new RuntimeException( 'Expected a title instance' );
+		}
+
+		$job = new UpdateJob( $title );
+		$job->run();
+
+		return $this;
+	}
+
+}