499a1fe497
We also bring along Database::attributesFromType(), which relied on the private ::getClass(). This requires us to inject DatabaseFactory through the LBFactory/LoadBalancer hierarchy. Database::factory() is now soft deprecated. All callers outside of includes/installer/ still need migration. Bug: T299691 Bug: T315270 Change-Id: I7d057a9438f1b097554679975e4e9b2fc99e7c2b
106 lines
3.1 KiB
PHP
106 lines
3.1 KiB
PHP
<?php
|
|
/**
|
|
* 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
|
|
*
|
|
* @file
|
|
*/
|
|
|
|
use MediaWiki\Config\ServiceOptions;
|
|
use Wikimedia\Rdbms\DatabaseDomain;
|
|
use Wikimedia\Rdbms\DatabaseFactory;
|
|
use Wikimedia\Rdbms\LBFactorySimple;
|
|
use Wikimedia\RequestTimeout\CriticalSectionProvider;
|
|
use Wikimedia\RequestTimeout\RequestTimeout;
|
|
|
|
/**
|
|
* @covers \Wikimedia\Rdbms\LBFactory
|
|
* @covers \Wikimedia\Rdbms\LBFactorySimple
|
|
* @covers \Wikimedia\Rdbms\LBFactoryMulti
|
|
*/
|
|
class MWLBFactoryTest extends MediaWikiUnitTestCase {
|
|
|
|
private function newMWLBFactory() {
|
|
return new MWLBFactory(
|
|
new ServiceOptions( [], [] ),
|
|
new ConfiguredReadOnlyMode( 'Test' ),
|
|
new EmptyBagOStuff(),
|
|
new EmptyBagOStuff(),
|
|
new WANObjectCache( [ 'cache' => new EmptyBagOStuff() ] ),
|
|
new CriticalSectionProvider( RequestTimeout::singleton(), 1, null, null ),
|
|
new NullStatsdDataFactory(),
|
|
new DatabaseFactory()
|
|
);
|
|
}
|
|
|
|
/**
|
|
* @covers MWLBFactory::getLBFactoryClass
|
|
* @dataProvider getLBFactoryClassProvider
|
|
*/
|
|
public function testGetLBFactoryClass( $config, $expected ) {
|
|
$this->assertEquals(
|
|
$expected,
|
|
$this->newMWLBFactory()->getLBFactoryClass( $config )
|
|
);
|
|
}
|
|
|
|
public function getLBFactoryClassProvider() {
|
|
yield 'undercore alias default' => [
|
|
[ 'class' => 'LBFactory_Simple' ],
|
|
Wikimedia\Rdbms\LBFactorySimple::class,
|
|
];
|
|
yield 'short alias multi' => [
|
|
[ 'class' => 'LBFactoryMulti' ],
|
|
Wikimedia\Rdbms\LBFactoryMulti::class,
|
|
];
|
|
}
|
|
|
|
/**
|
|
* @covers MWLBFactory::setDomainAliases()
|
|
* @dataProvider setDomainAliasesProvider
|
|
*/
|
|
public function testDomainAliases( $dbname, $prefix, $expectedDomain ) {
|
|
$servers = [ [
|
|
'type' => 'sqlite',
|
|
'dbname' => 'defaultdb',
|
|
'tablePrefix' => 'defaultprefix_',
|
|
'dbDirectory' => '~/sqldatadir/',
|
|
'load' => 0,
|
|
] ];
|
|
$lbFactory = new LBFactorySimple( [
|
|
'servers' => $servers,
|
|
'localDomain' => new DatabaseDomain( $dbname, null, $prefix )
|
|
] );
|
|
$this->newMWLBFactory()->setDomainAliases( $lbFactory );
|
|
|
|
$rawDomain = rtrim( "$dbname-$prefix", '-' );
|
|
$this->assertEquals(
|
|
$expectedDomain,
|
|
$lbFactory->resolveDomainID( $rawDomain ),
|
|
'Domain aliases set'
|
|
);
|
|
}
|
|
|
|
public function setDomainAliasesProvider() {
|
|
return [
|
|
[ 'enwiki', '', 'enwiki' ],
|
|
[ 'wikipedia', 'fr_', 'wikipedia-fr_' ],
|
|
[ 'wikipedia', 'zh', 'wikipedia-zh' ],
|
|
[ 'wiki-pedia', '', 'wiki?hpedia' ],
|
|
[ 'wiki-pedia', 'es_', 'wiki?hpedia-es_' ],
|
|
[ 'wiki-pedia', 'ru', 'wiki?hpedia-ru' ]
|
|
];
|
|
}
|
|
}
|