3 * Holds tests for LBFactory abstract MediaWiki class.
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation; either version 2 of the License, or
8 * (at your option) any later version.
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
15 * You should have received a copy of the GNU General Public License along
16 * with this program; if not, write to the Free Software Foundation, Inc.,
17 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
18 * http://www.gnu.org/copyleft/gpl.html
21 * @author Antoine Musso
22 * @copyright © 2013 Antoine Musso
23 * @copyright © 2013 Wikimedia Foundation Inc.
26 use Wikimedia\AtEase\AtEase
;
27 use Wikimedia\Rdbms\IDatabase
;
28 use Wikimedia\Rdbms\IMaintainableDatabase
;
29 use Wikimedia\Rdbms\LBFactory
;
30 use Wikimedia\Rdbms\LBFactorySimple
;
31 use Wikimedia\Rdbms\LBFactoryMulti
;
32 use Wikimedia\Rdbms\LoadBalancer
;
33 use Wikimedia\Rdbms\ChronologyProtector
;
34 use Wikimedia\Rdbms\MySQLMasterPos
;
35 use Wikimedia\Rdbms\DatabaseDomain
;
39 * @covers \Wikimedia\Rdbms\LBFactory
40 * @covers \Wikimedia\Rdbms\LBFactorySimple
41 * @covers \Wikimedia\Rdbms\LBFactoryMulti
43 class LBFactoryTest
extends MediaWikiTestCase
{
46 * @covers MWLBFactory::getLBFactoryClass()
47 * @dataProvider getLBFactoryClassProvider
49 public function testGetLBFactoryClass( $expected, $deprecated ) {
50 $mockDB = $this->getMockBuilder( IDatabase
::class )
51 ->disableOriginalConstructor()
55 'class' => $deprecated,
56 'connection' => $mockDB,
57 # Various other parameters required:
60 'serverTemplate' => [],
63 $this->hideDeprecated( '$wgLBFactoryConf must be updated. See RELEASE-NOTES for details' );
64 $result = MWLBFactory
::getLBFactoryClass( $config );
66 $this->assertEquals( $expected, $result );
69 public function getLBFactoryClassProvider() {
71 # Format: new class, old class
72 [ Wikimedia\Rdbms\LBFactorySimple
::class, 'LBFactory_Simple' ],
73 [ Wikimedia\Rdbms\LBFactorySingle
::class, 'LBFactory_Single' ],
74 [ Wikimedia\Rdbms\LBFactoryMulti
::class, 'LBFactory_Multi' ],
75 [ Wikimedia\Rdbms\LBFactorySimple
::class, 'LBFactorySimple' ],
76 [ Wikimedia\Rdbms\LBFactorySingle
::class, 'LBFactorySingle' ],
77 [ Wikimedia\Rdbms\LBFactoryMulti
::class, 'LBFactoryMulti' ],
82 * @covers \Wikimedia\Rdbms\LBFactory::getLocalDomainID()
83 * @covers \Wikimedia\Rdbms\LBFactory::resolveDomainID()
85 public function testLBFactorySimpleServer() {
86 global $wgDBserver, $wgDBname, $wgDBuser, $wgDBpassword, $wgDBtype, $wgSQLiteDataDir;
90 'host' => $wgDBserver,
91 'dbname' => $wgDBname,
93 'password' => $wgDBpassword,
95 'dbDirectory' => $wgSQLiteDataDir,
97 'flags' => DBO_TRX
// REPEATABLE-READ for consistency
101 $factory = new LBFactorySimple( [ 'servers' => $servers ] );
102 $lb = $factory->getMainLB();
104 $dbw = $lb->getConnection( DB_MASTER
);
105 $this->assertTrue( $dbw->getLBInfo( 'master' ), 'master shows as master' );
107 $dbr = $lb->getConnection( DB_REPLICA
);
108 $this->assertTrue( $dbr->getLBInfo( 'master' ), 'DB_REPLICA also gets the master' );
110 $this->assertSame( 'my_test_wiki', $factory->resolveDomainID( 'my_test_wiki' ) );
111 $this->assertSame( $factory->getLocalDomainID(), $factory->resolveDomainID( false ) );
113 $factory->shutdown();
117 public function testLBFactorySimpleServers() {
118 global $wgDBserver, $wgDBname, $wgDBuser, $wgDBpassword, $wgDBtype, $wgSQLiteDataDir;
122 'host' => $wgDBserver,
123 'dbname' => $wgDBname,
125 'password' => $wgDBpassword,
127 'dbDirectory' => $wgSQLiteDataDir,
129 'flags' => DBO_TRX
// REPEATABLE-READ for consistency
131 [ // emulated replica
132 'host' => $wgDBserver,
133 'dbname' => $wgDBname,
135 'password' => $wgDBpassword,
137 'dbDirectory' => $wgSQLiteDataDir,
139 'flags' => DBO_TRX
// REPEATABLE-READ for consistency
143 $factory = new LBFactorySimple( [
144 'servers' => $servers,
145 'loadMonitorClass' => LoadMonitorNull
::class
147 $lb = $factory->getMainLB();
149 $dbw = $lb->getConnection( DB_MASTER
);
150 $this->assertTrue( $dbw->getLBInfo( 'master' ), 'master shows as master' );
152 ( $wgDBserver != '' ) ?
$wgDBserver : 'localhost',
153 $dbw->getLBInfo( 'clusterMasterHost' ),
154 'cluster master set' );
156 $dbr = $lb->getConnection( DB_REPLICA
);
157 $this->assertTrue( $dbr->getLBInfo( 'replica' ), 'replica shows as replica' );
159 ( $wgDBserver != '' ) ?
$wgDBserver : 'localhost',
160 $dbr->getLBInfo( 'clusterMasterHost' ),
161 'cluster master set' );
163 $factory->shutdown();
167 public function testLBFactoryMultiConns() {
168 $factory = $this->newLBFactoryMultiLBs();
170 $dbw = $factory->getMainLB()->getConnection( DB_MASTER
);
171 $this->assertTrue( $dbw->getLBInfo( 'master' ), 'master shows as master' );
173 $dbr = $factory->getMainLB()->getConnection( DB_REPLICA
);
174 $this->assertTrue( $dbr->getLBInfo( 'replica' ), 'replica shows as replica' );
176 // Destructor should trigger without round stage errors
180 public function testLBFactoryMultiRoundCallbacks() {
182 $countLBsFunc = function ( LBFactoryMulti
$factory ) {
184 $factory->forEachLB( function () use ( &$count ) {
191 $factory = $this->newLBFactoryMultiLBs();
192 $this->assertEquals( 0, $countLBsFunc( $factory ) );
193 $dbw = $factory->getMainLB()->getConnection( DB_MASTER
);
194 $this->assertEquals( 1, $countLBsFunc( $factory ) );
195 // Test that LoadBalancer instances made during pre-commit callbacks in do not
196 // throw DBTransactionError due to transaction ROUND_* stages being mismatched.
197 $factory->beginMasterChanges( __METHOD__
);
198 $dbw->onTransactionPreCommitOrIdle( function () use ( $factory, &$called ) {
200 // Trigger s1 LoadBalancer instantiation during "finalize" stage.
201 // There is no s1wiki DB to select so it is not in getConnection(),
202 // but this fools getMainLB() at least.
203 $factory->getMainLB( 's1wiki' )->getConnection( DB_MASTER
);
205 $factory->commitMasterChanges( __METHOD__
);
206 $this->assertEquals( 1, $called );
207 $this->assertEquals( 2, $countLBsFunc( $factory ) );
208 $factory->shutdown();
209 $factory->closeAll();
212 $factory = $this->newLBFactoryMultiLBs();
213 $this->assertEquals( 0, $countLBsFunc( $factory ) );
214 $dbw = $factory->getMainLB()->getConnection( DB_MASTER
);
215 $this->assertEquals( 1, $countLBsFunc( $factory ) );
216 // Test that LoadBalancer instances made during pre-commit callbacks in do not
217 // throw DBTransactionError due to transaction ROUND_* stages being mismatched.hrow
218 // DBTransactionError due to transaction ROUND_* stages being mismatched.
219 $factory->beginMasterChanges( __METHOD__
);
220 $dbw->query( "SELECT 1 as t", __METHOD__
);
221 $dbw->onTransactionResolution( function () use ( $factory, &$called ) {
223 // Trigger s1 LoadBalancer instantiation during "finalize" stage.
224 // There is no s1wiki DB to select so it is not in getConnection(),
225 // but this fools getMainLB() at least.
226 $factory->getMainLB( 's1wiki' )->getConnection( DB_MASTER
);
228 $factory->commitMasterChanges( __METHOD__
);
229 $this->assertEquals( 1, $called );
230 $this->assertEquals( 2, $countLBsFunc( $factory ) );
231 $factory->shutdown();
232 $factory->closeAll();
234 $factory = $this->newLBFactoryMultiLBs();
235 $dbw = $factory->getMainLB()->getConnection( DB_MASTER
);
236 // DBTransactionError should not be thrown
238 $dbw->onTransactionPreCommitOrIdle( function () use ( &$ran ) {
241 $factory->commitAll( __METHOD__
);
242 $this->assertEquals( 1, $ran );
244 $factory->shutdown();
245 $factory->closeAll();
248 private function newLBFactoryMultiLBs() {
249 global $wgDBserver, $wgDBname, $wgDBuser, $wgDBpassword, $wgDBtype, $wgSQLiteDataDir;
251 return new LBFactoryMulti( [
265 'serverTemplate' => [
266 'dbname' => $wgDBname,
268 'password' => $wgDBpassword,
270 'dbDirectory' => $wgSQLiteDataDir,
271 'flags' => DBO_DEFAULT
274 'test-db1' => $wgDBserver,
275 'test-db2' => $wgDBserver,
276 'test-db3' => $wgDBserver,
277 'test-db4' => $wgDBserver
279 'loadMonitorClass' => LoadMonitorNull
::class
284 * @covers \Wikimedia\Rdbms\ChronologyProtector
286 public function testChronologyProtector() {
287 $now = microtime( true );
289 // (a) First HTTP request
290 $m1Pos = new MySQLMasterPos( 'db1034-bin.000976/843431247', $now );
291 $m2Pos = new MySQLMasterPos( 'db1064-bin.002400/794074907', $now );
294 $mockDB1 = $this->getMockBuilder( IDatabase
::class )
295 ->disableOriginalConstructor()
297 $mockDB1->method( 'writesOrCallbacksPending' )->willReturn( true );
298 $mockDB1->method( 'lastDoneWrites' )->willReturn( $now );
299 $mockDB1->method( 'getMasterPos' )->willReturn( $m1Pos );
300 // Load balancer for master DB 1
301 $lb1 = $this->getMockBuilder( LoadBalancer
::class )
302 ->disableOriginalConstructor()
304 $lb1->method( 'getConnection' )->willReturn( $mockDB1 );
305 $lb1->method( 'getServerCount' )->willReturn( 2 );
306 $lb1->method( 'hasReplicaServers' )->willReturn( true );
307 $lb1->method( 'hasStreamingReplicaServers' )->willReturn( true );
308 $lb1->method( 'getAnyOpenConnection' )->willReturn( $mockDB1 );
309 $lb1->method( 'hasOrMadeRecentMasterChanges' )->will( $this->returnCallback(
310 function () use ( $mockDB1 ) {
312 $p |
= call_user_func( [ $mockDB1, 'writesOrCallbacksPending' ] );
313 $p |
= call_user_func( [ $mockDB1, 'lastDoneWrites' ] );
318 $lb1->method( 'getMasterPos' )->willReturn( $m1Pos );
319 $lb1->method( 'getReplicaResumePos' )->willReturn( $m1Pos );
320 $lb1->method( 'getServerName' )->with( 0 )->willReturn( 'master1' );
322 $mockDB2 = $this->getMockBuilder( IDatabase
::class )
323 ->disableOriginalConstructor()
325 $mockDB2->method( 'writesOrCallbacksPending' )->willReturn( true );
326 $mockDB2->method( 'lastDoneWrites' )->willReturn( $now );
327 $mockDB2->method( 'getMasterPos' )->willReturn( $m2Pos );
328 // Load balancer for master DB 2
329 $lb2 = $this->getMockBuilder( LoadBalancer
::class )
330 ->disableOriginalConstructor()
332 $lb2->method( 'getConnection' )->willReturn( $mockDB2 );
333 $lb2->method( 'getServerCount' )->willReturn( 2 );
334 $lb2->method( 'hasReplicaServers' )->willReturn( true );
335 $lb2->method( 'hasStreamingReplicaServers' )->willReturn( true );
336 $lb2->method( 'getAnyOpenConnection' )->willReturn( $mockDB2 );
337 $lb2->method( 'hasOrMadeRecentMasterChanges' )->will( $this->returnCallback(
338 function () use ( $mockDB2 ) {
340 $p |
= call_user_func( [ $mockDB2, 'writesOrCallbacksPending' ] );
341 $p |
= call_user_func( [ $mockDB2, 'lastDoneWrites' ] );
346 $lb2->method( 'getMasterPos' )->willReturn( $m2Pos );
347 $lb2->method( 'getReplicaResumePos' )->willReturn( $m2Pos );
348 $lb2->method( 'getServerName' )->with( 0 )->willReturn( 'master2' );
350 $bag = new HashBagOStuff();
351 $cp = new ChronologyProtector(
355 'agent' => "Totally-Not-FireFox"
360 $mockDB1->expects( $this->exactly( 1 ) )->method( 'writesOrCallbacksPending' );
361 $mockDB1->expects( $this->exactly( 1 ) )->method( 'lastDoneWrites' );
362 $mockDB2->expects( $this->exactly( 1 ) )->method( 'writesOrCallbacksPending' );
363 $mockDB2->expects( $this->exactly( 1 ) )->method( 'lastDoneWrites' );
365 // Nothing to wait for on first HTTP request start
366 $cp->applySessionReplicationPosition( $lb1 );
367 $cp->applySessionReplicationPosition( $lb2 );
368 // Record positions in stash on first HTTP request end
369 $cp->storeSessionReplicationPosition( $lb1 );
370 $cp->storeSessionReplicationPosition( $lb2 );
372 $cp->shutdown( null, 'sync', $cpIndex );
374 $this->assertEquals( 1, $cpIndex, "CP write index set" );
376 // (b) Second HTTP request
378 // Load balancer for master DB 1
379 $lb1 = $this->getMockBuilder( LoadBalancer
::class )
380 ->disableOriginalConstructor()
382 $lb1->method( 'getServerCount' )->willReturn( 2 );
383 $lb1->method( 'hasReplicaServers' )->willReturn( true );
384 $lb1->method( 'hasStreamingReplicaServers' )->willReturn( true );
385 $lb1->method( 'getServerName' )->with( 0 )->willReturn( 'master1' );
386 $lb1->expects( $this->once() )
387 ->method( 'waitFor' )->with( $this->equalTo( $m1Pos ) );
388 // Load balancer for master DB 2
389 $lb2 = $this->getMockBuilder( LoadBalancer
::class )
390 ->disableOriginalConstructor()
392 $lb2->method( 'getServerCount' )->willReturn( 2 );
393 $lb2->method( 'hasReplicaServers' )->willReturn( true );
394 $lb2->method( 'hasStreamingReplicaServers' )->willReturn( true );
395 $lb2->method( 'getServerName' )->with( 0 )->willReturn( 'master2' );
396 $lb2->expects( $this->once() )
397 ->method( 'waitFor' )->with( $this->equalTo( $m2Pos ) );
399 $cp = new ChronologyProtector(
403 'agent' => "Totally-Not-FireFox"
408 // Wait for last positions to be reached on second HTTP request start
409 $cp->applySessionReplicationPosition( $lb1 );
410 $cp->applySessionReplicationPosition( $lb2 );
411 // Shutdown (nothing to record)
412 $cp->storeSessionReplicationPosition( $lb1 );
413 $cp->storeSessionReplicationPosition( $lb2 );
415 $cp->shutdown( null, 'sync', $cpIndex );
417 $this->assertEquals( null, $cpIndex, "CP write index retained" );
419 $this->assertEquals( '45e93a9c215c031d38b7c42d8e4700ca', $cp->getClientId() );
422 private function newLBFactoryMulti( array $baseOverride = [], array $serverOverride = [] ) {
423 global $wgDBserver, $wgDBuser, $wgDBpassword, $wgDBname, $wgDBprefix, $wgDBtype;
424 global $wgSQLiteDataDir;
426 return new LBFactoryMulti( $baseOverride +
[
427 'sectionsByDB' => [],
433 'serverTemplate' => $serverOverride +
[
434 'dbname' => $wgDBname,
435 'tablePrefix' => $wgDBprefix,
437 'password' => $wgDBpassword,
439 'dbDirectory' => $wgSQLiteDataDir,
440 'flags' => DBO_DEFAULT
443 'test-db1' => $wgDBserver,
445 'loadMonitorClass' => LoadMonitorNull
::class,
446 'localDomain' => new DatabaseDomain( $wgDBname, null, $wgDBprefix ),
447 'agent' => 'MW-UNIT-TESTS'
452 * @covers \Wikimedia\Rdbms\LoadBalancer::getConnection
453 * @covers \Wikimedia\Rdbms\DatabaseMysqlBase::doSelectDomain
454 * @covers \Wikimedia\Rdbms\DatabaseMysqlBase::selectDB
456 public function testNiceDomains() {
459 if ( wfGetDB( DB_MASTER
)->databasesAreIndependent() ) {
460 self
::markTestSkipped( "Skipping tests about selecting DBs: not applicable" );
464 $factory = $this->newLBFactoryMulti(
468 $lb = $factory->getMainLB();
470 $db = $lb->getConnectionRef( DB_MASTER
);
477 /** @var IMaintainableDatabase $db */
478 $db = $lb->getConnection( DB_MASTER
, [], $lb::DOMAIN_ANY
);
483 'Null domain ID handle used'
488 'Null domain ID handle used'
493 'Main domain ID handle used; prefix is empty though'
496 $this->quoteTable( $db, 'page' ),
497 $db->tableName( 'page' ),
498 "Correct full table name"
501 $this->quoteTable( $db, $wgDBname ) . '.' . $this->quoteTable( $db, 'page' ),
502 $db->tableName( "$wgDBname.page" ),
503 "Correct full table name"
506 $this->quoteTable( $db, 'nice_db' ) . '.' . $this->quoteTable( $db, 'page' ),
507 $db->tableName( 'nice_db.page' ),
508 "Correct full table name"
511 $lb->reuseConnection( $db ); // don't care
513 $db = $lb->getConnection( DB_MASTER
); // local domain connection
514 $factory->setLocalDomainPrefix( 'my_' );
516 $this->assertEquals( $wgDBname, $db->getDBname() );
522 $this->quoteTable( $db, 'my_page' ),
523 $db->tableName( 'page' ),
524 "Correct full table name"
527 $this->quoteTable( $db, 'other_nice_db' ) . '.' . $this->quoteTable( $db, 'page' ),
528 $db->tableName( 'other_nice_db.page' ),
529 "Correct full table name"
532 $factory->closeAll();
537 * @covers \Wikimedia\Rdbms\LoadBalancer::getConnection
538 * @covers \Wikimedia\Rdbms\DatabaseMysqlBase::doSelectDomain
539 * @covers \Wikimedia\Rdbms\DatabaseMysqlBase::selectDB
541 public function testTrickyDomain() {
544 if ( wfGetDB( DB_MASTER
)->databasesAreIndependent() ) {
545 self
::markTestSkipped( "Skipping tests about selecting DBs: not applicable" );
549 $dbname = 'unittest-domain'; // explodes if DB is selected
550 $factory = $this->newLBFactoryMulti(
551 [ 'localDomain' => ( new DatabaseDomain( $dbname, null, '' ) )->getId() ],
553 'dbname' => 'do_not_select_me' // explodes if DB is selected
556 $lb = $factory->getMainLB();
557 /** @var IMaintainableDatabase $db */
558 $db = $lb->getConnection( DB_MASTER
, [], $lb::DOMAIN_ANY
);
560 $this->assertEquals( '', $db->getDomainID(), "Null domain used" );
563 $this->quoteTable( $db, 'page' ),
564 $db->tableName( 'page' ),
565 "Correct full table name"
569 $this->quoteTable( $db, $dbname ) . '.' . $this->quoteTable( $db, 'page' ),
570 $db->tableName( "$dbname.page" ),
571 "Correct full table name"
575 $this->quoteTable( $db, 'nice_db' ) . '.' . $this->quoteTable( $db, 'page' ),
576 $db->tableName( 'nice_db.page' ),
577 "Correct full table name"
580 $lb->reuseConnection( $db ); // don't care
582 $factory->setLocalDomainPrefix( 'my_' );
583 $db = $lb->getConnection( DB_MASTER
, [], "$wgDBname-my_" );
586 $this->quoteTable( $db, 'my_page' ),
587 $db->tableName( 'page' ),
588 "Correct full table name"
591 $this->quoteTable( $db, 'other_nice_db' ) . '.' . $this->quoteTable( $db, 'page' ),
592 $db->tableName( 'other_nice_db.page' ),
593 "Correct full table name"
596 $this->quoteTable( $db, 'garbage-db' ) . '.' . $this->quoteTable( $db, 'page' ),
597 $db->tableName( 'garbage-db.page' ),
598 "Correct full table name"
601 $lb->reuseConnection( $db ); // don't care
603 $factory->closeAll();
608 * @covers \Wikimedia\Rdbms\LoadBalancer::getConnection
609 * @covers \Wikimedia\Rdbms\DatabaseMysqlBase::doSelectDomain
610 * @covers \Wikimedia\Rdbms\DatabaseMysqlBase::selectDB
612 public function testInvalidSelectDB() {
613 if ( wfGetDB( DB_MASTER
)->databasesAreIndependent() ) {
614 $this->markTestSkipped( "Not applicable per databasesAreIndependent()" );
617 $dbname = 'unittest-domain'; // explodes if DB is selected
618 $factory = $this->newLBFactoryMulti(
619 [ 'localDomain' => ( new DatabaseDomain( $dbname, null, '' ) )->getId() ],
621 'dbname' => 'do_not_select_me' // explodes if DB is selected
624 $lb = $factory->getMainLB();
625 /** @var IDatabase $db */
626 $db = $lb->getConnection( DB_MASTER
, [], $lb::DOMAIN_ANY
);
628 AtEase
::suppressWarnings();
630 $this->assertFalse( $db->selectDomain( 'garbagedb' ) );
631 $this->fail( "No error thrown." );
632 } catch ( \Wikimedia\Rdbms\DBQueryError
$e ) {
633 $this->assertRegExp( '/[\'"]garbagedb[\'"]/', $e->getMessage() );
635 AtEase
::restoreWarnings();
639 * @covers \Wikimedia\Rdbms\DatabaseSqlite::selectDB
640 * @covers \Wikimedia\Rdbms\DatabasePostgres::selectDB
641 * @expectedException \Wikimedia\Rdbms\DBConnectionError
643 public function testInvalidSelectDBIndependent() {
644 $dbname = 'unittest-domain'; // explodes if DB is selected
645 $factory = $this->newLBFactoryMulti(
646 [ 'localDomain' => ( new DatabaseDomain( $dbname, null, '' ) )->getId() ],
648 // Explodes with SQLite and Postgres during open/USE
649 'dbname' => 'bad_dir/do_not_select_me'
652 $lb = $factory->getMainLB();
654 if ( !$factory->getMainLB()->getServerAttributes( 0 )[Database
::ATTR_DB_IS_FILE
] ) {
655 $this->markTestSkipped( "Not applicable per ATTR_DB_IS_FILE" );
658 /** @var IDatabase $db */
659 $this->assertNotNull( $lb->getConnection( DB_MASTER
, [], $lb::DOMAIN_ANY
) );
663 * @covers \Wikimedia\Rdbms\DatabaseSqlite::selectDB
664 * @covers \Wikimedia\Rdbms\DatabasePostgres::selectDB
665 * @expectedException \Wikimedia\Rdbms\DBExpectedError
667 public function testInvalidSelectDBIndependent2() {
668 $dbname = 'unittest-domain'; // explodes if DB is selected
669 $factory = $this->newLBFactoryMulti(
670 [ 'localDomain' => ( new DatabaseDomain( $dbname, null, '' ) )->getId() ],
672 // Explodes with SQLite and Postgres during open/USE
673 'dbname' => 'bad_dir/do_not_select_me'
676 $lb = $factory->getMainLB();
678 if ( !$lb->getConnection( DB_MASTER
)->databasesAreIndependent() ) {
679 $this->markTestSkipped( "Not applicable per databasesAreIndependent()" );
682 $db = $lb->getConnection( DB_MASTER
);
683 $db->selectDomain( 'garbage-db' );
687 * @covers \Wikimedia\Rdbms\LoadBalancer::getConnection
688 * @covers \Wikimedia\Rdbms\LoadBalancer::redefineLocalDomain
689 * @covers \Wikimedia\Rdbms\DatabaseMysqlBase::selectDB
691 public function testRedefineLocalDomain() {
694 if ( wfGetDB( DB_MASTER
)->databasesAreIndependent() ) {
695 self
::markTestSkipped( "Skipping tests about selecting DBs: not applicable" );
699 $factory = $this->newLBFactoryMulti(
703 $lb = $factory->getMainLB();
705 $conn1 = $lb->getConnectionRef( DB_MASTER
);
708 $conn1->getDomainID()
712 $factory->redefineLocalDomain( 'somedb-prefix_' );
713 $this->assertEquals( 'somedb-prefix_', $factory->getLocalDomainID() );
715 $domain = new DatabaseDomain( $wgDBname, null, 'pref_' );
716 $factory->redefineLocalDomain( $domain );
719 $lb->forEachOpenConnection( function () use ( &$n ) {
722 $this->assertEquals( 0, $n, "Connections closed" );
724 $conn2 = $lb->getConnectionRef( DB_MASTER
);
727 $conn2->getDomainID()
731 $factory->closeAll();
735 private function quoteTable( IDatabase
$db, $table ) {
736 if ( $db->getType() === 'sqlite' ) {
739 return $db->addIdentifierQuotes( $table );
744 * @covers \Wikimedia\Rdbms\LBFactory::makeCookieValueFromCPIndex()
745 * @covers \Wikimedia\Rdbms\LBFactory::getCPInfoFromCookieValue()
747 public function testCPPosIndexCookieValues() {
749 $agentId = md5( 'Ramsey\'s Loyal Presa Canario' );
752 '3@542#c47dcfb0566e7d7bc110a6128a45c93a',
753 LBFactory
::makeCookieValueFromCPIndex( 3, 542, $agentId )
756 $lbFactory = $this->newLBFactoryMulti();
757 $lbFactory->setRequestInfo( [ 'IPAddress' => '10.64.24.52', 'UserAgent' => 'meow' ] );
759 '1@542#c47dcfb0566e7d7bc110a6128a45c93a',
760 LBFactory
::makeCookieValueFromCPIndex( 1, 542, $agentId )
765 LBFactory
::getCPInfoFromCookieValue( "5#$agentId", $time - 10 )['index'],
770 LBFactory
::getCPInfoFromCookieValue( "5@$time", $time - 10 )['index'],
775 LBFactory
::getCPInfoFromCookieValue( "0@$time#$agentId", $time - 10 )['index'],
781 LBFactory
::getCPInfoFromCookieValue( "2@$time#$agentId", $time - 10 )['index'],
786 LBFactory
::getCPInfoFromCookieValue( "2@$time#$agentId", $time +
9 - 10 )['index'],
791 LBFactory
::getCPInfoFromCookieValue( "0@$time#$agentId", $time +
9 - 10 )['index'],
792 'Almost stale; bad index'
796 LBFactory
::getCPInfoFromCookieValue( "2@$time#$agentId", $time +
11 - 10 )['index'],
802 LBFactory
::getCPInfoFromCookieValue( "5@$time#$agentId", $time - 10 )['clientId'],
807 LBFactory
::getCPInfoFromCookieValue( "5@$time#$agentId", $time +
11 - 10 )['clientId'],