1 files changed, 144 insertions, 0 deletions
diff --git a/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Utils/ByPageSemanticDataFinder.php b/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Utils/ByPageSemanticDataFinder.php
new file mode 100644
index 00000000..61bb4225
--- /dev/null
+++ b/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Utils/ByPageSemanticDataFinder.php
@@ -0,0 +1,144 @@
+<?php
+
+namespace SMW\Tests\Utils;
+
+use ParserOutput;
+use SMW\ParserData;
+use SMW\SemanticData;
+use SMW\Store;
+use Title;
+use UnexpectedValueException;
+use User;
+use WikiPage;
+
+/**
+ *
+ * @group SMW
+ * @group SMWExtension
+ *
+ * @licence GNU GPL v2+
+ * @since 1.9.1
+ *
+ * @author mwjames
+ */
+class ByPageSemanticDataFinder {
+
+	protected $store = null;
+	protected $title = null;
+
+	/**
+	 * @since 1.9.1
+	 *
+	 * @param Store $store
+	 *
+	 * @return PageDataFetcher
+	 */
+	public function setStore( Store $store ) {
+		$this->store = $store;
+		return $this;
+	}
+
+	/**
+	 * @since 1.9.1
+	 *
+	 * @param Title $title
+	 *
+	 * @return PageDataFetcher
+	 */
+	public function setTitle( Title $title ) {
+		$this->title = $title;
+		return $this;
+	}
+
+	/**
+	 * @since 1.9.2
+	 *
+	 * @return SemanticData
+	 */
+	public function fetchIncomingDataFromStore() {
+
+		$requestOptions = new \SMWRequestOptions();
+		$requestOptions->sort = true;
+
+		$subject = $this->getPageData()->getSubject();
+		$semanticData = new SemanticData( $subject );
+
+		$incomingProperties = $this->getStore()->getInProperties( $subject, $requestOptions );
+
+		foreach ( $incomingProperties as $property ) {
+			$values = $this->getStore()->getPropertySubjects( $property, null );
+
+			foreach ( $values as $value ) {
+				$semanticData->addPropertyObjectValue( $property, $value );
+			}
+		}
+
+		return $semanticData;
+	}
+
+	/**
+	 * @since 1.9.1
+	 *
+	 * @return SemanticData
+	 */
+	public function fetchFromStore() {
+		return $this->getStore()->getSemanticData( $this->getPageData()->getSubject() );
+	}
+
+	/**
+	 * @since 1.9.1
+	 *
+	 * @return SemanticData
+	 */
+	public function fetchFromOutput() {
+		return $this->getPageData()->getData();
+	}
+
+	protected function getPageData() {
+		return new ParserData( $this->getTitle(), $this->makeOutputFromPageRevision() );
+	}
+
+	protected function getPage() {
+		return WikiPage::factory( $this->getTitle() );
+	}
+
+	protected function makeOutputFromPageRevision() {
+
+		$wikiPage = $this->getPage();
+		$revision = $wikiPage->getRevision();
+
+		if ( $revision === null ) {
+			throw new UnexpectedValueException( 'Expected a valid Revision' );
+		}
+
+		$parserOutput = $wikiPage->getParserOutput(
+			$wikiPage->makeParserOptions( User::newFromId( $revision->getUser() ) ),
+			$revision->getId()
+		);
+
+		if ( $parserOutput instanceof ParserOutput ) {
+			return $parserOutput;
+		}
+
+		throw new UnexpectedValueException( 'Expected a ParserOutput object' );
+	}
+
+	protected function getTitle() {
+
+		if ( $this->title instanceof Title ) {
+			return $this->title;
+		}
+
+		throw new UnexpectedValueException( 'Expected a Title object' );
+	}
+
+	protected function getStore() {
+
+		if ( $this->store instanceof Store ) {
+			return $this->store;
+		}
+
+		throw new UnexpectedValueException( 'Expected a Store object' );
+	}
+
+}