1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38
|
<?php
namespace MediaWiki\Tests\Block;
use MediaWiki\Block\BlockRestrictionStore;
use MediaWiki\Block\BlockRestrictionStoreFactory;
use MediaWiki\DAO\WikiAwareEntity;
use MediaWikiUnitTestCase;
use Wikimedia\Rdbms\LBFactory;
use Wikimedia\Rdbms\LoadBalancer;
/**
* @covers \MediaWiki\Block\BlockRestrictionStoreFactory
*/
class BlockRestrictionStoreFactoryTest extends MediaWikiUnitTestCase {
/**
* @dataProvider provideDomains
*/
public function testGetBlockRestrictionStore( $domain ) {
$lb = $this->createMock( LoadBalancer::class );
$lbFactory = $this->createMock( LBFactory::class );
$lbFactory
->method( 'getMainLB' )
->with( $domain )
->willReturn( $lb );
$factory = new BlockRestrictionStoreFactory( $lbFactory );
$restrictionStore = $factory->getBlockRestrictionStore( $domain );
$this->assertInstanceOf( BlockRestrictionStore::class, $restrictionStore );
}
public static function provideDomains() {
yield 'local wiki' => [ WikiAwareEntity::LOCAL ];
yield 'foreign wiki' => [ 'meta' ];
}
}
|