1 files changed, 96 insertions, 0 deletions
diff --git a/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Unit/Connection/ConnectionManagerTest.php b/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Unit/Connection/ConnectionManagerTest.php
new file mode 100644
index 00000000..353032d7
--- /dev/null
+++ b/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Unit/Connection/ConnectionManagerTest.php
@@ -0,0 +1,96 @@
+<?php
+
+namespace SMW\Tests\Connection;
+
+use SMW\Connection\ConnectionManager;
+use SMW\Tests\PHPUnitCompat;
+
+/**
+ * @covers \SMW\Connection\ConnectionManager
+ * @group semantic-mediawiki
+ *
+ * @license GNU GPL v2+
+ * @since 2.1
+ *
+ * @author mwjames
+ */
+class ConnectionManagerTest extends \PHPUnit_Framework_TestCase {
+
+	use PHPUnitCompat;
+
+	public function testCanConstruct() {
+
+		$this->assertInstanceOf(
+			ConnectionManager::class,
+			new ConnectionManager()
+		);
+	}
+
+	public function testDefaultRegisteredConnectionProvided() {
+
+		$instance = new ConnectionManager();
+		$instance->releaseConnections();
+
+		$connection = $instance->getConnection( 'mw.db' );
+
+		$this->assertSame(
+			$connection,
+			$instance->getConnection( 'mw.db' )
+		);
+
+		$this->assertInstanceOf(
+			'\SMW\MediaWiki\Database',
+			$connection
+		);
+
+		$instance->releaseConnections();
+
+		$this->assertNotSame(
+			$connection,
+			$instance->getConnection( 'mw.db' )
+		);
+	}
+
+	public function testUnregisteredConnectionTypeThrowsException() {
+
+		$instance = new ConnectionManager();
+
+		$this->setExpectedException( 'RuntimeException' );
+		$instance->getConnection( 'mw.master' );
+	}
+
+	public function testRegisterConnectionProvider() {
+
+		$connectionProvider = $this->getMockBuilder( '\SMW\Connection\ConnectionProvider' )
+			->disableOriginalConstructor()
+			->getMock();
+
+		$connectionProvider->expects( $this->once() )
+			->method( 'getConnection' );
+
+		$instance = new ConnectionManager();
+		$instance->registerConnectionProvider( 'foo', $connectionProvider );
+
+		$instance->getConnection( 'FOO' );
+	}
+
+	public function testRegisterCallbackConnection() {
+
+		$connectionProvider = $this->getMockBuilder( '\SMW\Connection\ConnectionProvider' )
+			->disableOriginalConstructor()
+			->getMock();
+
+		$connectionProvider->expects( $this->once() )
+			->method( 'getConnection' );
+
+		$callback = function() use( $connectionProvider ) {
+			return $connectionProvider->getConnection();
+		};
+
+		$instance = new ConnectionManager();
+		$instance->registerCallbackConnection( 'foo', $callback );
+
+		$instance->getConnection( 'FOO' );
+	}
+
+}