1 files changed, 101 insertions, 0 deletions
diff --git a/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Unit/Query/ProfileAnnotators/SchemaLinkProfileAnnotatorTest.php b/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Unit/Query/ProfileAnnotators/SchemaLinkProfileAnnotatorTest.php
new file mode 100644
index 00000000..89150c66
--- /dev/null
+++ b/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Unit/Query/ProfileAnnotators/SchemaLinkProfileAnnotatorTest.php
@@ -0,0 +1,101 @@
+<?php
+
+namespace SMW\Tests\Query\ProfileAnnotators;
+
+use SMW\DIWikiPage;
+use SMW\Query\ProfileAnnotators\NullProfileAnnotator;
+use SMW\Query\ProfileAnnotators\SchemaLinkProfileAnnotator;
+use SMW\Tests\TestEnvironment;
+use SMWContainerSemanticData as ContainerSemanticData;
+use SMWDIContainer as DIContainer;
+use SMW\Tests\PHPUnitCompat;
+
+/**
+ * @covers \SMW\Query\ProfileAnnotators\SchemaLinkProfileAnnotator
+ * @group semantic-mediawiki
+ *
+ * @license GNU GPL v2+
+ * @since 3.0
+ *
+ * @author mwjames
+ */
+class SchemaLinkProfileAnnotatorTest extends \PHPUnit_Framework_TestCase {
+
+	use PHPUnitCompat;
+
+	private $semanticDataValidator;
+
+	protected function setUp() {
+		parent::setUp();
+
+		$this->semanticDataValidator = TestEnvironment::newValidatorFactory()->newSemanticDataValidator();
+	}
+
+	public function testCanConstruct() {
+
+		$profileAnnotator = $this->getMockBuilder( '\SMW\Query\ProfileAnnotator' )
+			->disableOriginalConstructor()
+			->getMock();
+
+		$this->assertInstanceOf(
+			SchemaLinkProfileAnnotator::class,
+			new SchemaLinkProfileAnnotator( $profileAnnotator, '' )
+		);
+	}
+
+	public function testAddAnnotationOnInvalidSchemaLinkTypeThrowsException() {
+
+		$profileAnnotator = $this->getMockBuilder( '\SMW\Query\ProfileAnnotator' )
+			->disableOriginalConstructor()
+			->getMock();
+
+		$instance = new SchemaLinkProfileAnnotator( $profileAnnotator, [] );
+
+		$this->setExpectedException( '\RuntimeException' );
+		$instance->addAnnotation();
+	}
+
+	/**
+	 * @dataProvider SchemaLinkProvider
+	 */
+	public function testAddAnnotation( $SchemaLink, $expected ) {
+
+		$subject = new DIWikiPage( __METHOD__, NS_MAIN, '', '_QUERYe7d20a88999' );
+
+		$container = new DIContainer(
+			new ContainerSemanticData( $subject	)
+		);
+
+		$instance = new SchemaLinkProfileAnnotator(
+			new NullProfileAnnotator( $container ),
+			$SchemaLink
+		);
+
+		$instance->addAnnotation();
+
+		$this->semanticDataValidator->assertThatPropertiesAreSet(
+			$expected,
+			$instance->getSemanticData()
+		);
+	}
+
+	public function SchemaLinkProvider() {
+
+		yield [
+			'',
+			[
+				'propertyCount' => 0
+			]
+		];
+
+		yield [
+			'Foo',
+			[
+				'propertyCount'  => 1,
+				'propertyKeys'   => [ '_SCHEMA_LINK' ],
+				'propertyValues' => [ 'Foo' ]
+			]
+		];
+	}
+
+}