array( 'LBFactoryFake', 'LBFactory_Fake' ),
);
}
+
+ public function testLBFactorySimpleServer() {
+ $this->setMwGlobals( 'wgDBservers', false );
+
+ $factory = new LBFactorySimple( array() );
+ $lb = $factory->getMainLB();
+
+ $dbw = $lb->getConnection( DB_MASTER );
+ $this->assertTrue( $dbw->getLBInfo( 'master' ), 'master shows as master' );
+
+ $dbr = $lb->getConnection( DB_SLAVE );
+ $this->assertTrue( $dbr->getLBInfo( 'master' ), 'DB_SLAVE also gets the master' );
+
+ $factory->shutdown();
+ $lb->closeAll();
+ }
+
+ public function testLBFactorySimpleServers() {
+ global $wgDBserver, $wgDBname, $wgDBuser, $wgDBpassword, $wgDBtype;
+
+ $this->setMwGlobals( 'wgDBservers', array(
+ array( // master
+ 'host' => $wgDBserver,
+ 'dbname' => $wgDBname,
+ 'user' => $wgDBuser,
+ 'password' => $wgDBpassword,
+ 'type' => $wgDBtype,
+ 'load' => 0,
+ 'flags' => DBO_TRX // REPEATABLE-READ for consistency
+ ),
+ array( // emulated slave
+ 'host' => $wgDBserver,
+ 'dbname' => $wgDBname,
+ 'user' => $wgDBuser,
+ 'password' => $wgDBpassword,
+ 'type' => $wgDBtype,
+ 'load' => 100,
+ 'flags' => DBO_TRX // REPEATABLE-READ for consistency
+ )
+ ) );
+
+ $factory = new LBFactorySimple( array( 'loadMonitorClass' => 'LoadMonitorNull' ) );
+ $lb = $factory->getMainLB();
+
+ $dbw = $lb->getConnection( DB_MASTER );
+ $this->assertTrue( $dbw->getLBInfo( 'master' ), 'master shows as master' );
+ $this->assertEquals(
+ $wgDBserver, $dbw->getLBInfo( 'clusterMasterHost' ), 'cluster master set' );
+
+ $dbr = $lb->getConnection( DB_SLAVE );
+ $this->assertTrue( $dbr->getLBInfo( 'slave' ), 'slave shows as slave' );
+ $this->assertEquals(
+ $wgDBserver, $dbr->getLBInfo( 'clusterMasterHost' ), 'cluster master set' );
+
+ $factory->shutdown();
+ $lb->closeAll();
+ }
+
+ public function testLBFactoryMulti() {
+ global $wgDBserver, $wgDBname, $wgDBuser, $wgDBpassword, $wgDBtype;
+
+ $factory = new LBFactoryMulti( array(
+ 'sectionsByDB' => array(),
+ 'sectionLoads' => array(
+ 'DEFAULT' => array(
+ 'test-db1' => 0,
+ 'test-db2' => 100,
+ ),
+ ),
+ 'serverTemplate' => array(
+ 'dbname' => $wgDBname,
+ 'user' => $wgDBuser,
+ 'password' => $wgDBpassword,
+ 'type' => $wgDBtype,
+ 'flags' => DBO_DEFAULT
+ ),
+ 'hostsByName' => array(
+ 'test-db1' => $wgDBserver,
+ 'test-db2' => $wgDBserver
+ ),
+ 'loadMonitorClass' => 'LoadMonitorNull'
+ ) );
+ $lb = $factory->getMainLB();
+
+ $dbw = $lb->getConnection( DB_MASTER );
+ $this->assertTrue( $dbw->getLBInfo( 'master' ), 'master shows as master' );
+
+ $dbr = $lb->getConnection( DB_SLAVE );
+ $this->assertTrue( $dbr->getLBInfo( 'slave' ), 'slave shows as slave' );
+
+ $factory->shutdown();
+ $lb->closeAll();
+ }
+
+ public function testChronologyProtector() {
+ // (a) First HTTP request
+ $mPos = new MySQLMasterPos( 'db1034-bin.000976', '843431247' );
+
+ $mockDB = $this->getMockBuilder( 'DatabaseMysql' )
+ ->disableOriginalConstructor()
+ ->getMock();
+ $mockDB->expects( $this->any() )
+ ->method( 'doneWrites' )->will( $this->returnValue( true ) );
+ $mockDB->expects( $this->any() )
+ ->method( 'getMasterPos' )->will( $this->returnValue( $mPos ) );
+
+ $lb = $this->getMockBuilder( 'LoadBalancer' )
+ ->disableOriginalConstructor()
+ ->getMock();
+ $lb->expects( $this->any() )
+ ->method( 'getConnection' )->will( $this->returnValue( $mockDB ) );
+ $lb->expects( $this->any() )
+ ->method( 'getServerCount' )->will( $this->returnValue( 2 ) );
+ $lb->expects( $this->any() )
+ ->method( 'parentInfo' )->will( $this->returnValue( array( 'id' => "main-DEFAULT" ) ) );
+ $lb->expects( $this->any() )
+ ->method( 'getAnyOpenConnection' )->will( $this->returnValue( $mockDB ) );
+
+ $bag = new HashBagOStuff();
+ $cp = new ChronologyProtector(
+ $bag,
+ array(
+ 'ip' => '127.0.0.1',
+ 'agent' => "Totally-Not-FireFox"
+ )
+ );
+
+ $mockDB->expects( $this->exactly( 2 ) )->method( 'doneWrites' );
+
+ // Nothing to wait for
+ $cp->initLB( $lb );
+ // Record in stash
+ $cp->shutdownLB( $lb );
+ $cp->shutdown();
+
+ // (b) Second HTTP request
+ $cp = new ChronologyProtector(
+ $bag,
+ array(
+ 'ip' => '127.0.0.1',
+ 'agent' => "Totally-Not-FireFox"
+ )
+ );
+
+ $lb->expects( $this->once() )
+ ->method( 'waitFor' )->with( $this->equalTo( $mPos ) );
+
+ // Wait
+ $cp->initLB( $lb );
+ // Record in stash
+ $cp->shutdownLB( $lb );
+ $cp->shutdown();
+ }
}