wiki.techinc.nl/tests/phpunit/includes/db/LBFactoryTest.php

<?php
/**
 * Holds tests for LBFactory abstract MediaWiki class.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 * http://www.gnu.org/copyleft/gpl.html
 *
 * @group Database
 * @file
 * @author Antoine Musso
 * @copyright © 2013 Antoine Musso
 * @copyright © 2013 Wikimedia Foundation Inc.
 */
class LBFactoryTest extends MediaWikiTestCase {

	/**
	 * @dataProvider getLBFactoryClassProvider
	 */
	public function testGetLBFactoryClass( $expected, $deprecated ) {
		$mockDB = $this->getMockBuilder( 'DatabaseMysql' )
			->disableOriginalConstructor()
			->getMock();

		$config = [
			'class'          => $deprecated,
			'connection'     => $mockDB,
			# Various other parameters required:
			'sectionsByDB'   => [],
			'sectionLoads'   => [],
			'serverTemplate' => [],
		];

		$this->hideDeprecated( '$wgLBFactoryConf must be updated. See RELEASE-NOTES for details' );
		$result = LBFactoryMW::getLBFactoryClass( $config );

		$this->assertEquals( $expected, $result );
	}

	public function getLBFactoryClassProvider() {
		return [
			# Format: new class, old class
			[ 'LBFactorySimple', 'LBFactory_Simple' ],
			[ 'LBFactorySingle', 'LBFactory_Single' ],
			[ 'LBFactoryMulti', 'LBFactory_Multi' ],
		];
	}

	public function testLBFactorySimpleServer() {
		global $wgDBserver, $wgDBname, $wgDBuser, $wgDBpassword, $wgDBtype;

		$servers = [
			[
				'host'      => $wgDBserver,
				'dbname'    => $wgDBname,
				'user'      => $wgDBuser,
				'password'  => $wgDBpassword,
				'type'      => $wgDBtype,
				'load'      => 0,
				'flags'     => DBO_TRX // REPEATABLE-READ for consistency
			],
		];

		$factory = new LBFactorySimple( [ 'servers' => $servers ] );
		$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;

		$servers = [
			[ // master
				'host'     => $wgDBserver,
				'dbname'   => $wgDBname,
				'user'     => $wgDBuser,
				'password' => $wgDBpassword,
				'type'     => $wgDBtype,
				'load'     => 0,
				'flags'    => DBO_TRX // REPEATABLE-READ for consistency
			],
			[ // emulated slave
				'host'     => $wgDBserver,
				'dbname'   => $wgDBname,
				'user'     => $wgDBuser,
				'password' => $wgDBpassword,
				'type'     => $wgDBtype,
				'load'     => 100,
				'flags'    => DBO_TRX // REPEATABLE-READ for consistency
			]
		];

		$factory = new LBFactorySimple( [
			'servers' => $servers,
			'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( 'replica' ), '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( [
			'sectionsByDB' => [],
			'sectionLoads' => [
				'DEFAULT' => [
					'test-db1' => 0,
					'test-db2' => 100,
				],
			],
			'serverTemplate' => [
				'dbname'	  => $wgDBname,
				'user'		  => $wgDBuser,
				'password'	  => $wgDBpassword,
				'type'		  => $wgDBtype,
				'flags'		  => DBO_DEFAULT
			],
			'hostsByName' => [
				'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( 'replica' ), 'slave shows as slave' );

		$factory->shutdown();
		$lb->closeAll();
	}

	public function testChronologyProtector() {
		// (a) First HTTP request
		$mPos = new MySQLMasterPos( 'db1034-bin.000976', '843431247' );

		$now = microtime( true );
		$mockDB = $this->getMockBuilder( 'DatabaseMysql' )
			->disableOriginalConstructor()
			->getMock();
		$mockDB->method( 'writesOrCallbacksPending' )->willReturn( true );
		$mockDB->method( 'lastDoneWrites' )->willReturn( $now );
		$mockDB->method( 'getMasterPos' )->willReturn( $mPos );

		$lb = $this->getMockBuilder( 'LoadBalancer' )
			->disableOriginalConstructor()
			->getMock();
		$lb->method( 'getConnection' )->willReturn( $mockDB );
		$lb->method( 'getServerCount' )->willReturn( 2 );
		$lb->method( 'parentInfo' )->willReturn( [ 'id' => "main-DEFAULT" ] );
		$lb->method( 'getAnyOpenConnection' )->willReturn( $mockDB );
		$lb->method( 'hasOrMadeRecentMasterChanges' )->will( $this->returnCallback(
				function () use ( $mockDB ) {
					$p = 0;
					$p |= call_user_func( [ $mockDB, 'writesOrCallbacksPending' ] );
					$p |= call_user_func( [ $mockDB, 'lastDoneWrites' ] );

					return (bool)$p;
				}
			) );
		$lb->method( 'getMasterPos' )->willReturn( $mPos );

		$bag = new HashBagOStuff();
		$cp = new ChronologyProtector(
			$bag,
			[
				'ip' => '127.0.0.1',
				'agent' => "Totally-Not-FireFox"
			]
		);

		$mockDB->expects( $this->exactly( 2 ) )->method( 'writesOrCallbacksPending' );
		$mockDB->expects( $this->exactly( 2 ) )->method( 'lastDoneWrites' );

		// Nothing to wait for
		$cp->initLB( $lb );
		// Record in stash
		$cp->shutdownLB( $lb );
		$cp->shutdown();

		// (b) Second HTTP request
		$cp = new ChronologyProtector(
			$bag,
			[
				'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();
	}

	private function newLBFactoryMulti( array $baseOverride = [], array $serverOverride = [] ) {
		global $wgDBserver, $wgDBuser, $wgDBpassword, $wgDBname, $wgDBtype;

		return new LBFactoryMulti( $baseOverride + [
			'sectionsByDB' => [],
			'sectionLoads' => [
				'DEFAULT' => [
					'test-db1' => 1,
				],
			],
			'serverTemplate' => $serverOverride + [
				'dbname' => $wgDBname,
				'user' => $wgDBuser,
				'password' => $wgDBpassword,
				'type' => $wgDBtype,
				'flags' => DBO_DEFAULT
			],
			'hostsByName' => [
				'test-db1' => $wgDBserver,
			],
			'loadMonitorClass' => 'LoadMonitorNull',
			'localDomain' => wfWikiID()
		] );
	}

	public function testNiceDomains() {
		global $wgDBname;

		$factory = $this->newLBFactoryMulti();
		$lb = $factory->getMainLB();

		$db = $lb->getConnectionRef( DB_MASTER );
		$this->assertEquals(
			$wgDBname,
			$db->getDomainID()
		);
		unset( $db );

		/** @var DatabaseBase $db */
		$db = $lb->getConnection( DB_MASTER, [], '' );

		$this->assertEquals(
			'',
			$db->getDomainID()
		);

		$this->assertEquals(
			$db->addIdentifierQuotes( 'page' ),
			$db->tableName( 'page' ),
			"Correct full table name"
		);

		$this->assertEquals(
			$db->addIdentifierQuotes( $wgDBname ) . '.' . $db->addIdentifierQuotes( 'page' ),
			$db->tableName( "$wgDBname.page" ),
			"Correct full table name"
		);

		$this->assertEquals(
			$db->addIdentifierQuotes( 'nice_db' ) . '.' . $db->addIdentifierQuotes( 'page' ),
			$db->tableName( 'nice_db.page' ),
			"Correct full table name"
		);

		$factory->setDomainPrefix( 'my_' );
		$this->assertEquals(
			'',
			$db->getDomainID()
		);
		$this->assertEquals(
			$db->addIdentifierQuotes( 'my_page' ),
			$db->tableName( 'page' ),
			"Correct full table name"
		);
		$this->assertEquals(
			$db->addIdentifierQuotes( 'other_nice_db' ) . '.' . $db->addIdentifierQuotes( 'page' ),
			$db->tableName( 'other_nice_db.page' ),
			"Correct full table name"
		);

		$factory->closeAll();
		$factory->destroy();
	}

	public function testTrickyDomain() {
		$dbname = 'unittest-domain';
		$factory = $this->newLBFactoryMulti(
			[ 'localDomain' => $dbname ], [ 'dbname' => $dbname ] );
		$lb = $factory->getMainLB();
		/** @var DatabaseBase $db */
		$db = $lb->getConnection( DB_MASTER, [], '' );

		$this->assertEquals(
			'',
			$db->getDomainID()
		);

		$this->assertEquals(
			$db->addIdentifierQuotes( 'page' ),
			$db->tableName( 'page' ),
			"Correct full table name"
		);

		$this->assertEquals(
			$db->addIdentifierQuotes( $dbname ) . '.' . $db->addIdentifierQuotes( 'page' ),
			$db->tableName( "$dbname.page" ),
			"Correct full table name"
		);

		$this->assertEquals(
			$db->addIdentifierQuotes( 'nice_db' ) . '.' . $db->addIdentifierQuotes( 'page' ),
			$db->tableName( 'nice_db.page' ),
			"Correct full table name"
		);

		$factory->setDomainPrefix( 'my_' );

		$this->assertEquals(
			$db->addIdentifierQuotes( 'my_page' ),
			$db->tableName( 'page' ),
			"Correct full table name"
		);
		$this->assertEquals(
			$db->addIdentifierQuotes( 'other_nice_db' ) . '.' . $db->addIdentifierQuotes( 'page' ),
			$db->tableName( 'other_nice_db.page' ),
			"Correct full table name"
		);

		\MediaWiki\suppressWarnings();
		$this->assertFalse( $db->selectDB( 'garbage-db' ) );
		\MediaWiki\restoreWarnings();

		$this->assertEquals(
			$db->addIdentifierQuotes( 'garbage-db' ) . '.' . $db->addIdentifierQuotes( 'page' ),
			$db->tableName( 'garbage-db.page' ),
			"Correct full table name"
		);

		$factory->closeAll();
		$factory->destroy();
	}
}