tests/phpunit/includes/db/LBFactoryTest.php

   1 <?php
   2 /**
   3  * Holds tests for LBFactory abstract MediaWiki class.
   4  *
   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.
   9  *
  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.
  14  *
  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
  19  *
  20  * @group Database
  21  * @file
  22  * @author Antoine Musso
  23  * @copyright © 2013 Antoine Musso
  24  * @copyright © 2013 Wikimedia Foundation Inc.
  25  */
  26 class LBFactoryTest extends MediaWikiTestCase {
  27
  28         /**
  29          * @dataProvider getLBFactoryClassProvider
  30          */
  31         public function testGetLBFactoryClass( $expected, $deprecated ) {
  32                 $mockDB = $this->getMockBuilder( 'DatabaseMysql' )
  33                         ->disableOriginalConstructor()
  34                         ->getMock();
  35
  36                 $config = array(
  37                         'class'          => $deprecated,
  38                         'connection'     => $mockDB,
  39                         # Various other parameters required:
  40                         'sectionsByDB'   => array(),
  41                         'sectionLoads'   => array(),
  42                         'serverTemplate' => array(),
  43                 );
  44
  45                 $this->hideDeprecated( '$wgLBFactoryConf must be updated. See RELEASE-NOTES for details' );
  46                 $result = LBFactory::getLBFactoryClass( $config );
  47
  48                 $this->assertEquals( $expected, $result );
  49         }
  50
  51         public function getLBFactoryClassProvider() {
  52                 return array(
  53                         # Format: new class, old class
  54                         array( 'LBFactorySimple', 'LBFactory_Simple' ),
  55                         array( 'LBFactorySingle', 'LBFactory_Single' ),
  56                         array( 'LBFactoryMulti', 'LBFactory_Multi' ),
  57                         array( 'LBFactoryFake', 'LBFactory_Fake' ),
  58                 );
  59         }
  60
  61         public function testLBFactorySimpleServer() {
  62                 $this->setMwGlobals( 'wgDBservers', false );
  63
  64                 $factory = new LBFactorySimple( array() );
  65                 $lb = $factory->getMainLB();
  66
  67                 $dbw = $lb->getConnection( DB_MASTER );
  68                 $this->assertTrue( $dbw->getLBInfo( 'master' ), 'master shows as master' );
  69
  70                 $dbr = $lb->getConnection( DB_SLAVE );
  71                 $this->assertTrue( $dbr->getLBInfo( 'master' ), 'DB_SLAVE also gets the master' );
  72
  73                 $factory->shutdown();
  74                 $lb->closeAll();
  75         }
  76
  77         public function testLBFactorySimpleServers() {
  78                 global $wgDBserver, $wgDBname, $wgDBuser, $wgDBpassword, $wgDBtype;
  79
  80                 $this->setMwGlobals( 'wgDBservers', array(
  81                         array( // master
  82                                 'host'          => $wgDBserver,
  83                                 'dbname'    => $wgDBname,
  84                                 'user'          => $wgDBuser,
  85                                 'password'      => $wgDBpassword,
  86                                 'type'          => $wgDBtype,
  87                                 'load'      => 0,
  88                                 'flags'     => DBO_TRX // REPEATABLE-READ for consistency
  89                         ),
  90                         array( // emulated slave
  91                                 'host'          => $wgDBserver,
  92                                 'dbname'    => $wgDBname,
  93                                 'user'          => $wgDBuser,
  94                                 'password'      => $wgDBpassword,
  95                                 'type'          => $wgDBtype,
  96                                 'load'      => 100,
  97                                 'flags'     => DBO_TRX // REPEATABLE-READ for consistency
  98                         )
  99                 ) );
 100
 101                 $factory = new LBFactorySimple( array( 'loadMonitorClass' => 'LoadMonitorNull' ) );
 102                 $lb = $factory->getMainLB();
 103
 104                 $dbw = $lb->getConnection( DB_MASTER );
 105                 $this->assertTrue( $dbw->getLBInfo( 'master' ), 'master shows as master' );
 106
 107                 $dbr = $lb->getConnection( DB_SLAVE );
 108                 $this->assertTrue( $dbr->getLBInfo( 'slave' ), 'slave shows as slave' );
 109
 110                 $factory->shutdown();
 111                 $lb->closeAll();
 112         }
 113
 114         public function testLBFactoryMulti() {
 115                 global $wgDBserver, $wgDBname, $wgDBuser, $wgDBpassword, $wgDBtype;
 116
 117                 $factory = new LBFactoryMulti( array(
 118                         'sectionsByDB' => array(),
 119                         'sectionLoads' => array(
 120                                 'DEFAULT' => array(
 121                                         'test-db1' => 0,
 122                                         'test-db2' => 100,
 123                                 ),
 124                         ),
 125                         'serverTemplate' => array(
 126                                 'dbname'          => $wgDBname,
 127                                 'user'            => $wgDBuser,
 128                                 'password'        => $wgDBpassword,
 129                                 'type'            => $wgDBtype,
 130                                 'flags'           => DBO_DEFAULT
 131                         ),
 132                         'hostsByName' => array(
 133                                 'test-db1'  => $wgDBserver,
 134                                 'test-db2'  => $wgDBserver
 135                         ),
 136                         'loadMonitorClass' => 'LoadMonitorNull'
 137                 ) );
 138                 $lb = $factory->getMainLB();
 139
 140                 $dbw = $lb->getConnection( DB_MASTER );
 141                 $this->assertTrue( $dbw->getLBInfo( 'master' ), 'master shows as master' );
 142
 143                 $dbr = $lb->getConnection( DB_SLAVE );
 144                 $this->assertTrue( $dbr->getLBInfo( 'slave' ), 'slave shows as slave' );
 145
 146                 $factory->shutdown();
 147                 $lb->closeAll();
 148         }
 149
 150         public function testChronologyProtector() {
 151                 // (a) First HTTP request
 152                 $mPos = new MySQLMasterPos( 'db1034-bin.000976', '843431247' );
 153
 154                 $mockDB = $this->getMockBuilder( 'DatabaseMysql' )
 155                         ->disableOriginalConstructor()
 156                         ->getMock();
 157                 $mockDB->expects( $this->any() )
 158                         ->method( 'doneWrites' )->will( $this->returnValue( true ) );
 159                 $mockDB->expects( $this->any() )
 160                         ->method( 'getMasterPos' )->will( $this->returnValue( $mPos ) );
 161
 162                 $lb = $this->getMockBuilder( 'LoadBalancer' )
 163                         ->disableOriginalConstructor()
 164                         ->getMock();
 165                 $lb->expects( $this->any() )
 166                         ->method( 'getConnection' )->will( $this->returnValue( $mockDB ) );
 167                 $lb->expects( $this->any() )
 168                         ->method( 'getServerCount' )->will( $this->returnValue( 2 ) );
 169                 $lb->expects( $this->any() )
 170                         ->method( 'parentInfo' )->will( $this->returnValue( array( 'id' => "main-DEFAULT" ) ) );
 171                 $lb->expects( $this->any() )
 172                         ->method( 'getAnyOpenConnection' )->will( $this->returnValue( $mockDB ) );
 173
 174                 $bag = new HashBagOStuff();
 175                 $cp = new ChronologyProtector(
 176                         $bag,
 177                         array(
 178                                 'ip' => '127.0.0.1',
 179                                 'agent' => "Totally-Not-FireFox"
 180                         )
 181                 );
 182
 183                 $mockDB->expects( $this->exactly( 2 ) )->method( 'doneWrites' );
 184
 185                 // Nothing to wait for
 186                 $cp->initLB( $lb );
 187                 // Record in stash
 188                 $cp->shutdownLB( $lb );
 189                 $cp->shutdown();
 190
 191                 // (b) Second HTTP request
 192                 $cp = new ChronologyProtector(
 193                         $bag,
 194                         array(
 195                                 'ip' => '127.0.0.1',
 196                                 'agent' => "Totally-Not-FireFox"
 197                         )
 198                 );
 199
 200                 $lb->expects( $this->once() )
 201                         ->method( 'waitFor' )->with( $this->equalTo( $mPos ) );
 202
 203                 // Wait
 204                 $cp->initLB( $lb );
 205                 // Record in stash
 206                 $cp->shutdownLB( $lb );
 207                 $cp->shutdown();
 208         }
 209 }