X-Git-Url: https://git.heureux-cyclage.org/?p=lhc%2Fweb%2Fwiklou.git;a=blobdiff_plain;f=tests%2Fphpunit%2Fincludes%2Fdb%2FLoadBalancerTest.php;h=6cc41d1f41774666aea86e6a95ef8b608dfa4f6d;hp=54aa2acbdfee06568320512bba7149d7a06b6814;hb=a2c8c2969420a0f150c03f76e3a0bf9028fcda43;hpb=1a21a63d52b9ebf940cd35f041d675d39c9d474a

diff --git a/tests/phpunit/includes/db/LoadBalancerTest.php b/tests/phpunit/includes/db/LoadBalancerTest.php
index 54aa2acbdf..6cc41d1f41 100644
--- a/tests/phpunit/includes/db/LoadBalancerTest.php
+++ b/tests/phpunit/includes/db/LoadBalancerTest.php
@@ -1,10 +1,5 @@
 <?php
 
-use Wikimedia\Rdbms\DBError;
-use Wikimedia\Rdbms\LoadBalancer;
-use Wikimedia\Rdbms\DatabaseDomain;
-use Wikimedia\Rdbms\Database;
-
 /**
  * Holds tests for LoadBalancer MediaWiki class.
  *
@@ -28,6 +23,13 @@ use Wikimedia\Rdbms\Database;
  *
  * @covers \Wikimedia\Rdbms\LoadBalancer
  */
+
+use Wikimedia\Rdbms\DBError;
+use Wikimedia\Rdbms\DatabaseDomain;
+use Wikimedia\Rdbms\Database;
+use Wikimedia\Rdbms\LoadBalancer;
+use Wikimedia\Rdbms\LoadMonitorNull;
+
 class LoadBalancerTest extends MediaWikiTestCase {
 	public function testWithoutReplica() {
 		global $wgDBserver, $wgDBname, $wgDBuser, $wgDBpassword, $wgDBtype, $wgSQLiteDataDir;
@@ -111,7 +113,7 @@ class LoadBalancerTest extends MediaWikiTestCase {
 		$lb = new LoadBalancer( [
 			'servers' => $servers,
 			'localDomain' => new DatabaseDomain( $wgDBname, null, $this->dbPrefix() ),
-			'loadMonitorClass' => 'LoadMonitorNull'
+			'loadMonitorClass' => LoadMonitorNull::class
 		] );
 
 		$dbw = $lb->getConnection( DB_MASTER );
@@ -190,4 +192,52 @@ class LoadBalancerTest extends MediaWikiTestCase {
 		}
 	}
 
+	public function testServerAttributes() {
+		$servers = [
+			[ // master
+				'dbname'      => 'my_unittest_wiki',
+				'tablePrefix' => 'unittest_',
+				'type'        => 'sqlite',
+				'dbDirectory' => "some_directory",
+				'load'        => 0
+			]
+		];
+
+		$lb = new LoadBalancer( [
+			'servers' => $servers,
+			'localDomain' => new DatabaseDomain( 'my_unittest_wiki', null, 'unittest_' ),
+			'loadMonitorClass' => LoadMonitorNull::class
+		] );
+
+		$this->assertTrue( $lb->getServerAttributes( 0 )[Database::ATTR_DB_LEVEL_LOCKING] );
+
+		$servers = [
+			[ // master
+				'host'        => 'db1001',
+				'user'        => 'wikiuser',
+				'password'    => 'none',
+				'dbname'      => 'my_unittest_wiki',
+				'tablePrefix' => 'unittest_',
+				'type'        => 'mysql',
+				'load'        => 100
+			],
+			[ // emulated replica
+				'host'        => 'db1002',
+				'user'        => 'wikiuser',
+				'password'    => 'none',
+				'dbname'      => 'my_unittest_wiki',
+				'tablePrefix' => 'unittest_',
+				'type'        => 'mysql',
+				'load'        => 100
+			]
+		];
+
+		$lb = new LoadBalancer( [
+			'servers' => $servers,
+			'localDomain' => new DatabaseDomain( 'my_unittest_wiki', null, 'unittest_' ),
+			'loadMonitorClass' => LoadMonitorNull::class
+		] );
+
+		$this->assertFalse( $lb->getServerAttributes( 1 )[Database::ATTR_DB_LEVEL_LOCKING] );
+	}
 }