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' );
66 $this->assertEquals( $expected,
$result );
71 # Format: new class, old class
102 $lb = $factory->getMainLB();
105 $this->assertTrue( $dbw->getLBInfo(
'master' ),
'master shows as master' );
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();
144 'servers' => $servers,
147 $lb = $factory->getMainLB();
150 $this->assertTrue( $dbw->getLBInfo(
'master' ),
'master shows as master' );
153 $dbw->getLBInfo(
'clusterMasterHost' ),
154 'cluster master set' );
157 $this->assertTrue(
$dbr->getLBInfo(
'replica' ),
'replica shows as replica' );
160 $dbr->getLBInfo(
'clusterMasterHost' ),
161 'cluster master set' );
163 $factory->shutdown();
170 $dbw = $factory->getMainLB()->getConnection(
DB_MASTER );
171 $this->assertTrue( $dbw->getLBInfo(
'master' ),
'master shows as master' );
174 $this->assertTrue(
$dbr->getLBInfo(
'replica' ),
'replica shows as replica' );
184 $factory->forEachLB(
function ()
use ( &$count ) {
192 $this->assertEquals( 0, $countLBsFunc( $factory ) );
193 $dbw = $factory->getMainLB()->getConnection(
DB_MASTER );
194 $this->assertEquals( 1, $countLBsFunc( $factory ) );
197 $factory->beginMasterChanges( __METHOD__ );
198 $dbw->onTransactionPreCommitOrIdle(
function ()
use ( $factory, &
$called ) {
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();
213 $this->assertEquals( 0, $countLBsFunc( $factory ) );
214 $dbw = $factory->getMainLB()->getConnection(
DB_MASTER );
215 $this->assertEquals( 1, $countLBsFunc( $factory ) );
219 $factory->beginMasterChanges( __METHOD__ );
220 $dbw->query(
"SELECT 1 as t", __METHOD__ );
221 $dbw->onTransactionResolution(
function ()
use ( $factory, &
$called ) {
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();
235 $dbw = $factory->getMainLB()->getConnection(
DB_MASTER );
238 $dbw->onTransactionPreCommitOrIdle(
function ()
use ( &$ran ) {
241 $factory->commitAll( __METHOD__ );
242 $this->assertEquals( 1, $ran );
244 $factory->shutdown();
245 $factory->closeAll();
265 'serverTemplate' => [
287 $now = microtime(
true );
290 $m1Pos =
new MySQLMasterPos(
'db1034-bin.000976/843431247', $now );
291 $m2Pos =
new MySQLMasterPos(
'db1064-bin.002400/794074907', $now );
295 ->disableOriginalConstructor()
297 $mockDB1->method(
'writesOrCallbacksPending' )->willReturn(
true );
298 $mockDB1->method(
'lastDoneWrites' )->willReturn( $now );
299 $mockDB1->method(
'getMasterPos' )->willReturn( $m1Pos );
302 ->disableOriginalConstructor()
304 $lb1->method(
'getConnection' )->willReturn( $mockDB1 );
305 $lb1->method(
'getServerCount' )->willReturn( 2 );
306 $lb1->method(
'getAnyOpenConnection' )->willReturn( $mockDB1 );
307 $lb1->method(
'hasOrMadeRecentMasterChanges' )->will( $this->returnCallback(
308 function ()
use ( $mockDB1 ) {
310 $p |= call_user_func( [ $mockDB1,
'writesOrCallbacksPending' ] );
311 $p |= call_user_func( [ $mockDB1,
'lastDoneWrites' ] );
316 $lb1->method(
'getMasterPos' )->willReturn( $m1Pos );
317 $lb1->method(
'getServerName' )->with( 0 )->willReturn(
'master1' );
320 ->disableOriginalConstructor()
322 $mockDB2->method(
'writesOrCallbacksPending' )->willReturn(
true );
323 $mockDB2->method(
'lastDoneWrites' )->willReturn( $now );
324 $mockDB2->method(
'getMasterPos' )->willReturn( $m2Pos );
327 ->disableOriginalConstructor()
329 $lb2->method(
'getConnection' )->willReturn( $mockDB2 );
330 $lb2->method(
'getServerCount' )->willReturn( 2 );
331 $lb2->method(
'getAnyOpenConnection' )->willReturn( $mockDB2 );
332 $lb2->method(
'hasOrMadeRecentMasterChanges' )->will( $this->returnCallback(
333 function ()
use ( $mockDB2 ) {
335 $p |= call_user_func( [ $mockDB2,
'writesOrCallbacksPending' ] );
336 $p |= call_user_func( [ $mockDB2,
'lastDoneWrites' ] );
341 $lb2->method(
'getMasterPos' )->willReturn( $m2Pos );
342 $lb2->method(
'getServerName' )->with( 0 )->willReturn(
'master2' );
349 'agent' =>
"Totally-Not-FireFox"
353 $mockDB1->expects( $this->exactly( 1 ) )->method(
'writesOrCallbacksPending' );
354 $mockDB1->expects( $this->exactly( 1 ) )->method(
'lastDoneWrites' );
355 $mockDB2->expects( $this->exactly( 1 ) )->method(
'writesOrCallbacksPending' );
356 $mockDB2->expects( $this->exactly( 1 ) )->method(
'lastDoneWrites' );
362 $cp->shutdownLB( $lb1 );
363 $cp->shutdownLB( $lb2 );
365 $cp->shutdown(
null,
'sync', $cpIndex );
367 $this->assertEquals( 1, $cpIndex,
"CP write index set" );
373 ->disableOriginalConstructor()
375 $lb1->method(
'getServerCount' )->willReturn( 2 );
376 $lb1->method(
'getServerName' )->with( 0 )->willReturn(
'master1' );
377 $lb1->expects( $this->once() )
378 ->method(
'waitFor' )->with( $this->equalTo( $m1Pos ) );
381 ->disableOriginalConstructor()
383 $lb2->method(
'getServerCount' )->willReturn( 2 );
384 $lb2->method(
'getServerName' )->with( 0 )->willReturn(
'master2' );
385 $lb2->expects( $this->once() )
386 ->method(
'waitFor' )->with( $this->equalTo( $m2Pos ) );
392 'agent' =>
"Totally-Not-FireFox"
401 $cp->shutdownLB( $lb1 );
402 $cp->shutdownLB( $lb2 );
404 $cp->shutdown(
null,
'sync', $cpIndex );
406 $this->assertEquals(
null, $cpIndex,
"CP write index retained" );
408 $this->assertEquals(
'45e93a9c215c031d38b7c42d8e4700ca', $cp->getClientId() );
416 'sectionsByDB' => [],
422 'serverTemplate' => $serverOverride + [
436 'agent' =>
'MW-UNIT-TESTS'
451 self::markTestSkipped(
"Skipping tests about selecting DBs: not applicable" );
459 $lb = $factory->getMainLB();
474 'Null domain ID handle used'
479 'Null domain ID handle used'
484 'Main domain ID handle used; prefix is empty though'
488 $db->tableName(
'page' ),
489 "Correct full table name"
493 $db->tableName(
"$wgDBname.page" ),
494 "Correct full table name"
498 $db->tableName(
'nice_db.page' ),
499 "Correct full table name"
502 $lb->reuseConnection(
$db );
505 $factory->setLocalDomainPrefix(
'my_' );
514 $db->tableName(
'page' ),
515 "Correct full table name"
519 $db->tableName(
'other_nice_db.page' ),
520 "Correct full table name"
523 $factory->closeAll();
538 self::markTestSkipped(
"Skipping tests about selecting DBs: not applicable" );
542 $dbname =
'unittest-domain';
544 [
'localDomain' => (
new DatabaseDomain( $dbname,
null,
'' ) )->getId() ],
546 'dbname' =>
'do_not_select_me'
549 $lb = $factory->getMainLB();
553 $this->assertEquals(
'',
$db->getDomainID(),
"Null domain used" );
557 $db->tableName(
'page' ),
558 "Correct full table name"
563 $db->tableName(
"$dbname.page" ),
564 "Correct full table name"
569 $db->tableName(
'nice_db.page' ),
570 "Correct full table name"
573 $lb->reuseConnection(
$db );
575 $factory->setLocalDomainPrefix(
'my_' );
576 $db = $lb->getConnection(
DB_MASTER, [],
"$wgDBname-my_" );
580 $db->tableName(
'page' ),
581 "Correct full table name"
585 $db->tableName(
'other_nice_db.page' ),
586 "Correct full table name"
590 $db->tableName(
'garbage-db.page' ),
591 "Correct full table name"
594 $lb->reuseConnection(
$db );
596 $factory->closeAll();
609 $this->markTestSkipped(
"Not applicable per databasesAreIndependent()" );
612 $dbname =
'unittest-domain';
614 [
'localDomain' => (
new DatabaseDomain( $dbname,
null,
'' ) )->getId() ],
616 'dbname' =>
'do_not_select_me'
619 $lb = $factory->getMainLB();
623 \Wikimedia\suppressWarnings();
625 $this->assertFalse(
$db->selectDB(
'garbage-db' ) );
626 $this->fail(
"No error thrown." );
628 $this->assertRegExp(
'/[\'"]garbage-db[\'"]/',
$e->getMessage() );
630 \Wikimedia\restoreWarnings();
639 $dbname =
'unittest-domain';
641 [
'localDomain' => (
new DatabaseDomain( $dbname,
null,
'' ) )->getId() ],
643 'dbname' =>
'do_not_select_me'
646 $lb = $factory->getMainLB();
649 $this->markTestSkipped(
"Not applicable per databasesAreIndependent()" );
662 $dbname =
'unittest-domain';
664 [
'localDomain' => (
new DatabaseDomain( $dbname,
null,
'' ) )->getId() ],
666 'dbname' =>
'do_not_select_me'
669 $lb = $factory->getMainLB();
672 $this->markTestSkipped(
"Not applicable per databasesAreIndependent()" );
676 \Wikimedia\suppressWarnings();
677 $db->selectDB(
'garbage-db' );
678 \Wikimedia\restoreWarnings();
692 self::markTestSkipped(
"Skipping tests about selecting DBs: not applicable" );
700 $lb = $factory->getMainLB();
702 $conn1 = $lb->getConnectionRef(
DB_MASTER );
705 $conn1->getDomainID()
709 $factory->redefineLocalDomain(
'somedb-prefix_' );
710 $this->assertEquals(
'somedb-prefix_', $factory->getLocalDomainID() );
713 $factory->redefineLocalDomain( $domain );
716 $lb->forEachOpenConnection(
function ()
use ( &$n ) {
719 $this->assertEquals( 0, $n,
"Connections closed" );
721 $conn2 = $lb->getConnectionRef(
DB_MASTER );
724 $conn2->getDomainID()
728 $factory->closeAll();
733 if (
$db->getType() ===
'sqlite' ) {
736 return $db->addIdentifierQuotes( $table );
746 $agentId = md5(
'Ramsey\'s Loyal Presa Canario' );
749 '3@542#c47dcfb0566e7d7bc110a6128a45c93a',
750 LBFactory::makeCookieValueFromCPIndex( 3, 542, $agentId )
754 $lbFactory->setRequestInfo( [
'IPAddress' =>
'10.64.24.52',
'UserAgent' =>
'meow' ] );
756 '1@542#c47dcfb0566e7d7bc110a6128a45c93a',
757 LBFactory::makeCookieValueFromCPIndex( 1, 542, $agentId )
762 LBFactory::getCPInfoFromCookieValue(
"5#$agentId",
$time - 10 )[
'index'],
767 LBFactory::getCPInfoFromCookieValue(
"5@$time",
$time - 10 )[
'index'],
772 LBFactory::getCPInfoFromCookieValue(
"0@$time#$agentId",
$time - 10 )[
'index'],
778 LBFactory::getCPInfoFromCookieValue(
"2@$time#$agentId",
$time - 10 )[
'index'],
783 LBFactory::getCPInfoFromCookieValue(
"2@$time#$agentId",
$time + 9 - 10 )[
'index'],
788 LBFactory::getCPInfoFromCookieValue(
"0@$time#$agentId",
$time + 9 - 10 )[
'index'],
789 'Almost stale; bad index'
793 LBFactory::getCPInfoFromCookieValue(
"2@$time#$agentId",
$time + 11 - 10 )[
'index'],
799 LBFactory::getCPInfoFromCookieValue(
"5@$time#$agentId",
$time - 10 )[
'clientId'],
804 LBFactory::getCPInfoFromCookieValue(
"5@$time#$agentId",
$time + 11 - 10 )[
'clientId'],