3 use Wikimedia\Rdbms\LoadBalancer
;
6 * Holds tests for LoadBalancer MediaWiki class.
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 2 of the License, or
11 * (at your option) any later version.
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
18 * You should have received a copy of the GNU General Public License along
19 * with this program; if not, write to the Free Software Foundation, Inc.,
20 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
21 * http://www.gnu.org/copyleft/gpl.html
26 class LoadBalancerTest
extends MediaWikiTestCase
{
27 public function testLBSimpleServer() {
28 global $wgDBserver, $wgDBname, $wgDBuser, $wgDBpassword, $wgDBtype, $wgSQLiteDataDir;
32 'host' => $wgDBserver,
33 'dbname' => $wgDBname,
35 'password' => $wgDBpassword,
37 'dbDirectory' => $wgSQLiteDataDir,
39 'flags' => DBO_TRX
// REPEATABLE-READ for consistency
43 $lb = new LoadBalancer( [
44 'servers' => $servers,
45 'localDomain' => wfWikiID()
48 $dbw = $lb->getConnection( DB_MASTER
);
49 $this->assertTrue( $dbw->getLBInfo( 'master' ), 'master shows as master' );
50 $this->assertTrue( $dbw->getFlag( $dbw::DBO_TRX
), "DBO_TRX set on master" );
52 $dbr = $lb->getConnection( DB_REPLICA
);
53 $this->assertTrue( $dbr->getLBInfo( 'master' ), 'DB_REPLICA also gets the master' );
54 $this->assertTrue( $dbw->getFlag( $dbw::DBO_TRX
), "DBO_TRX set on replica" );
56 $dbwAuto = $lb->getConnection( DB_MASTER
, [], false, $lb::CONN_TRX_AUTO
);
57 $this->assertFalse( $dbwAuto->getFlag( $dbw::DBO_TRX
), "No DBO_TRX with CONN_TRX_AUTO" );
58 $this->assertTrue( $dbw->getFlag( $dbw::DBO_TRX
), "DBO_TRX still set on master" );
59 $this->assertNotEquals( $dbw, $dbwAuto, "CONN_TRX_AUTO uses separate connection" );
61 $dbrAuto = $lb->getConnection( DB_REPLICA
, [], false, $lb::CONN_TRX_AUTO
);
62 $this->assertFalse( $dbrAuto->getFlag( $dbw::DBO_TRX
), "No DBO_TRX with CONN_TRX_AUTO" );
63 $this->assertTrue( $dbr->getFlag( $dbw::DBO_TRX
), "DBO_TRX still set on replica" );
64 $this->assertNotEquals( $dbr, $dbrAuto, "CONN_TRX_AUTO uses separate connection" );
66 $dbwAuto2 = $lb->getConnection( DB_MASTER
, [], false, $lb::CONN_TRX_AUTO
);
67 $this->assertEquals( $dbwAuto2, $dbwAuto, "CONN_TRX_AUTO reuses connections" );
72 public function testLBSimpleServers() {
73 global $wgDBserver, $wgDBname, $wgDBuser, $wgDBpassword, $wgDBtype, $wgSQLiteDataDir;
77 'host' => $wgDBserver,
78 'dbname' => $wgDBname,
80 'password' => $wgDBpassword,
82 'dbDirectory' => $wgSQLiteDataDir,
84 'flags' => DBO_TRX
// REPEATABLE-READ for consistency
87 'host' => $wgDBserver,
88 'dbname' => $wgDBname,
90 'password' => $wgDBpassword,
92 'dbDirectory' => $wgSQLiteDataDir,
94 'flags' => DBO_TRX
// REPEATABLE-READ for consistency
98 $lb = new LoadBalancer( [
99 'servers' => $servers,
100 'localDomain' => wfWikiID(),
101 'loadMonitorClass' => 'LoadMonitorNull'
104 $dbw = $lb->getConnection( DB_MASTER
);
105 $this->assertTrue( $dbw->getLBInfo( 'master' ), 'master shows as master' );
107 ( $wgDBserver != '' ) ?
$wgDBserver : 'localhost',
108 $dbw->getLBInfo( 'clusterMasterHost' ),
109 'cluster master set' );
110 $this->assertTrue( $dbw->getFlag( $dbw::DBO_TRX
), "DBO_TRX set on master" );
112 $dbr = $lb->getConnection( DB_REPLICA
);
113 $this->assertTrue( $dbr->getLBInfo( 'replica' ), 'slave shows as slave' );
115 ( $wgDBserver != '' ) ?
$wgDBserver : 'localhost',
116 $dbr->getLBInfo( 'clusterMasterHost' ),
117 'cluster master set' );
118 $this->assertTrue( $dbw->getFlag( $dbw::DBO_TRX
), "DBO_TRX set on replica" );
120 $dbwAuto = $lb->getConnection( DB_MASTER
, [], false, $lb::CONN_TRX_AUTO
);
121 $this->assertFalse( $dbwAuto->getFlag( $dbw::DBO_TRX
), "No DBO_TRX with CONN_TRX_AUTO" );
122 $this->assertTrue( $dbw->getFlag( $dbw::DBO_TRX
), "DBO_TRX still set on master" );
123 $this->assertNotEquals( $dbw, $dbwAuto, "CONN_TRX_AUTO uses separate connection" );
125 $dbrAuto = $lb->getConnection( DB_REPLICA
, [], false, $lb::CONN_TRX_AUTO
);
126 $this->assertFalse( $dbrAuto->getFlag( $dbw::DBO_TRX
), "No DBO_TRX with CONN_TRX_AUTO" );
127 $this->assertTrue( $dbr->getFlag( $dbw::DBO_TRX
), "DBO_TRX still set on replica" );
128 $this->assertNotEquals( $dbr, $dbrAuto, "CONN_TRX_AUTO uses separate connection" );
130 $dbwAuto2 = $lb->getConnection( DB_MASTER
, [], false, $lb::CONN_TRX_AUTO
);
131 $this->assertEquals( $dbwAuto2, $dbwAuto, "CONN_TRX_AUTO reuses connections" );