1 files changed, 91 insertions, 0 deletions
diff --git a/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Utils/Validators/ExportDataValidator.php b/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Utils/Validators/ExportDataValidator.php
new file mode 100644
index 00000000..fc89a262
--- /dev/null
+++ b/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Utils/Validators/ExportDataValidator.php
@@ -0,0 +1,91 @@
+<?php
+
+namespace SMW\Tests\Utils\Validators;
+
+use SMWExpData as ExpData;
+use SMWExpResource as ExpResource;
+
+/**
+ * @license GNU GPL v2+
+ * @since   2.0
+ *
+ * @author mwjames
+ */
+class ExportDataValidator extends \PHPUnit_Framework_Assert {
+
+	/**
+	 * @since 2.0
+	 *
+	 * @param mixed $expected
+	 * @param ExpData $exportData
+	 */
+	public function assertThatExportDataContainsProperty( $expectedProperties, ExpData $exportData ) {
+
+		$expProperties = $exportData->getProperties();
+
+		$this->assertNotEmpty( $expProperties );
+
+		$expectedProperties = is_array( $expectedProperties ) ? $expectedProperties : [ $expectedProperties ];
+		$expectedToCount  = count( $expectedProperties );
+		$actualComparedToCount = 0;
+
+		$assertThatExportDataContainsProperty = false;
+
+		foreach ( $expProperties as $expProperty ) {
+			foreach ( $expectedProperties as $expectedProperty ) {
+				if ( $expectedProperty->getHash() === $expProperty->getHash() ) {
+					$actualComparedToCount++;
+					$assertThatExportDataContainsProperty = true;
+				}
+			}
+		}
+
+		$this->assertTrue( $assertThatExportDataContainsProperty );
+
+		$this->assertEquals(
+			$expectedToCount,
+			$actualComparedToCount
+		);
+	}
+
+	/**
+	 * @since 2.0
+	 *
+	 * @param mixed $expected
+	 * @param ExpResource $selectedElement
+	 * @param ExpData $exportData
+	 */
+	public function assertThatExportDataContainsResource( $expectedResources, ExpResource $selectedElement, ExpData $exportData ) {
+
+		$expElements = $exportData->getValues( $selectedElement );
+
+		$this->assertNotEmpty( $expElements );
+
+		$expectedResources = is_array( $expectedResources ) ? $expectedResources : [ $expectedResources ];
+		$expectedToCount  = count( $expectedResources );
+		$actualComparedToCount = 0;
+
+		$assertThatExportDataContainsResource = false;
+
+		foreach ( $expElements as $expElement ) {
+			foreach ( $expectedResources as $expectedResource ) {
+				if ( $expectedResource->getHash() === $expElement->getHash() ) {
+					$actualComparedToCount++;
+					$assertThatExportDataContainsResource = true;
+				}
+			}
+		}
+
+		$this->assertTrue(
+			$assertThatExportDataContainsResource,
+			'Asserts that a resource is set'
+		);
+
+		$this->assertEquals(
+			$expectedToCount,
+			$actualComparedToCount,
+			'Asserts that all listed resources are set'
+		);
+	}
+
+}