NameTableStoreFactoryTest.php

Go to the documentation of this file.

<?php
 
namespace MediaWiki\Tests\Storage;
 
use MediaWiki\Logger\LoggerFactory;
use MediaWiki\MediaWikiServices;
use MediaWiki\Storage\NameTableStore;
use MediaWiki\Storage\NameTableStoreFactory;
use MediaWikiTestCase;
use Wikimedia\Rdbms\ILBFactory;
use Wikimedia\Rdbms\ILoadBalancer;
 
class NameTableStoreFactoryTest extends MediaWikiTestCase {
    private function getMockLoadBalancer( $localDomain ) {
        $mock = $this->getMockBuilder( ILoadBalancer::class )
            ->disableOriginalConstructor()->getMock();
 
        $mock->expects( $this->any() )
            ->method( 'getLocalDomainID' )
            ->willReturn( $localDomain );
 
        return $mock;
    }
 
    private function getMockLoadBalancerFactory( $expectedWiki ) {
        $mock = $this->getMockBuilder( ILBFactory::class )
            ->disableOriginalConstructor()->getMock();
 
        $lbFactory = MediaWikiServices::getInstance()->getDBLoadBalancerFactory();
        $localDomain = $lbFactory->getLocalDomainID();
 
        $mock->expects( $this->any() )->method( 'getLocalDomainID' )->willReturn( $localDomain );
 
        $mock->expects( $this->once() )
            ->method( 'getMainLB' )
            ->with( $this->equalTo( $expectedWiki ) )
            ->willReturnCallback( function ( $domain ) use ( $localDomain ) {
                return $this->getMockLoadBalancer( $localDomain );
            } );
 
        return $mock;
    }
 
    public static function provideTestGet() {
        return [
            [
                'change_tag_def',
                false,
                false,
            ],
            [
                'content_models',
                false,
                false,
            ],
            [
                'slot_roles',
                false,
                false,
            ],
            [
                'change_tag_def',
                'test7245',
                'test7245',
            ],
        ];
    }
 
    public function testGet( $tableName, $wiki, $expectedWiki ) {
        $services = MediaWikiServices::getInstance();
        $wiki2 = ( $wiki === false )
            ? $services->getDBLoadBalancerFactory()->getLocalDomainID()
            : $wiki;
        $names = new NameTableStoreFactory(
            $this->getMockLoadBalancerFactory( $expectedWiki ),
            $services->getMainWANObjectCache(),
            LoggerFactory::getInstance( 'NameTableStoreFactory' )
        );
 
        $table = $names->get( $tableName, $wiki );
        $table2 = $names->get( $tableName, $wiki2 );
        $this->assertSame( $table, $table2 );
        $this->assertInstanceOf( NameTableStore::class, $table );
    }
 
    /*
     * The next three integration tests verify that the schema information is correct by loading
     * the relevant information from the database.
     */
 
    public function testIntegratedGetChangeTagDef() {
        $services = MediaWikiServices::getInstance();
        $factory = $services->getNameTableStoreFactory();
        $store = $factory->getChangeTagDef();
        $this->assertType( 'array', $store->getMap() );
    }
 
    public function testIntegratedGetContentModels() {
        $services = MediaWikiServices::getInstance();
        $factory = $services->getNameTableStoreFactory();
        $store = $factory->getContentModels();
        $this->assertType( 'array', $store->getMap() );
    }
 
    public function testIntegratedGetSlotRoles() {
        $services = MediaWikiServices::getInstance();
        $factory = $services->getNameTableStoreFactory();
        $store = $factory->getSlotRoles();
        $this->assertType( 'array', $store->getMap() );
    }
}